Add additional test for corner cases in "bounce"

jojolebarjos · May 11, 2024 · 6372d79 · 6372d79
1 parent 37205cd
commit 6372d79
Showing 1 changed file with 66 additions and 0 deletions.
diff --git a/tests/test_bounce.py b/tests/test_bounce.py
@@ -228,3 +228,69 @@ def test_sequence():
     assert len(state.actions) == 0
     assert state.winner == 0
     assert state.reward.tolist() == [1, -1]
+
+
+def test_larger_values():
+    state, action = assert_state(
+        """
+             [*]
+         1 . . * . *
+         . . * . * .
+         . * . * . *
+         * . * . * .
+         . * . * . *
+         * . * . * .
+         . . . .[7].
+              .
+        """,
+    )
+    state = action.sample_next_state()
+
+    assert state.has_ended
+    assert len(state.actions) == 0
+    assert state.winner == 0
+    assert state.reward.tolist() == [1, -1]
+
+
+def test_block_victory():
+    state, action = assert_state(
+        """
+              .
+         1 . * .[2].
+         . . . * . *
+         2 2 2 2[*]2
+         3 . 3 . 3 .
+         . . . . . .
+         . . . . . .
+         . . . . . .
+              .
+        """,
+    )
+    state = action.sample_next_state()
+
+    assert state.has_ended
+    assert len(state.actions) == 0
+    assert state.winner == 1
+    assert state.reward.tolist() == [-1, 1]
+
+
+def test_draw():
+    state, action = assert_state(
+        """
+              .
+         . . . . . .
+         2 2 2 2 2 2
+         3 3 3 3 3 3
+         3 .[*]. 3 .
+         . * . * . .
+         * . * . * .
+         . .[3]. . *
+              .
+        """,
+    )
+    state = action.sample_next_state()
+
+    assert state.has_ended
+    assert len(state.actions) == 0
+    assert state.winner is None
+    assert state.reward.tolist() == [0, 0]