diff --git a/api/tests/integration_tests/vdb/milvus/test_milvus.py b/api/tests/integration_tests/vdb/milvus/test_milvus.py index 00c414000..2ce85445f 100644 --- a/api/tests/integration_tests/vdb/milvus/test_milvus.py +++ b/api/tests/integration_tests/vdb/milvus/test_milvus.py @@ -1,12 +1,12 @@ from core.rag.datasource.vdb.milvus.milvus_vector import MilvusConfig, MilvusVector from tests.integration_tests.vdb.test_vector_store import ( - AbstractTestVector, + AbstractVectorTest, get_example_text, setup_mock_redis, ) -class TestMilvusVector(AbstractTestVector): +class MilvusVectorTest(AbstractVectorTest): def __init__(self): super().__init__() self.vector = MilvusVector( @@ -33,4 +33,4 @@ class TestMilvusVector(AbstractTestVector): def test_milvus_vector(setup_mock_redis): - TestMilvusVector().run_all_tests() + MilvusVectorTest().run_all_tests() diff --git a/api/tests/integration_tests/vdb/qdrant/test_qdrant.py b/api/tests/integration_tests/vdb/qdrant/test_qdrant.py index ba6920660..34beb25d4 100644 --- a/api/tests/integration_tests/vdb/qdrant/test_qdrant.py +++ b/api/tests/integration_tests/vdb/qdrant/test_qdrant.py @@ -1,11 +1,11 @@ from core.rag.datasource.vdb.qdrant.qdrant_vector import QdrantConfig, QdrantVector from tests.integration_tests.vdb.test_vector_store import ( - AbstractTestVector, + AbstractVectorTest, setup_mock_redis, ) -class TestQdrantVector(AbstractTestVector): +class QdrantVectorTest(AbstractVectorTest): def __init__(self): super().__init__() self.attributes = ['doc_id', 'dataset_id', 'document_id', 'doc_hash'] @@ -20,4 +20,4 @@ class TestQdrantVector(AbstractTestVector): def test_qdrant_vector(setup_mock_redis): - TestQdrantVector().run_all_tests() + QdrantVectorTest().run_all_tests() diff --git a/api/tests/integration_tests/vdb/test_vector_store.py b/api/tests/integration_tests/vdb/test_vector_store.py index fd64d445e..f7698c903 100644 --- a/api/tests/integration_tests/vdb/test_vector_store.py +++ b/api/tests/integration_tests/vdb/test_vector_store.py @@ -41,7 +41,7 @@ def setup_mock_redis() -> None: ext_redis.redis_client.lock = mock_redis_lock -class AbstractTestVector: +class AbstractVectorTest: def __init__(self): self.vector = None self.dataset_id = str(uuid.uuid4()) diff --git a/api/tests/integration_tests/vdb/weaviate/test_weaviate.py b/api/tests/integration_tests/vdb/weaviate/test_weaviate.py index 20a2d2be0..3d540cee3 100644 --- a/api/tests/integration_tests/vdb/weaviate/test_weaviate.py +++ b/api/tests/integration_tests/vdb/weaviate/test_weaviate.py @@ -1,11 +1,11 @@ from core.rag.datasource.vdb.weaviate.weaviate_vector import WeaviateConfig, WeaviateVector from tests.integration_tests.vdb.test_vector_store import ( - AbstractTestVector, + AbstractVectorTest, setup_mock_redis, ) -class TestWeaviateVector(AbstractTestVector): +class WeaviateVectorTest(AbstractVectorTest): def __init__(self): super().__init__() self.attributes = ['doc_id', 'dataset_id', 'document_id', 'doc_hash'] @@ -20,4 +20,4 @@ class TestWeaviateVector(AbstractTestVector): def test_weaviate_vector(setup_mock_redis): - TestWeaviateVector().run_all_tests() + WeaviateVectorTest().run_all_tests()