diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 286e7ace5967480bf23ba7f5c6945086c3b59d72..92577f34da95006c8772c46b39dcafeaff40b5cc 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -119,6 +119,10 @@ class TestRouteLayer: route_layer_openai = RouteLayer(encoder=openai_encoder) assert route_layer_openai.score_threshold == 0.82 + def test_initialization_no_encoder(self, openai_encoder): + route_layer_none = RouteLayer(encoder=None) + assert route_layer_none.score_threshold == openai_encoder.score_threshold + 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