test_experiment_parameters: test evaluation of default parameters

OxfordIonTrapGroup · May 11, 2023 · 14e1f4f · 14e1f4f
1 parent bcb1c79
commit 14e1f4f
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/test/test_experiment_parameters.py b/test/test_experiment_parameters.py
@@ -21,6 +21,16 @@ def test_describe(self):
                 "type": "float"
             })
 
+    def test_evaluate_default(self):
+        def mock_get_dataset(key: str, default=None):
+            return {"baz": 42.}[key]
+
+        param = FloatParam("foo", "bar", 0.)
+        self.assertEqual(param.eval_default(mock_get_dataset), 0.)
+
+        param = FloatParam("foo", "bar", "dataset('baz', 0.)")
+        self.assertEqual(param.eval_default(mock_get_dataset), 42.)
+
 
 class IntParamCase(unittest.TestCase):
     def test_describe(self):
@@ -40,6 +50,16 @@ def test_describe(self):
                 "type": "int"
             })
 
+    def test_evaluate_default(self):
+        def mock_get_dataset(key: str, default=None):
+            return {"baz": 42}[key]
+
+        param = IntParam("foo", "bar", 0)
+        self.assertEqual(param.eval_default(mock_get_dataset), 0)
+
+        param = IntParam("foo", "bar", "dataset('baz', 0)")
+        self.assertEqual(param.eval_default(mock_get_dataset), 42)
+
 
 class BoolParamCase(unittest.TestCase):
     def test_describe(self):
@@ -54,3 +74,13 @@ def test_describe(self):
                     "is_scannable": True
                 }
             })
+
+    def test_evaluate_default(self):
+        def mock_get_dataset(key: str, default=None):
+            return {"baz": True}[key]
+
+        param = BoolParam("foo", "bar", True)
+        self.assertEqual(param.eval_default(mock_get_dataset), True)
+
+        param = BoolParam("foo", "bar", "dataset('baz', False)")
+        self.assertEqual(param.eval_default(mock_get_dataset), True)