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)