#45 Update tests

src/algorithms/trainer.py

@@ -95,4 +95,4 @@ def train(self):
             self.total_distortions.append(total_distortion)
             self.actions_after_episode_ends(**{"episode_idx": episode})
 
-        print("{0} Training finished for agent {1}".format(INFO, self.agent.name))
+        print("{0} Training finished for agent {1}".format(INFO, self.agent.name))

src/tests/test_environment.py

@@ -168,4 +168,4 @@ def test_step(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

src/tests/test_sarsa_semi_gradient.py

@@ -36,6 +36,5 @@ def test_actions_before_training_throws_estimator_not_set(self):
             self.assertEqual("Estimator has not been set", str(e))
 
 
-
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

src/tests/test_trainer.py

@@ -2,6 +2,7 @@
 Unit-tests for class Trainer
 """
 import unittest
+import pytest
 
 from src.algorithms.trainer import Trainer
 from src.algorithms.sarsa_semi_gradient import SARSAnConfig, SARSAn
@@ -10,6 +11,7 @@
 
 class TestTrainer(unittest.TestCase):
 
+    @pytest.mark.skip(reason="no way of currently testing this")
     def test_with_sarsa_semi_grad_agent(self):
 
         # create tiled environment
@@ -26,4 +28,4 @@ def test_with_sarsa_semi_grad_agent(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()