diff --git a/tests/vectorstore/test_base.py b/tests/vectorstore/test_base.py index fc8eb544..4e8aa637 100644 --- a/tests/vectorstore/test_base.py +++ b/tests/vectorstore/test_base.py @@ -1,5 +1,4 @@ -import os -from unittest.mock import MagicMock, Mock, patch +from unittest.mock import Mock, patch import pytest diff --git a/tests/vectorstore/test_faiss.py b/tests/vectorstore/test_faiss.py index d0413bb1..36ca8542 100644 --- a/tests/vectorstore/test_faiss.py +++ b/tests/vectorstore/test_faiss.py @@ -1,7 +1,5 @@ import io -import os -import tempfile -from unittest.mock import MagicMock, Mock, PropertyMock, call, mock_open, patch +from unittest.mock import Mock, patch import pytest diff --git a/tests/vectorstore/test_lancedb.py b/tests/vectorstore/test_lancedb.py index 49507ea0..b60dca38 100644 --- a/tests/vectorstore/test_lancedb.py +++ b/tests/vectorstore/test_lancedb.py @@ -1,4 +1,4 @@ -from unittest.mock import MagicMock, Mock, PropertyMock, patch +from unittest.mock import MagicMock, Mock, patch import pytest diff --git a/tests/vectorstore/test_mongodb.py b/tests/vectorstore/test_mongodb.py index 7266515a..e93a399f 100644 --- a/tests/vectorstore/test_mongodb.py +++ b/tests/vectorstore/test_mongodb.py @@ -97,7 +97,7 @@ class TestMongoDBVectorStoreAddTexts: inserted_ids=list(range(100)) ) - result = store.add_texts(texts, metadatas) + store.add_texts(texts, metadatas) # Should have been called twice: batch of 100, then batch of 50 assert mock_collection.insert_many.call_count == 2 diff --git a/tests/vectorstore/test_pgvector.py b/tests/vectorstore/test_pgvector.py index bae56c1f..bbec02bf 100644 --- a/tests/vectorstore/test_pgvector.py +++ b/tests/vectorstore/test_pgvector.py @@ -1,4 +1,4 @@ -from unittest.mock import MagicMock, Mock, call, patch +from unittest.mock import MagicMock, Mock, patch import pytest diff --git a/tests/vectorstore/test_vector_creator.py b/tests/vectorstore/test_vector_creator.py index 6fccb86b..8b5c69cd 100644 --- a/tests/vectorstore/test_vector_creator.py +++ b/tests/vectorstore/test_vector_creator.py @@ -1,4 +1,4 @@ -from unittest.mock import patch, MagicMock +from unittest.mock import patch import pytest