def test_get_action_list(self): mdp = MDP() mdp.add_action(0) mdp.add_action(1) action_list = mdp.get_action_list() self.assertEqual(len(action_list), 2) self.assertIn(mdp.get_action(0), action_list)
def test_add_action(self): """ Test that you can add an action. Named actions make more sense than named states """ mdp = MDP() mdp.add_action(0) self.assertEqual(mdp.num_actions(), 1) self.assertEqual(type(mdp.get_action(0)), Action) self.assertIn(mdp.get_action(0), mdp.get_action_list())
def test_get_num_actions(self): """ It will be helpful to be able to return the number of distinct actions that an MDP has. """ mdp = MDP() mdp.add_action(0) mdp.add_action(1) mdp.add_action(2) mdp.add_action(3) mdp.add_action(4) mdp.add_action(5) mdp.add_action(6) action_list = mdp.get_action_list() self.assertEqual(len(action_list), 7) self.assertIn(mdp.get_action(0), action_list)
def test_add_string_action(self): mdp = MDP() mdp.add_action("jump") self.assertEqual(mdp.num_actions(), 1) self.assertEqual(type(mdp.get_action("jump")), Action) self.assertIn(mdp.get_action("jump"), mdp.get_action_list())