test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-01 10:48:37 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

chore: skip warning messages when pytest auto-collecting the vdb test class by removing Test prefix (#3906)

Browse Source
This commit is contained in:
Bowen Liang 2024-04-27 16:36:09 +08:00 committed by GitHub
parent 43a5ba9415
commit 661b30784e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/tests/integration_tests/vdb/milvus/test_milvus.py
View File

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

6
api/tests/integration_tests/vdb/qdrant/test_qdrant.py
View File

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

2
api/tests/integration_tests/vdb/test_vector_store.py
View File

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

6
api/tests/integration_tests/vdb/weaviate/test_weaviate.py
View File

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