test: Actually test subscan analysis result collection [nfc]

OxfordIonTrapGroup · Jan 9, 2021 · b902f09 · b902f09
1 parent 90644a2
commit b902f09
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/test/fixtures.py b/test/fixtures.py
@@ -133,7 +133,11 @@ def build_fragment(self):
         self.setattr_param("b", FloatParam, "b", 0.0)
 
     def get_default_analyses(self):
+        def analyse(coords, values, results):
+            next(iter(results.values())).push(42.0)
+            return []
+
         return [
-            CustomAnalysis([self.a], lambda *_: []),
-            CustomAnalysis([self.b], lambda *_: [])
+            CustomAnalysis([self.a], analyse, [FloatChannel("result_a")]),
+            CustomAnalysis([self.b], analyse, [FloatChannel("result_b")])
         ]
diff --git a/test/test_experiment_subscan.py b/test/test_experiment_subscan.py
@@ -213,6 +213,10 @@ def run_once(self):
 
 
 class SubscanAnalysisCase(ExpFragmentCase):
-    def test_analysis_filtering(self):
+    def test_simple_filtering(self):
         parent = self.create(SubscanAnalysisFragment)
-        run_fragment_once(parent)
+        results = run_fragment_once(parent)
+
+        spec = json.loads(results[parent.scan_spec])
+        self.assertEqual(spec["analysis_results"], {"result_a": "scan_result_a"})
+        self.assertEqual(results[parent.scan_result_a], 42.0)