From c726eb8ebd6256c2f0744df7f0885f0155c0089d Mon Sep 17 00:00:00 2001 From: Alex Date: Thu, 26 Mar 2026 18:48:53 +0000 Subject: [PATCH] fix: ruff --- tests/vectorstore/test_base.py | 3 +-- tests/vectorstore/test_faiss.py | 4 +--- tests/vectorstore/test_lancedb.py | 2 +- tests/vectorstore/test_mongodb.py | 2 +- tests/vectorstore/test_pgvector.py | 2 +- tests/vectorstore/test_vector_creator.py | 2 +- 6 files changed, 6 insertions(+), 9 deletions(-) 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