def setUp(self): from body import Body from database import AbilitiesDataBase from abilities import a_test ab_db = AbilitiesDataBase() ab_db.addAbility(0,a_test.Test()) self.b = Body() self.b.abilitiesDataBase(ab_db)
def setUp(self): from database import AbilitiesDataBase self.a_db = AbilitiesDataBase()
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)
def generateDataBase(self): """ Generates abilities list Input: Nothing Returns: Abilities """ from jarvis import Jarvis from database import AbilitiesDataBase from abilities import a_test from abilities import a_time from abilities import a_hello from abilities import a_exit from abilities import a_mediacenter from abilities import a_weather from abilities import a_didyouknow if not isinstance(self._jarvis, Jarvis): raise DataBaseBuilderException abilities_list = AbilitiesDataBase() abilities_list.addAbility(0, a_test.Test()) abilities_list.addAbility(1, a_time.Time()) abilities_list.addAbility(2, a_hello.Hello()) exit_ability = a_exit.Exit() exit_ability.jarvis(self._jarvis) abilities_list.addAbility(3, exit_ability) mc = a_mediacenter.MediaCenter() abilities_list.addAbility(4, mc.factoryBBCRadioStart()) abilities_list.addAbility(5, mc.factoryStop()) abilities_list.addAbility(6, a_weather.Weather()) abilities_list.addAbility(7, a_didyouknow.DidYouKnow()) return abilities_list