Refactor tests

parthvshah · May 19, 2023 · 68d5a14 · 68d5a14
1 parent 4648bbf
commit 68d5a14
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/integration/drivers/storage/vector/test_pinecone_vector_storage_driver.py b/tests/integration/drivers/storage/vector/test_pinecone_vector_storage_driver.py
@@ -10,7 +10,7 @@ class TestOpenAiEmbeddingDriver:
     """
 
     @pytest.fixture(autouse=True)
-    def mock_openai_embedding_create(self, mocker):
+    def mock_pinecone(self, mocker):
         # Create a fake response
         fake_query_response = {
             "matches": [

diff --git a/tests/unit/drivers/embedding/test_openai_embedding_driver.py b/tests/unit/drivers/embedding/test_openai_embedding_driver.py
@@ -4,7 +4,7 @@
 
 class TestOpenAiEmbeddingDriver:
     @pytest.fixture(autouse=True)
-    def mock_openai_embedding_create(self, mocker):
+    def mock_openai(self, mocker):
         fake_response = {
             "data": [
                 {