From 2a5080362ec8e4d18057468c430b020dbeafeff4 Mon Sep 17 00:00:00 2001 From: Alec Koumjian Date: Thu, 5 Dec 2024 21:33:53 -0500 Subject: [PATCH] Benchmark multiple size inputs to transform coordinates --- .../coordinates/tests/test_benchmarks.py | 19 ++++++++++--------- 1 file changed, 10 insertions(+), 9 deletions(-) diff --git a/src/adam_core/coordinates/tests/test_benchmarks.py b/src/adam_core/coordinates/tests/test_benchmarks.py index e44188d5..ea2f893a 100644 --- a/src/adam_core/coordinates/tests/test_benchmarks.py +++ b/src/adam_core/coordinates/tests/test_benchmarks.py @@ -24,9 +24,10 @@ [OriginCodes.SUN, OriginCodes.SOLAR_SYSTEM_BARYCENTER], ids=lambda x: f"origin={x.name},", ) +@pytest.mark.parametrize("size", [1, 50, 100]) @pytest.mark.benchmark(group="coord_transforms") def test_benchmark_transform_cartesian_coordinates( - benchmark, representation, frame, origin + benchmark, representation, frame, origin, size ): if origin == OriginCodes.SOLAR_SYSTEM_BARYCENTER: pytest.skip("barycenter transform not yet implemented") @@ -37,14 +38,14 @@ def test_benchmark_transform_cartesian_coordinates( frame_in = "ecliptic" from_coords = CartesianCoordinates.from_kwargs( - x=np.array([1]), - y=np.array([1]), - z=np.array([1]), - vx=np.array([1]), - vy=np.array([1]), - vz=np.array([1]), - time=Timestamp.from_mjd([50000]), - origin=Origin.from_kwargs(code=["SUN"]), + x=np.array([1] * size), + y=np.array([1] * size), + z=np.array([1] * size), + vx=np.array([1] * size), + vy=np.array([1] * size), + vz=np.array([1] * size), + time=Timestamp.from_mjd([50000] * size), + origin=Origin.from_kwargs(code=[OriginCodes.SUN] * size), frame=frame_in, ) benchmark(