From 04d2a87c89f4affa3de239844a00157f8b313d86 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sun, 15 Dec 2024 14:27:41 +0400
Subject: [PATCH] fix: better test coverage
---
tests/unit/test_router.py | 17 +++++++++++++++++
1 file changed, 17 insertions(+)
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index be10e3b6..067e78a9 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -272,6 +272,23 @@ class TestSemanticRouter:
assert route_layer.score_threshold == openai_encoder.score_threshold
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be updated
+ _ = route_layer("Hello")
+ assert len(route_layer.index.get_utterances()) == 6
+
+ def test_init_and_add_single_utterance(
+ self, route_single_utterance, openai_encoder, index_cls
+ ):
+ index = init_index(index_cls)
+ route_layer = SemanticRouter(
+ encoder=openai_encoder,
+ index=index,
+ auto_sync="local",
+ )
+ if index_cls is PineconeIndex:
+ time.sleep(PINECONE_SLEEP) # allow for index to be updated
+ route_layer.add(routes=[route_single_utterance])
+ assert route_layer.score_threshold == openai_encoder.score_threshold
+ _ = route_layer("Hello")
assert len(route_layer.index.get_utterances()) == 1
def test_delete_index(self, openai_encoder, routes, index_cls):
--
GitLab