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 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
