From d0f30887546372e05d9a405820b0ea1ac905610d Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Wed, 13 Nov 2024 11:06:29 +0100
Subject: [PATCH] fix: sync tests
---
tests/unit/test_sync.py | 33 +++++++++++++++++++--------------
1 file changed, 19 insertions(+), 14 deletions(-)
diff --git a/tests/unit/test_sync.py b/tests/unit/test_sync.py
index 3cb425bb..b7acebe9 100644
--- a/tests/unit/test_sync.py
+++ b/tests/unit/test_sync.py
@@ -12,7 +12,7 @@ from semantic_router.route import Route
from platform import python_version
-PINECONE_SLEEP = 12
+PINECONE_SLEEP = 6
def mock_encoder_call(utterances):
@@ -199,7 +199,7 @@ class TestRouteLayer:
def test_second_initialization_sync(self, openai_encoder, routes, index_cls):
index = init_index(index_cls)
route_layer = RouteLayer(
- encoder=openai_encoder, routes=routes, top_k=10, index=index
+ encoder=openai_encoder, routes=routes, index=index, auto_sync="local"
)
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be populated
@@ -212,7 +212,10 @@ class TestRouteLayer:
self, openai_encoder, routes, routes_2, index_cls
):
index = init_index(index_cls, sync=None)
- _ = RouteLayer(encoder=openai_encoder, routes=routes, index=index)
+ _ = RouteLayer(
+ encoder=openai_encoder, routes=routes, index=index,
+ auto_sync="local"
+ )
route_layer = RouteLayer(encoder=openai_encoder, routes=routes_2, index=index)
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be populated
@@ -223,21 +226,23 @@ class TestRouteLayer:
)
def test_utterance_diff(self, openai_encoder, routes, routes_2, index_cls):
index = init_index(index_cls)
- _ = RouteLayer(encoder=openai_encoder, routes=routes, top_k=10, index=index)
- if index_cls is PineconeIndex:
- time.sleep(PINECONE_SLEEP) # allow for index to be populated
+ _ = RouteLayer(
+ encoder=openai_encoder, routes=routes, index=index,
+ auto_sync="local"
+ )
route_layer_2 = RouteLayer(
- encoder=openai_encoder, routes=routes_2, top_k=10, index=index
+ encoder=openai_encoder, routes=routes_2, index=index
)
if index_cls is PineconeIndex:
time.sleep(PINECONE_SLEEP) # allow for index to be populated
- diff = route_layer_2.get_utterance_diff()
- assert " Route 1: Hello" in diff
- assert "+ Route 1: Hi" in diff
- assert "+ Route 2: Au revoir" in diff
- assert "+ Route 2: Bye" in diff
- assert "+ Route 2: Goodbye" in diff
- assert " Route 2: Hi" in diff
+ diff = route_layer_2.get_utterance_diff(include_metadata=True)
+ assert "+ Route 1: Hello | None | {'type': 'default'}" in diff
+ assert "+ Route 1: Hi | None | {'type': 'default'}" in diff
+ assert "- Route 1: Hello | None | {}" in diff
+ assert "+ Route 2: Au revoir | None | {}" in diff
+ assert "- Route 2: Hi | None | {}" in diff
+ assert "+ Route 2: Bye | None | {}" in diff
+ assert "+ Route 2: Goodbye | None | {}" in diff
@pytest.mark.skipif(
os.environ.get("PINECONE_API_KEY") is None, reason="Pinecone API key required"
--
GitLab