| @@ -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() | |||
| @@ -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() | |||
| @@ -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()) | |||
| @@ -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() | |||