Expand tests

emdgroup · Sep 5, 2024 · e8c9658 · e8c9658
1 parent bcb68d3
commit e8c9658
Showing 1 changed file with 32 additions and 2 deletions.
diff --git a/tests/test_pending_experiments.py b/tests/test_pending_experiments.py
@@ -30,14 +30,17 @@
 _flags = dict(
     allow_repeated_recommendations=True,
     allow_recommending_already_measured=True,
-    allow_recommending_pending_experiments=False,
 )
 
 
 @pytest.mark.parametrize(
     "parameter_names, recommender",
     [
-        param(_discrete_params, FPSRecommender(**_flags), id="fps_discrete"),
+        param(
+            _discrete_params,
+            FPSRecommender(**_flags),
+            id="fps_discrete",
+        ),
         param(_discrete_params, PAMClusteringRecommender(**_flags), id="pam_discrete"),
         param(
             _discrete_params,
@@ -64,6 +67,33 @@
             TwoPhaseMetaRecommender(recommender=BotorchRecommender(**_flags)),
             id="botorch_hybrid",
         ),
+        param(
+            _discrete_params,
+            TwoPhaseMetaRecommender(
+                recommender=BotorchRecommender(
+                    **_flags, allow_recommending_pending_experiments=True
+                )
+            ),
+            id="botorch_discrete_allow",
+        ),
+        param(
+            _continuous_params,
+            TwoPhaseMetaRecommender(
+                recommender=BotorchRecommender(
+                    **_flags, allow_recommending_pending_experiments=True
+                )
+            ),
+            id="botorch_continuous_allow",
+        ),
+        param(
+            _hybrid_params,
+            TwoPhaseMetaRecommender(
+                recommender=BotorchRecommender(
+                    **_flags, allow_recommending_pending_experiments=True
+                )
+            ),
+            id="botorch_hybrid_allow",
+        ),
         param(
             _discrete_params,
             NaiveHybridSpaceRecommender(