From 19d80e5eb338f7007c46a580874de0a16a54ce4f Mon Sep 17 00:00:00 2001 From: Drew Leonard Date: Thu, 6 Jun 2024 15:36:59 +0100 Subject: [PATCH] Benchmark dataset slicing --- dkist/tests/test_benchmarks.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/dkist/tests/test_benchmarks.py b/dkist/tests/test_benchmarks.py index 16072eaf..11dfbfbe 100644 --- a/dkist/tests/test_benchmarks.py +++ b/dkist/tests/test_benchmarks.py @@ -96,3 +96,10 @@ def test_raveled_tab1d_model(benchmark): inputs = tuple(raw_inputs * units) benchmark(raveled_tab, *inputs) + + +@pytest.mark.benchmark +def test_slice_dataset(benchmark, large_visp_dataset): + @benchmark + def slice_dataset(dataset=large_visp_dataset, idx = np.s_[:2, 10:15, 0]): + sliced = dataset[idx]