From a7f7fca3936f89e2cfc3f7037247450b908a82cc Mon Sep 17 00:00:00 2001
From: James Briggs <james.briggs@hotmail.com>
Date: Sat, 16 Nov 2024 13:19:23 +0100
Subject: [PATCH] fix: sync tests
---
tests/unit/test_sync.py | 13 +++++++------
1 file changed, 7 insertions(+), 6 deletions(-)
diff --git a/tests/unit/test_sync.py b/tests/unit/test_sync.py
index 579e0629..5f4ddc6b 100644
--- a/tests/unit/test_sync.py
+++ b/tests/unit/test_sync.py
@@ -37,14 +37,13 @@ def init_index(
index_cls,
dimensions: Optional[int] = None,
namespace: Optional[str] = "",
- sync: Optional[str] = None,
):
"""We use this function to initialize indexes with different names to avoid
issues during testing.
"""
if index_cls is PineconeIndex:
index = index_cls(
- index_name=TEST_ID, dimensions=dimensions, namespace=namespace, sync=sync
+ index_name=TEST_ID, dimensions=dimensions, namespace=namespace,
)
else:
index = index_cls()
@@ -191,8 +190,11 @@ class TestRouteLayer:
os.environ.get("PINECONE_API_KEY") is None, reason="Pinecone API key required"
)
def test_initialization(self, openai_encoder, routes, index_cls):
- index = init_index(index_cls, sync="local")
- _ = RouteLayer(encoder=openai_encoder, routes=routes, top_k=10, index=index)
+ index = init_index(index_cls)
+ _ = RouteLayer(
+ encoder=openai_encoder, routes=routes, top_k=10, index=index,
+ auto_sync="local",
+ )
@pytest.mark.skipif(
os.environ.get("PINECONE_API_KEY") is None, reason="Pinecone API key required"
@@ -212,7 +214,7 @@ class TestRouteLayer:
def test_second_initialization_not_synced(
self, openai_encoder, routes, routes_2, index_cls
):
- index = init_index(index_cls, sync=None)
+ index = init_index(index_cls)
_ = RouteLayer(
encoder=openai_encoder, routes=routes, index=index, auto_sync="local"
)
@@ -272,7 +274,6 @@ class TestRouteLayer:
)
def test_auto_sync_remote(self, openai_encoder, routes, routes_2, index_cls):
if index_cls is PineconeIndex:
-
# TEST REMOTE
pinecone_index = init_index(index_cls)
_ = RouteLayer(
--
GitLab