put tests in right spot

microsoft · Jan 15, 2025 · 3dd77d2 · 3dd77d2
1 parent 6cc63a7
commit 3dd77d2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/...sts/unit/test_mongodb_atlas_collection.py → ...db_atlas/test_mongodb_atlas_collection.py b/...sts/unit/test_mongodb_atlas_collection.py → ...db_atlas/test_mongodb_atlas_collection.py
@@ -1,15 +1,16 @@
 # Copyright (c) Microsoft. All rights reserved.
 
-import pytest
 from unittest.mock import MagicMock, patch
+
+import pytest
 from pymongo import MongoClient
+
 from semantic_kernel.connectors.memory.mongodb_atlas.mongodb_atlas_collection import MongoDBAtlasCollection
-from semantic_kernel.data.record_definition import VectorStoreRecordDefinition
 
 
 @pytest.fixture
 def mock_mongo_client():
-    with patch("pymongo.MongoClient") as mock:
+    with patch("pymongo.AsyncMongoClient") as mock:
         yield mock
 
 

diff --git a/...on/tests/unit/test_mongodb_atlas_store.py → ...mongodb_atlas/test_mongodb_atlas_store.py b/...on/tests/unit/test_mongodb_atlas_store.py → ...mongodb_atlas/test_mongodb_atlas_store.py
@@ -1,16 +1,18 @@
 # Copyright (c) Microsoft. All rights reserved.
 
-import pytest
 from unittest.mock import MagicMock, patch
+
+import pytest
 from pymongo import MongoClient
-from semantic_kernel.connectors.memory.mongodb_atlas.mongodb_atlas_store import MongoDBAtlasStore
+
 from semantic_kernel.connectors.memory.mongodb_atlas.mongodb_atlas_collection import MongoDBAtlasCollection
+from semantic_kernel.connectors.memory.mongodb_atlas.mongodb_atlas_store import MongoDBAtlasStore
 from semantic_kernel.data.record_definition import VectorStoreRecordDefinition
 
 
 @pytest.fixture
 def mock_mongo_client():
-    with patch("pymongo.MongoClient") as mock:
+    with patch("pymongo.AsyncMongoClient") as mock:
         yield mock