Added test coverage for compute_head_direction_vector() alias

neuroinformatics-unit · Sep 24, 2024 · b2de46a · b2de46a
1 parent be75092
commit b2de46a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/test_unit/test_kinematics.py b/tests/test_unit/test_kinematics.py
@@ -271,6 +271,12 @@ def test_compute_forward_vector(valid_data_array_for_forward_vector):
         "right_ear",
         camera_view="top_down",
     )
+    head_vector = kinematics.compute_head_direction_vector(
+        valid_data_array_for_forward_vector,
+        "left_ear",
+        "right_ear",
+        camera_view="bottom_up",
+    )
     known_vectors = np.array([[[0, -1]], [[1, 0]], [[0, 1]], [[-1, 0]]])
 
     assert (
@@ -280,6 +286,7 @@ def test_compute_forward_vector(valid_data_array_for_forward_vector):
     )
     assert np.equal(forward_vector.values, known_vectors).all()
     assert np.equal(forward_vector_flipped.values, known_vectors * -1).all()
+    assert head_vector.equals(forward_vector)
 
 
 @pytest.mark.parametrize(