class AbilitiesDataBaseTest(unittest.TestCase): def setUp(self): from database import AbilitiesDataBase self.a_db = AbilitiesDataBase() def test_addAbility_exception(self): from database import DataBaseException from abilities import Ability with self.assertRaises(DataBaseException): self.a_db.addAbility(0,None) ab1 = Ability() ab2 = Ability() id1 = id2 = 1 self.a_db.addAbility(id1,ab1) with self.assertRaises(DataBaseException): self.a_db.addAbility(id1,ab2) def test_addAbility(self): from abilities import Ability ab = Ability() id = 0 self.a_db.addAbility(id,ab) self.assertEqual(ab,self.a_db.getAbility(id)) def test_getAbility_exception(self): from database import DataBaseException with self.assertRaises(DataBaseException): self.a_db.getAbility(0)