milvus-io
/
milvus
mirror of https://github.com/milvus-io/milvus.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update query de-dup cases (#11637) 

Signed-off-by: zhuwenxing <wenxing.zhu@zilliz.com>
pull/11666/head
zhuwenxing 2021-11-11 18:45:02 +08:00 committed by GitHub
parent 77b6439c65
commit ce7c1637cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/python_client/testcases/test_query.py
View File

@ -90,8 +90,9 @@ class TestQueryParams(TestcaseBase):
check_items={exp_res: res[:1]}) check_items={exp_res: res[:1]})
@pytest.mark.tags(CaseLabel.L1) @pytest.mark.tags(CaseLabel.L1)
@pytest.mark.parametrize("dup_times",[1,2,3])
@pytest.mark.parametrize("dim", [8, 128]) @pytest.mark.parametrize("dim", [8, 128])
def test_query_with_dup_primary_key(self, dim): def test_query_with_dup_primary_key(self, dim, dup_times):
""" """
target: test query with duplicate primary key target: test query with duplicate primary key
method: 1.insert same data twice method: 1.insert same data twice
@ -100,8 +101,9 @@ class TestQueryParams(TestcaseBase):
""" """
nb = ct.default_nb nb = ct.default_nb
collection_w, insert_data, _, _ = self.init_collection_general(prefix, True, nb, dim=dim)[0:4] collection_w, insert_data, _, _ = self.init_collection_general(prefix, True, nb, dim=dim)[0:4]
# insert data again # insert dup data multi times
collection_w.insert(insert_data[0]) for i in range(dup_times):
collection_w.insert(insert_data[0])
# query # query
res, _ = collection_w.query(default_term_expr) res, _ = collection_w.query(default_term_expr)
# assert that query results are de-duplicated # assert that query results are de-duplicated