diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py index 63fe31d7234e756c355a0b367b9c42024d5a22ec..dd4167fa0850b2b8e656bba87f51a93bcbca2c81 100644 --- a/tests/unit/test_layer.py +++ b/tests/unit/test_layer.py @@ -602,6 +602,13 @@ class TestRouteLayer: result.name in ["Route 1", "Route 2"] for result in results ), "Expected the result to be either 'Route 1' or 'Route 2'" + def test_retrieve_without_text_or_vector(self, openai_encoder, routes, index_cls): + route_layer = RouteLayer( + encoder=openai_encoder, routes=routes, index=index_cls() + ) + with pytest.raises(ValueError, match="Either text or vector must be provided"): + route_layer.retrieve_multiple_routes() + def test_retrieve_no_matches(self, openai_encoder, routes, index_cls): route_layer = RouteLayer( encoder=openai_encoder, routes=routes, index=index_cls()