Merge pull request #123 from ISISComputingGroup/ibc93_standard_callbacks

add system test for standard callbacks

Author: jackbdoughty

requirements.txt

@@ -9,4 +9,5 @@ six
 unittest-xml-reporting
 ibex_bluesky_core
 genie_python
-configobj
+matplotlib
+pyqt6

test_bluesky.py

@@ -2,13 +2,17 @@
 import os
 import unittest
 import uuid
+from pathlib import Path
 
 import bluesky.plan_stubs as bps
 import bluesky.plans as bp
+import matplotlib
 from bluesky.callbacks import LiveTable
 from bluesky.preprocessors import subs_decorator
 from bluesky.run_engine import RunEngine, RunEngineResult
 from genie_python import genie as g  # type: ignore
+from ibex_bluesky_core.callbacks import ISISCallbacks
+from ibex_bluesky_core.callbacks.fitting.fitting_utils import Linear
 from ibex_bluesky_core.devices import get_pv_prefix
 from ibex_bluesky_core.devices.block import block_r, block_rw_rbv
 from ibex_bluesky_core.devices.simpledae import SimpleDae
@@ -20,7 +24,10 @@
     GoodFramesNormalizer,
     PeriodGoodFramesNormalizer,
 )
-from ibex_bluesky_core.devices.simpledae.waiters import GoodFramesWaiter, PeriodGoodFramesWaiter
+from ibex_bluesky_core.devices.simpledae.waiters import (
+    GoodFramesWaiter,
+    PeriodGoodFramesWaiter,
+)
 from ibex_bluesky_core.log import set_bluesky_log_levels
 from ibex_bluesky_core.run_engine import get_run_engine
 from ophyd_async.plan_stubs import ensure_connected
@@ -30,6 +37,7 @@
     set_genie_python_raises_exceptions,
 )
 
+matplotlib.use("qtagg")
 RE: RunEngine = get_run_engine()
 
 P3_INIT_VALUE: float = 123.456
@@ -147,6 +155,29 @@ def _plan():
         self.assertTrue(any("|    123.456 |  -10.00000 |" in line for line in livetable_lines))
         self.assertTrue(any("|    123.456 |   10.00000 |" in line for line in livetable_lines))
 
+    def test_scan_with_standard_callbacks(self) -> None:
+        icc = ISISCallbacks(
+            x="p5",
+            y="p3",
+            fit=Linear().fit(),
+            human_readable_file_output_dir=Path(LOG_FOLDER) / "output_files",
+            live_fit_logger_output_dir=Path(LOG_FOLDER) / "fitting",
+        )
+
+        @icc
+        def _plan():
+            p3 = block_r(float, "p3")
+            p5 = block_rw_rbv(float, "p5")
+            yield from ensure_connected(p3, p5)
+            yield from bp.scan([p3], p5, -10, 10, num=10)
+
+        RE(_plan())
+
+        self.assertAlmostEqual(icc.peak_stats["com"], 0)
+        print(icc.live_fit.result.params["c0"])
+        print(icc.live_fit.result.fit_report())
+        self.assertAlmostEqual(icc.live_fit.result.params["c0"], P3_INIT_VALUE)
+
     def test_count_simple_dae(self) -> None:
         start_run_number = int(g.get_runnumber())
 
@@ -211,7 +242,9 @@ def test_GIVEN_logging_is_requested_THEN_the_log_file_exists(self) -> None:
         )
         self.assertTrue(os.path.exists(self.qualified_log_filename))
 
-    def test_GIVEN_logging_is_requested_THEN_the_log_file_contains_the_message(self) -> None:
+    def test_GIVEN_logging_is_requested_THEN_the_log_file_contains_the_message(
+        self,
+    ) -> None:
         # Log invocation.
         bluesky_message = LOG_MESSAGE + str(uuid.uuid4())
         other_message = LOG_MESSAGE + str(uuid.uuid4())
@@ -240,7 +273,9 @@ def test_GIVEN_logging_is_configured_at_info_level_THEN_debug_messages_do_not_go
             self.assertNotIn(debug_msg, content)
             self.assertIn(info_msg, content)
 
-    def test_GIVEN_logging_is_configured_at_debug_level_THEN_debug_messages_do_go_to_log_file(self):
+    def test_GIVEN_logging_is_configured_at_debug_level_THEN_debug_messages_do_go_to_log_file(
+        self,
+    ):
         debug_msg = LOG_MESSAGE + str(uuid.uuid4())
         info_msg = LOG_MESSAGE + str(uuid.uuid4())