Move rock paper scissors test in a dedicated test file

airbus · Nov 30, 2023 · 09f5700 · 09f5700
1 parent 1a45047
commit 09f5700
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/skdecide/hub/domain/rock_paper_scissors/rock_paper_scissors.py b/skdecide/hub/domain/rock_paper_scissors/rock_paper_scissors.py
@@ -103,15 +103,3 @@ def _get_observation(
 
     def _get_observation_space_(self) -> D.T_agent[Space[D.T_observation]]:
         return {"player1": EnumSpace(Move), "player2": EnumSpace(Move)}
-
-
-if __name__ == "__main__":
-    from skdecide.utils import rollout
-
-    domain = RockPaperScissors()
-    rollout(
-        domain,
-        action_formatter=lambda a: str({k: v.name for k, v in a.items()}),
-        outcome_formatter=lambda o: f"{ {k: v.name for k, v in o.observation.items()} }"
-        f" - rewards: { {k: v.reward for k, v in o.value.items()} }",
-    )
diff --git a/tests/domains/test_rock_paper_scissors.py b/tests/domains/test_rock_paper_scissors.py
@@ -0,0 +1,14 @@
+from skdecide.hub.domain.rock_paper_scissors.rock_paper_scissors import (
+    RockPaperScissors,
+)
+from skdecide.utils import rollout
+
+
+def test_rock_paper_scissors():
+    domain = RockPaperScissors()
+    rollout(
+        domain,
+        action_formatter=lambda a: str({k: v.name for k, v in a.items()}),
+        outcome_formatter=lambda o: f"{ {k: v.name for k, v in o.observation.items()} }"
+        f" - rewards: { {k: v.reward for k, v in o.value.items()} }",
+    )