Skip to content
Snippets Groups Projects
Unverified Commit 9f16810c authored by James Briggs's avatar James Briggs
Browse files

add delete and increase sleep

parent c4dd453a
No related branches found
No related tags found
No related merge requests found
...@@ -280,7 +280,7 @@ class TestRouteLayer: ...@@ -280,7 +280,7 @@ class TestRouteLayer:
route_layer = RouteLayer( route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, index=pineconeindex encoder=openai_encoder, routes=routes, index=pineconeindex
) )
time.sleep(5) # allow for index to be populated time.sleep(10) # allow for index to be populated
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
try: try:
...@@ -288,6 +288,9 @@ class TestRouteLayer: ...@@ -288,6 +288,9 @@ class TestRouteLayer:
except ValueError: except ValueError:
assert True assert True
# delete index
pineconeindex.delete_index()
assert query_result in ["Route 1"] assert query_result in ["Route 1"]
def test_namespace_pinecone_index(self, openai_encoder, routes, index_cls): def test_namespace_pinecone_index(self, openai_encoder, routes, index_cls):
...@@ -296,7 +299,7 @@ class TestRouteLayer: ...@@ -296,7 +299,7 @@ class TestRouteLayer:
route_layer = RouteLayer( route_layer = RouteLayer(
encoder=openai_encoder, routes=routes, index=pineconeindex encoder=openai_encoder, routes=routes, index=pineconeindex
) )
time.sleep(5) # allow for index to be populated time.sleep(10) # allow for index to be populated
query_result = route_layer(text="Hello", route_filter=["Route 1"]).name query_result = route_layer(text="Hello", route_filter=["Route 1"]).name
try: try:
...@@ -304,6 +307,9 @@ class TestRouteLayer: ...@@ -304,6 +307,9 @@ class TestRouteLayer:
except ValueError: except ValueError:
assert True assert True
# delete index
pineconeindex.delete_index()
assert query_result in ["Route 1"] assert query_result in ["Route 1"]
def test_query_with_no_index(self, openai_encoder, index_cls): def test_query_with_no_index(self, openai_encoder, index_cls):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment