class TestDataBase(TestCase): def setUp(self): self.db = DataBase() def test_create_non_existing_table(self): self.assertFalse(self.db.table_exists('persons')) self.db.create_table('persons') self.assertTrue(self.db.table_exists('persons')) def test_create_existing_table(self): self.assertFalse(self.db.table_exists('persons')) self.db.create_table('persons') bTableExists = False try: self.db.create_table('persons') except DBError, e: bTableExists = True self.assertTrue(bTableExists)