test single-slice case

mehta-lab · Mar 18, 2024 · 797654f · 797654f
1 parent 8e39e41
commit 797654f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_focus_estimator.py b/tests/test_focus_estimator.py
@@ -45,6 +45,15 @@ def test_focus_estimator(tmp_path):
     assert slice <= data3D.shape[0]
     assert plot_path.exists()
 
+    # Check single slice
+    slice = focus.focus_from_transverse_band(
+        np.random.random((1, 10, 10)),
+        NA_det,
+        lambda_ill,
+        ps,
+    )
+    assert slice == 0
+
 
 def test_focus_estimator_snr(tmp_path):
     ps = 6.5 / 100