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())
