set delete_many only if count_documents is not 0

langchain-ai · Mar 1, 2024 · d41690d · d41690d
1 parent 2b93206
commit d41690d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/libs/partners/mongodb/tests/integration_tests/test_vectorstores.py b/libs/partners/mongodb/tests/integration_tests/test_vectorstores.py
@@ -50,7 +50,8 @@ class TestMongoDBAtlasVectorSearch:
     def setup_class(cls) -> None:
         # insure the test collection is empty
         collection = get_collection()
-        assert collection.count_documents({}) == 0  # type: ignore[index]  # noqa: E501
+        if collection.count_documents({}):
+            collection.delete_many({})  # type: ignore[index]  # noqa: E501
 
     @classmethod
     def teardown_class(cls) -> None: