JBAujogue commited on
Commit
4145e1a
·
1 Parent(s): cff7f74

add unit tests

Browse files
Files changed (1) hide show
  1. tests/unit/test_action.py +4 -1
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}', but got '{observed}' instead"
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(