chore: rename vdb tests for PGVector and PGvectoRS (#4973)

This commit is contained in:
Bowen Liang 2024-06-06 07:22:49 +08:00 committed by GitHub
parent 3579fd1b09
commit 7fa735a43b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 4 deletions

View File

@ -6,7 +6,7 @@ from tests.integration_tests.vdb.test_vector_store import (
) )
class TestPgvectoRSVector(AbstractVectorTest): class PGVectoRSVectorTest(AbstractVectorTest):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.vector = PGVectoRS( self.vector = PGVectoRS(
@ -34,4 +34,4 @@ class TestPgvectoRSVector(AbstractVectorTest):
assert len(ids) == 1 assert len(ids) == 1
def test_pgvecot_rs(setup_mock_redis): def test_pgvecot_rs(setup_mock_redis):
TestPgvectoRSVector().run_all_tests() PGVectoRSVectorTest().run_all_tests()

View File

@ -7,7 +7,7 @@ from tests.integration_tests.vdb.test_vector_store import (
) )
class TestPGVector(AbstractVectorTest): class PGVectorTest(AbstractVectorTest):
def __init__(self): def __init__(self):
super().__init__() super().__init__()
self.vector = PGVector( self.vector = PGVector(
@ -27,4 +27,4 @@ class TestPGVector(AbstractVectorTest):
def test_pgvector(setup_mock_redis): def test_pgvector(setup_mock_redis):
TestPGVector().run_all_tests() PGVectorTest().run_all_tests()