class TestCSV(unittest.TestCase):
def setUp(self):
self.csv = CSV(INFILE)
def test_echo_output(self):
self.assertEqual(self.csv.toStr(), read(INFILE))
def test_columns(self):
self.csv.colspec = '2,7,8'
self.assertEqual(self.csv.toStr(), read(TEST_2_7_8))
def test_columns_rearrange(self):
self.csv.colspec = '26,2'
self.assertEqual(self.csv.toStr(), read(TEST_26_2))
def test_columns_range(self):
self.csv.colspec = '24-26,2'
self.assertEqual(self.csv.toStr(), read(TEST_24_to_26_2))
def test_pretty(self):
self.csv.pretty_print = True
self.assertEqual(self.csv.toStr(), read(TEST_PRETTY))
def test_show_header(self):
self.csv.show_header = True
self.assertEqual(self.csv.toStr(), read(TEST_SHOW_HEADER))
def setUp(self):
self.csv = CSV(INFILE)
