From 43c119e1dae9cc1569f61995772de698c879b6f3 Mon Sep 17 00:00:00 2001 From: James Briggs <35938317+jamescalam@users.noreply.github.com> Date: Fri, 10 Jan 2025 12:49:45 +0000 Subject: [PATCH] fix: add clear index to troublesome tests --- tests/unit/test_router.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py index d720e1f7..abe0b2e7 100644 --- a/tests/unit/test_router.py +++ b/tests/unit/test_router.py @@ -764,6 +764,14 @@ def check_index_populated(): check_index_populated() + # clear index + @retry(max_retries=RETRY_COUNT, delay=PINECONE_SLEEP) + def clear_index(): + route_layer.index.index.delete(delete_all=True) + assert route_layer.index.get_utterances() == [] + + clear_index() + def test_query_and_classification(self, routes, index_cls, encoder_cls, router_cls): encoder = encoder_cls() index = init_index(index_cls, index_name=encoder.__class__.__name__)