From c5d6d0220b0891fcb0f2a38be1325254fdf4804d Mon Sep 17 00:00:00 2001
From: Siraj R Aizlewood <siraj@aurelio.ai>
Date: Wed, 3 Apr 2024 03:05:56 +0400
Subject: [PATCH] More test coverage.
---
tests/unit/test_layer.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/tests/unit/test_layer.py b/tests/unit/test_layer.py
index 63fe31d7..dd4167fa 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()
--
GitLab