even more tests

Author: Thomas Morris

.codecov.yml

@@ -7,4 +7,4 @@ coverage:
         target: auto
     patch:
       default:
-        target: auto
+        target: 80

src/blop/generation/__init__.py

@@ -1,14 +1,15 @@
-from ax.generation_strategy.generation_strategy import GenerationStrategy
-
-from .nodes import latent_gp_node, sobol_node
+from .beamline import beamline_generation_strategy
 
 all_strategies = ["beamline"]
 
 
+class InvalidStrategyError(Exception): ...
+
+
 def get_generation_strategy(name):
 
     if name not in all_strategies:
-        raise ValueError(f"Invalid strategy '{name}', valid strategies are one of {all_strategies}")
+        raise InvalidStrategyError(f"Invalid strategy '{name}', valid strategies are one of {all_strategies}")
 
     if name == "beamline":
-        return GenerationStrategy(name="Custom Generation Strategy", nodes=[sobol_node, latent_gp_node])
+        return beamline_generation_strategy

src/blop/tests/ax/test_generation.py …blop/tests/generation/test_generation.pysrc/blop/tests/ax/test_generation.py renamed to src/blop/tests/generation/test_generation.py src/blop/tests/ax/test_generation.py renamed to src/blop/tests/generation/test_generation.py

@@ -11,7 +11,7 @@
 from blop.ax.dof import RangeDOF
 from blop.ax.objective import Objective
 from blop.evaluation.test_functions import TestFunctionEvaluation
-from blop.generation import all_strategies, get_generation_strategy
+from blop.generation import InvalidStrategyError, all_strategies, get_generation_strategy
 
 from ..conftest import MovableSignal
 from ..test_plans import _collect_optimize_events
@@ -27,6 +27,15 @@ def test_get_generation_strategy(name):
     get_generation_strategy(name)
 
 
+def test_invalid_strategy():
+    try:
+        get_generation_strategy("invalid_strategy")
+    except InvalidStrategyError:
+        pass
+    except Exception as error:
+        raise error
+
+
 def test_optimize(RE):
 
     dofs = [