def setUp(self): self.test_table = CompareSectionCsvTable() self.doc = create_compare_test() odml.tools.xmlparser.XMLWriter(self.doc).write_file('comparetest.odml') self.test_table.load_from_file('comparetest.odml') self.test_table.choose_sections_startwith('')
class TestCompareCsv(unittest.TestCase): def setUp(self): self.test_table = CompareSectionCsvTable() self.doc = create_compare_test() odml.tools.xmlparser.XMLWriter(self.doc).write_file('comparetest.odml') self.test_table.load_from_file('comparetest.odml') self.test_table.choose_sections_startwith('') def tearDown(self): os.remove('comparetest.odml') os.remove('test.csv') def test_includeall_true(self): self.test_table.include_all = True self.test_table.switch = False self.test_table.write2file('test.csv') with open('test.csv', 'rb') as csvfile: csvreader = csv.reader(csvfile) row_num = 0 for row in csvreader: self.assertEqual(row, include_true_expected[row_num]) row_num += 1 def test_includeall_false(self): self.test_table.include_all = False self.test_table.switch = False self.test_table.choose_sections_startwith('S') self.test_table.write2file('test.csv') with open('test.csv', 'rb') as csvfile: csvreader = csv.reader(csvfile) row_num = 0 for row in csvreader: self.assertEqual(row, include_false_expected[row_num]) row_num += 1 def test_switch(self): self.test_table.switch = True self.test_table.include_all = True self.test_table.write2file('test.csv') with open('test.csv', 'rb') as csvfile: csvreader = csv.reader(csvfile) row_num = 0 for row in csvreader: self.assertEqual(row, switch_expected[row_num]) row_num += 1
class TestCompareCsv(unittest.TestCase): def setUp(self): self.test_table = CompareSectionCsvTable() self.doc = create_compare_test() odml.tools.xmlparser.XMLWriter(self.doc).write_file('comparetest.odml') self.test_table.load_from_file('comparetest.odml') self.test_table.choose_sections_startwith('') def tearDown(self): os.remove('comparetest.odml') os.remove('test.csv') def test_includeall_true(self): self.test_table.include_all = True self.test_table.switch = False self.test_table.write2file('test.csv') with open('test.csv', 'r') as csvfile: csvreader = csv.reader(csvfile) row_num = 0 for row in csvreader: self.assertEqual(row, include_true_expected[row_num]) row_num += 1 def test_includeall_false(self): self.test_table.include_all = False self.test_table.switch = False self.test_table.choose_sections_startwith('S') self.test_table.write2file('test.csv') with open('test.csv', 'r') as csvfile: csvreader = csv.reader(csvfile) row_num = 0 for row in csvreader: self.assertEqual(row, include_false_expected[row_num]) row_num += 1 def test_switch(self): self.test_table.switch = True self.test_table.include_all = True self.test_table.write2file('test.csv') with open('test.csv', 'r') as csvfile: csvreader = csv.reader(csvfile) row_num = 0 for row in csvreader: self.assertEqual(row, switch_expected[row_num]) row_num += 1