From b5f4ca4cda088fd6983f415bb6597db8eb081cfd Mon Sep 17 00:00:00 2001 From: lochhh Date: Mon, 9 Sep 2024 16:12:50 +0100 Subject: [PATCH] Update test function args --- tests/test_unit/test_kinematics.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/tests/test_unit/test_kinematics.py b/tests/test_unit/test_kinematics.py index c2db89c27..54edaa98f 100644 --- a/tests/test_unit/test_kinematics.py +++ b/tests/test_unit/test_kinematics.py @@ -226,9 +226,7 @@ def test_approximate_derivative_with_invalid_order(order): ), ], ) - def test_cdist( - self, dim, pairs, expected_data, pairwise_distances_dataset - ): + def test_cdist(dim, pairs, expected_data, pairwise_distances_dataset): """Test the computation of pairwise distances with known values.""" core_dim = "keypoints" if dim == "individuals" else "individuals" input_dataarray = pairwise_distances_dataset.position @@ -249,7 +247,7 @@ def test_cdist( expected, ) - def expected_pairwise_distances(self, pairs, input_ds, dim): + def expected_pairwise_distances(pairs, input_ds, dim): """Return a list of the expected data variable names for pairwise distances tests. """ @@ -284,7 +282,7 @@ def expected_pairwise_distances(self, pairs, input_ds, dim): ], ) def test_compute_pairwise_distances_with_valid_pairs( - self, pairwise_distances_dataset, dim, pairs + pairwise_distances_dataset, dim, pairs ): """Test that the expected pairwise distances are computed for valid ``pairs`` inputs. @@ -292,7 +290,7 @@ def test_compute_pairwise_distances_with_valid_pairs( result = getattr(kinematics, f"compute_inter{dim[:-1]}_distances")( pairwise_distances_dataset.position, pairs=pairs ) - expected_data_vars = self.expected_pairwise_distances( + expected_data_vars = expected_pairwise_distances( pairs, pairwise_distances_dataset, dim ) if isinstance(result, dict): @@ -301,7 +299,7 @@ def test_compute_pairwise_distances_with_valid_pairs( assert isinstance(result, xr.DataArray) def test_compute_pairwise_distances_with_invalid_dim( - self, pairwise_distances_dataset + pairwise_distances_dataset, ): """Test that an error is raised when an invalid dimension is passed.""" with pytest.raises(ValueError):