admin/dify
1
0
Fork 0
Code Issues Pull Requests Actions 1 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
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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