add unit tests

tests/unit/test_action.py

@@ -8,6 +8,10 @@ from rubik.action import (
     POS_SHIFTS,
     FACE_ROTATIONS,
     build_actions_tensor,
+    build_action_tensor,
+    parse_action_str,
+    parse_actions_str,
+    sample_actions_str,
 )
 
 
@@ -97,3 +101,79 @@ def test_build_actions_tensor_shape(size: int):
     assert expected == observed, (
         f"'build_actions_tensor' output has incorrect shape: expected shape '{expected}', got '{observed}' instead"
     )
+
+
+@pytest.mark.parametrize(
+    "size, axis, slice, inverse",
+    [
+        (2, 2, 1, 0),
+        (3, 0, 1, 1),
+        (5, 1, 4, 0),
+    ],
+)
+def test_build_action_tensor_shape(size: int, axis: int, slice: int, inverse: int):
+    """
+    Test that "build_actions_tensor" output has expected shape.
+    """
+    expected = (3, size, 2, 6 * (size**2), 6 * (size**2))
+    observed = build_action_tensor(size, axis, slice, inverse).shape
+    assert expected == observed, (
+        f"'build_action_tensor' output has incorrect shape: expected shape '{expected}', got '{observed}' instead"
+    )
+
+
+@pytest.mark.parametrize(
+    "move, expected",
+    [
+        ["X1", (0, 1, 0)],
+        ["X25i", (0, 25, 1)],
+        ["Y0", (1, 0, 0)],
+        ["Y5i", (1, 5, 1)],
+        ["Z30", (2, 30, 0)],
+        ["Z512ijk", (2, 512, 1)],
+    ],
+)
+def test_parse_action_str(move: str, expected: tuple[int, int, int]):
+    """
+    Test that "parse_action_str" behaves as expected.
+    """
+    observed = parse_action_str(move)
+    assert expected == observed, (
+        f"'parse_action_str' output is incorrect: expected '{expected}', got '{observed}' instead"
+    )
+
+
+@pytest.mark.parametrize(
+    "moves, expected",
+    [
+        ["  X1 Y0 X25i Z512ijk Z30 Y5i ", [(0, 1, 0), (1, 0, 0), (0, 25, 1), (2, 512, 1), (2, 30, 0), (1, 5, 1)]],
+    ],
+)
+def test_parse_actions_str(moves: str, expected: tuple[int, int, int]):
+    """
+    Test that "parse_action_str" behaves as expected.
+    """
+    observed = parse_actions_str(moves)
+    assert expected == observed, (
+        f"'parse_actions_str' output is incorrect: expected '{expected}', got '{observed}' instead"
+    )
+
+
+@pytest.mark.parametrize(
+    "num_moves, size, seed",
+    [
+        [1, 3, 0],
+        [1, 20, 42],
+        [256, 5, 21],
+    ],
+)
+def test_sample_actions_str(num_moves: int, size: int, seed: int):
+    """
+    Test that "sample_actions_str" is deterministic and outputs parsable content.
+    """
+    moves_1 = sample_actions_str(num_moves, size, seed)
+    moves_2 = sample_actions_str(num_moves, size, seed)
+    assert moves_1 == moves_2, f"'sample_actions_str' is non-deterministic: {moves_1} != {moves_2}"
+
+    parsed = parse_actions_str(moves_1)
+    assert len(parsed) == len(moves_1.split()), "'sample_actions_str' output cannot be parsed correctly"