Modified estimate cost to have an assert

flexcompute · Sep 17, 2024 · 3d3a076 · 3d3a076
1 parent 16d0077
commit 3d3a076
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/tests/test_plugins/test_design.py b/tests/test_plugins/test_design.py
@@ -495,6 +495,11 @@ def emulated_job_delete(self):
     # Test with dict pre function
     estimate = design_space.estimate_cost(scs_pre_dict)
 
+    if est_cost_func == emulated_estimate_cost_return:
+        assert isinstance(estimate, float)
+    else:
+        assert estimate is None
+
 
 @pytest.mark.parametrize(
     "sweep_method",