diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 327549979209d02be921236e48af0d5e87668598..1511a9833e8c901811990c5ba2b4d637f0c8f6a5 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -91,6 +91,13 @@ def routes(): Route(name="Route 2", utterances=["Goodbye", "Bye", "Au revoir"]), ] +@pytest.fixture +def dynamic_routes(): + return [ + Route(name="Route 1", utterances=["Hello", "Hi"], function_schema="test"), + Route(name="Route 2", utterances=["Goodbye", "Bye", "Au revoir"]), + ] + class TestRouteLayer: def test_initialization(self, openai_encoder, routes): @@ -106,7 +113,12 @@ class TestRouteLayer: def test_initialization_different_encoders(self, cohere_encoder, openai_encoder): route_layer_cohere = RouteLayer(encoder=cohere_encoder) assert route_layer_cohere.score_threshold == 0.3 + route_layer_openai = RouteLayer(encoder=openai_encoder) + assert route_layer_openai.score_threshold == 0.82 + def test_initialization_dynamic_route(self, cohere_encoder, openai_encoder): + route_layer_cohere = RouteLayer(encoder=cohere_encoder) + assert route_layer_cohere.score_threshold == 0.3 route_layer_openai = RouteLayer(encoder=openai_encoder) assert route_layer_openai.score_threshold == 0.82