Benchmark dataset slicing

DKISTDC · Jun 6, 2024 · 19d80e5 · 19d80e5
1 parent 78eb098
commit 19d80e5
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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]