Achieve 100% test coverage for agent_base class (#106)

test: Added test for agent base

tests/units/agents/test_agent_base.py

@@ -80,3 +80,16 @@ def test_learn(self):
 
         # THEN
         assert learn_return is None
+
+    def test_learn_with_previous_state(self):
+        # GIVEN
+        fake_model = FakeModel()
+        fake_state_space = FakeState()
+        fake_agent = FakeAgent(None, fake_state_space, fake_model, None)
+        fake_agent.previous_state = "previous_state"
+
+        # WHEN
+        learn_return = fake_agent.learn("obs", "reward", "done")
+
+        # THEN
+        assert learn_return is None

tests/units/models/test_model_base.py

@@ -15,7 +15,7 @@ def __init__(self):
 
 		fake_model = FakeModel()
 		learn_return = fake_model.learn("current_state",
-			"action", "reward", "next_state")
+			"action", "reward", "next_state", "done")
 		predict_return = fake_model.predict("state")
 		learning_data = fake_model.learning_data
 

urnai/models/model_base.py

@@ -10,7 +10,7 @@ def __init__(self):
 		self.learning_data = {}
 
 	@abstractmethod
-	def learn(self, current_state, action, reward, next_state):
+	def learn(self, current_state, action, reward, next_state, done):
 		"""Learning strategy"""
 		...