python/python-columnize
SHA256
15
0
Fork 0
forked from pool/python-columnize
Code Pull Requests Activity
Files
b9a5c89c4d8368f6beb29ce95b5f4e519a1c440a3258b00a5721e4d7f36aa643
python-columnize/test_columnize.py

179 lines
6.5 KiB
Python
Raw Normal View History

Accepting request 951766 from home:badshah400:branches:science New package - A python module to display simple lists in formatted columns. OBS-URL: https://build.opensuse.org/request/show/951766 OBS-URL: https://build.opensuse.org/package/show/devel:languages:python/python-columnize?expand=0&rev=1
2022-02-07 21:13:53 +00:00
#!/usr/bin/env python
# -*- Python -*-
"Unit test for Columnize"
import mock, operator, os, sys, unittest
top_builddir = os.path.join(os.path.dirname(__file__), os.path.pardir)
if top_builddir[-1] != os.path.sep:
top_builddir += os.path.sep
sys.path.insert(0, top_builddir)
from columnize import columnize
class TestColumize(unittest.TestCase):
def test_basic(self):
"""Basic sanity and status testing."""
self.assertEqual("1, 2, 3\n",
columnize(['1', '2', '3'], 10, ', '))
self.assertEqual("1 3\n2 4\n",
columnize(['1', '2', '3', '4'], 4))
self.assertEqual("1 3\n2 4\n",
columnize(['1', '2', '3', '4'], 7))
self.assertEqual("0 1 2\n3\n",
columnize(['0', '1', '2', '3'], 7,
arrange_vertical=False))
self.assertEqual("<empty>\n", columnize([]))
self.assertEqual("oneitem\n", columnize(["oneitem"]))
data = [str(i) for i in range(55)]
self.assertEqual(
"0, 6, 12, 18, 24, 30, 36, 42, 48, 54\n" +
"1, 7, 13, 19, 25, 31, 37, 43, 49\n" +
"2, 8, 14, 20, 26, 32, 38, 44, 50\n" +
"3, 9, 15, 21, 27, 33, 39, 45, 51\n" +
"4, 10, 16, 22, 28, 34, 40, 46, 52\n" +
"5, 11, 17, 23, 29, 35, 41, 47, 53\n",
columnize(data, displaywidth=39, ljust=False,
arrange_vertical=True, colsep=', '))
self.assertEqual(
" 0, 7, 14, 21, 28, 35, 42, 49\n" +
" 1, 8, 15, 22, 29, 36, 43, 50\n" +
" 2, 9, 16, 23, 30, 37, 44, 51\n" +
" 3, 10, 17, 24, 31, 38, 45, 52\n" +
" 4, 11, 18, 25, 32, 39, 46, 53\n" +
" 5, 12, 19, 26, 33, 40, 47, 54\n" +
" 6, 13, 20, 27, 34, 41, 48\n",
columnize(data, displaywidth=39, ljust=False,
arrange_vertical=True, colsep=', ',
lineprefix=' '))
self.assertEqual(
" 0, 1, 2, 3, 4, 5, 6, 7, 8, 9\n" +
"10, 11, 12, 13, 14, 15, 16, 17, 18, 19\n" +
"20, 21, 22, 23, 24, 25, 26, 27, 28, 29\n" +
"30, 31, 32, 33, 34, 35, 36, 37, 38, 39\n" +
"40, 41, 42, 43, 44, 45, 46, 47, 48, 49\n" +
"50, 51, 52, 53, 54\n",
columnize(data, displaywidth=39, ljust=False,
arrange_vertical=False, colsep=', '))
self.maxDiff = None
self.assertEqual(
" 0, 1, 2, 3, 4, 5, 6, 7\n" +
" 8, 9, 10, 11, 12, 13, 14, 15\n" +
" 16, 17, 18, 19, 20, 21, 22, 23\n" +
" 24, 25, 26, 27, 28, 29, 30, 31\n" +
" 32, 33, 34, 35, 36, 37, 38, 39\n" +
" 40, 41, 42, 43, 44, 45, 46, 47\n" +
" 48, 49, 50, 51, 52, 53, 54\n",
columnize(data, displaywidth=34, ljust=False,
arrange_vertical=False, colsep=', ',
lineprefix=' '))
data = (
"one", "two", "three",
"for", "five", "six",
"seven", "eight", "nine",
"ten", "eleven", "twelve",
"thirteen", "fourteen", "fifteen",
"sixteen", "seventeen", "eightteen",
"nineteen", "twenty", "twentyone",
"twentytwo", "twentythree", "twentyfour",
"twentyfive","twentysix", "twentyseven",)
self.assertEqual(
"one two three for five six \n" +
"seven eight nine ten eleven twelve \n" +
"thirteen fourteen fifteen sixteen seventeen eightteen \n" +
"nineteen twenty twentyone twentytwo twentythree twentyfour\n" +
"twentyfive twentysix twentyseven\n", columnize(data, arrange_vertical=False))
self.assertEqual(
"one five nine thirteen seventeen twentyone twentyfive \n" +
"two six ten fourteen eightteen twentytwo twentysix \n" +
"three seven eleven fifteen nineteen twentythree twentyseven\n" +
"for eight twelve sixteen twenty twentyfour \n", columnize(data))
self.assertEqual('0 1 2 3\n', columnize(list(range(4))))
self.assertEqual(
"[ 0, 1, 2, 3, 4, 5, 6, 7, 8,\n"+
" 9, 10, 11, 12, 13, 14, 15, 16, 17,\n"+
" 18, 19, 20, 21, 22, 23, 24, 25, 26,\n"+
" 27, 28, 29, 30, 31, 32, 33, 34, 35,\n"+
" 36, 37, 38, 39, 40, 41, 42, 43, 44,\n"+
" 45, 46, 47, 48, 49, 50, 51, 52, 53,\n"+
" 54]\n\n", columnize(list(range(55)),
opts={'displaywidth':38, 'arrange_array':True}))
self.assertEqual("""[ 0,
1,
2,
3,
4,
5,
6,
7,
8,
9,
10,
11]
""", columnize(list(range(12)),
opts={'displaywidth':6, 'arrange_array':True}))
self.assertEqual("""[ 0, 1,
2, 3,
4, 5,
6, 7,
8, 9,
10, 11]
""", columnize(list(range(12)),
opts={'displaywidth':9, 'arrange_array':True}))
return
def test_colfmt(self):
self.assertEqual(' 0 1 2 3\n',
columnize([0, 1, 2, 3], 7,
arrange_vertical=False,
opts={'colfmt': '%5d'}))
def test_lineprefix(self):
self.assertEqual('>>> 0\n>>> 1\n>>> 2\n>>> 3\n',
columnize([0, 1, 2, 3], 7,
arrange_vertical=False,
opts={'colfmt': '%5d',
'displaywidth': 16,
'lineprefix': '>>> '}))
def test_lineprefix_just_wide_enough(self):
self.assertEqual('>>>10 12\n>>>11 13\n',
columnize([10, 11, 12, 13],
opts={'lineprefix': '>>>',
'displaywidth': 9}))
if sys.version_info[:2] >= (3, 6):
@mock.patch.dict('os.environ', {'COLUMNS': '87'}, clear=True)
def test_computed_displaywidth_environ_COLUMNS_set(self):
from columnize import computed_displaywidth
width = computed_displaywidth()
self.assertEqual(width, 87)
def test_errors(self):
"""Test various error conditions."""
self.assertRaises(TypeError, columnize, 5, 'reject input - not array')
return
pass
if __name__ == '__main__':
unittest.main()
Copy Permalink