diff --git a/src/openflexure_microscope_server/things/autofocus.py b/src/openflexure_microscope_server/things/autofocus.py index ddfb76bb..43cc3e2b 100644 --- a/src/openflexure_microscope_server/things/autofocus.py +++ b/src/openflexure_microscope_server/things/autofocus.py @@ -283,7 +283,7 @@ class JPEGSharpnessMonitor: self.camera = camera self.stage = stage self.method = method - self.record = method if record is None or record == 0 else record + self.record = record if record else method if not self.method & self.record: raise ValueError( diff --git a/tests/unit_tests/test_autofocus.py b/tests/unit_tests/test_autofocus.py index 0880712b..c568872b 100644 --- a/tests/unit_tests/test_autofocus.py +++ b/tests/unit_tests/test_autofocus.py @@ -129,6 +129,7 @@ def test_record_defaults_to_method(mock_stage, mock_camera): assert monitor.record == SharpnessMethod.FOCUS_FOM + def test_zero_record_defaults_to_method(mock_stage, mock_camera): """If record=0, only the selected method is recorded.""" monitor = JPEGSharpnessMonitor(