Cosmetic changes

neuroinformatics-unit · Sep 16, 2024 · d96f465 · d96f465
1 parent 6ba7271
commit d96f465
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/test_integration/test_kinematics_vector_transform.py b/tests/test_integration/test_kinematics_vector_transform.py
@@ -15,7 +15,7 @@
     ],
 )
 @pytest.mark.parametrize(
-    "kinematic_variable, expected_2D_pol_array_per_individual",
+    "kinematic_variable, expected_kinematics_polar",
     [
         (
             "displacement",
@@ -48,16 +48,16 @@
         (
             "acceleration",
             [
-                np.zeros((10, 2)),
-                np.zeros((10, 2)),
+                np.zeros((10, 2)),  # Individual 0
+                np.zeros((10, 2)),  # Individual 1
             ],
         ),
     ],
 )
 def test_cart2pol_transform_on_kinematics(
     valid_dataset_uniform_linear_motion,
     kinematic_variable,
-    expected_2D_pol_array_per_individual,
+    expected_kinematics_polar,
     request,
 ):
     """Test transformation between Cartesian and polar coordinates
@@ -69,7 +69,7 @@ def test_cart2pol_transform_on_kinematics(
 
     # Build expected data array
     expected_array_pol = xr.DataArray(
-        np.stack(expected_2D_pol_array_per_individual, axis=1),
+        np.stack(expected_kinematics_polar, axis=1),
         # Stack along the "individuals" axis
         dims=["time", "individuals", "space"],
     )