fix: add clear index to troublesome tests

aurelio-labs · Jan 10, 2025 · 43c119e · 43c119e
1 parent 19169e7
commit 43c119e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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__)