class DBHandlerTester(unittest.TestCase): def setUp(self): self.handler = DBHandler() def tearDown(self): self.handler.closeCursor() def testCreateNewTableAndInsertRow(self): self.handler.createNewTable(['c1', 'c2', 'c3', 'c4']) row1 = ['r1c1', 'r1c2', 'r1c3', 'r1c4'] self.handler.insertRow(row1) row2 = ['r2c1', 'r2c2', 'r2c3', 'r3c4'] self.handler.insertRow(row2) results = self.handler.executeQuery('select * from xmldata') # each returned row is a tuple, so has to be converted to list for comparison assert list(results[0]) == row1 and list(results[1]) == row2, results def testGetColumnNames(self): columns = ['c1', 'c2', 'c3', 'c4'] self.handler.createNewTable(columns) results = self.handler.getColumnNames() assert results == columns, results