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

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

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