jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: add tests covering all methods of vector store (#3849)

Browse source
This commit is contained in:
Bowen Liang 2024-04-25 22:27:30 +08:00 • committed by GitHub
commit 45dd1683fd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 48 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
api/tests/integration_tests/vdb/milvus/test_milvus.py
View file

@ -24,6 +24,13 @@ class TestMilvusVector(AbstractTestVector):
hits_by_full_text = self.vector.search_by_full_text(query=get_sample_text())
assert len(hits_by_full_text) == 0
def delete_document_by_id(self):
self.vector.delete_by_document_id(self.dataset_id)
def get_ids_by_metadata_field(self):
ids = self.vector.get_ids_by_metadata_field('document_id', self.dataset_id)
assert len(ids) >= 1
def test_milvus_vector(setup_mock_redis):
TestMilvusVector().run_all_test()
TestMilvusVector().run_all_tests()