From 7a2fffcada48aea1da03dacd337f8b1152c6046d Mon Sep 17 00:00:00 2001 From: "Matthew R. Becker" Date: Wed, 29 Jan 2025 13:48:54 -0600 Subject: [PATCH] test: add benchmark for moffat init (#139) --- tests/jax/test_benchmarks.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/jax/test_benchmarks.py b/tests/jax/test_benchmarks.py index f5eb0802..f73d569a 100644 --- a/tests/jax/test_benchmarks.py +++ b/tests/jax/test_benchmarks.py @@ -289,3 +289,15 @@ def test_benchmark_invert_ab_noraise(benchmark, kind): lambda: _run_benchmark_invert_ab_noraise(u, v, ab).block_until_ready(), ) print(f"time: {dt:0.4g} ms", end=" ") + + +def _run_benchmark_moffat_init(): + return jgs.Moffat(beta=2.5, half_light_radius=0.6, trunc=1.2).scale_radius + + +@pytest.mark.parametrize("kind", ["compile", "run"]) +def test_benchmark_moffat_init(benchmark, kind): + dt = _run_benchmarks( + benchmark, kind, lambda: _run_benchmark_moffat_init().block_until_ready() + ) + print(f"time: {dt:0.4g} ms", end=" ")