From 8f282aa11156e7c0d666c807ac6bf5e13993c1c7 Mon Sep 17 00:00:00 2001 From: Stella <30465823+stellaprins@users.noreply.github.com> Date: Wed, 5 Feb 2025 09:04:46 +0000 Subject: [PATCH] change trajectory inputs, replace keypoint and individual with selection, remove title --- tests/test_unit/test_plot.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/tests/test_unit/test_plot.py b/tests/test_unit/test_plot.py index 76e97924a..2608831d2 100644 --- a/tests/test_unit/test_plot.py +++ b/tests/test_unit/test_plot.py @@ -82,10 +82,11 @@ def test_trajectory(sample_data, image, tmp_path): kwargs = {"image_path": image_path} else: kwargs = {"image_path": None} - da_position = sample_data.position - _, ax_centre = trajectory(da_position, keypoint="centre", **kwargs) + _, ax_centre = trajectory( + sample_data.position, keypoint="centre", **kwargs + ) _, ax_left_right = trajectory( - da_position, keypoint=["left", "right"], **kwargs + sample_data.position, keypoint=["left", "right"], **kwargs ) expected_data = np.array([[0, 0], [0, 1], [0, 2], [0, 3]], dtype=float)