Spaces:
Sleeping
Sleeping
add unit tests
Browse files
tests/unit/test_action.py
CHANGED
@@ -10,9 +10,12 @@ from rubik.action import (
|
|
10 |
|
11 |
|
12 |
def test_position_rotation_shape():
|
|
|
|
|
|
|
13 |
expected = (3, 4, 4)
|
14 |
observed = POS_ROTATIONS.shape
|
15 |
-
assert expected == observed, f"Position rotation tensor expected shape '{expected}',
|
16 |
|
17 |
|
18 |
@pytest.mark.parametrize(
|
|
|
10 |
|
11 |
|
12 |
def test_position_rotation_shape():
|
13 |
+
"""
|
14 |
+
Test that POS_ROTATIONS has expected shape.
|
15 |
+
"""
|
16 |
expected = (3, 4, 4)
|
17 |
observed = POS_ROTATIONS.shape
|
18 |
+
assert expected == observed, f"Position rotation tensor expected shape '{expected}', got '{observed}' instead"
|
19 |
|
20 |
|
21 |
@pytest.mark.parametrize(
|