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(
|