From 3f3a025e0f867b8f137d50cc42e5e14e79b90ebe Mon Sep 17 00:00:00 2001 From: Alex Date: Tue, 15 Feb 2022 11:47:24 +0000 Subject: [PATCH] #45 Update tests --- src/algorithms/trainer.py | 2 +- src/tests/test_environment.py | 2 +- src/tests/test_sarsa_semi_gradient.py | 3 +-- src/tests/test_trainer.py | 4 +++- 4 files changed, 6 insertions(+), 5 deletions(-) diff --git a/src/algorithms/trainer.py b/src/algorithms/trainer.py index 4d8de98..f92422b 100644 --- a/src/algorithms/trainer.py +++ b/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)) \ No newline at end of file + print("{0} Training finished for agent {1}".format(INFO, self.agent.name)) diff --git a/src/tests/test_environment.py b/src/tests/test_environment.py index 609a441..cfbd63e 100644 --- a/src/tests/test_environment.py +++ b/src/tests/test_environment.py @@ -168,4 +168,4 @@ def test_step(self): if __name__ == '__main__': - unittest.main() \ No newline at end of file + unittest.main() diff --git a/src/tests/test_sarsa_semi_gradient.py b/src/tests/test_sarsa_semi_gradient.py index d2a2125..1944e4e 100644 --- a/src/tests/test_sarsa_semi_gradient.py +++ b/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() \ No newline at end of file + unittest.main() diff --git a/src/tests/test_trainer.py b/src/tests/test_trainer.py index 5d885d7..c2175e3 100644 --- a/src/tests/test_trainer.py +++ b/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() \ No newline at end of file + unittest.main()