From b2de46aa5b37eda2a4ee37fe4a0e27c7fb7799ed Mon Sep 17 00:00:00 2001 From: b-peri Date: Tue, 24 Sep 2024 17:57:48 +0100 Subject: [PATCH] Added test coverage for `compute_head_direction_vector()` alias --- tests/test_unit/test_kinematics.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/tests/test_unit/test_kinematics.py b/tests/test_unit/test_kinematics.py index 39bcb4c0..01bf3ef6 100644 --- 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(