From 642f0386cf2d017737ad837ce4103c25a8edcb11 Mon Sep 17 00:00:00 2001 From: James Parkhurst Date: Mon, 21 Oct 2024 17:05:32 +0100 Subject: [PATCH] Fix test --- tests/test_sample.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/test_sample.py b/tests/test_sample.py index e4dbd46..0d001d7 100644 --- a/tests/test_sample.py +++ b/tests/test_sample.py @@ -573,6 +573,7 @@ def test_sample_with_motion(tmp_path): position = np.array(position) direction = np.random.uniform(-np.pi, np.pi, size=position.shape[0]) + global_drift = config["motion"]["global_drift"] interaction_range = config["motion"]["interaction_range"] velocity = config["motion"]["velocity"] noise_magnitude = np.radians(config["motion"]["noise_magnitude"]) @@ -582,7 +583,12 @@ def test_sample_with_motion(tmp_path): ): position, direction = ( parakeet.sample.motion.update_particle_position_and_direction( - position, direction, interaction_range, velocity, noise_magnitude + position, + direction, + global_drift, + interaction_range, + velocity, + noise_magnitude, ) ) print(