admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-07 19:31:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

update testcases of query interfaces (#6175)

Browse Source 
* [skip ci] Update query test cases

Signed-off-by: wangting0128 <ting.wang@zilliz.com>

* [skip ci] Conflict resolution

Signed-off-by: wangting0128 <ting.wang@zilliz.com>

* [skip ci] Update teardown

Signed-off-by: wangting0128 <ting.wang@zilliz.com>

* [skip ci] Update testcases of query interfaces

Signed-off-by: wangting0128 <ting.wang@zilliz.com>
This commit is contained in:
紫晴 2021-06-28 17:00:13 +08:00 committed by GitHub
parent 2bc93a8eae
commit 63d1f049d1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

48
tests20/python_client/testcases/test_query.py
View File

@ -555,19 +555,6 @@ class TestQueryOperation(TestcaseBase):
res, _ = collection_w.query(term_expr)
assert len(res) == len(int_values)
# half = ct.default_nb // 2
# collection_w, partition_w, _, df_default = self.insert_entities_into_two_partitions_in_half(half)
# int_values = df_default[ct.default_int64_field_name].values.tolist()
# float_values = df_default[ct.default_float_field_name].values.tolist()
# vec_values = df_default[ct.default_float_vec_field_name].values.tolist()
# term_expr = f'{ct.default_int64_field_name} in {int_values}'
# res, _ = collection_w.query(term_expr)
# assert len(res) == half
# for i in half:
# assert res[i][ct.default_int64_field_name] == int_values[i]
# assert res[i][ct.default_float_field_name] == float_values[i]
# assert res[i][ct.default_float_vec_field_name] == vec_values[i]
@pytest.mark.xfail(reason="fail")
@pytest.mark.tags(ct.CaseLabel.L3)
def test_query_expr_repeated_term_array(self):
@ -604,14 +591,6 @@ class TestQueryOperation(TestcaseBase):
check_vec = vectors[0].iloc[:, [0, 1]][0:len(int_values)].to_dict('records')
collection_w.query(term_expr, check_task=CheckTasks.check_query_results, check_items={exp_res: check_vec})
# entities, ids = init_data(connect, collection)
# assert len(ids) == ut.default_nb
# connect.create_index(collection, ut.default_float_vec_field_name, get_simple_index)
# connect.load_collection(collection)
# res = connect.query(collection, default_term_expr)
# logging.getLogger().info(res)
@pytest.mark.xfail(reason='')
@pytest.mark.tags(ct.CaseLabel.L3)
def test_query_after_search(self):
"""
@ -634,22 +613,10 @@ class TestQueryOperation(TestcaseBase):
# check number of entities and that method calls the flush interface
assert collection_w.num_entities == nb_old
term_expr = f'{ct.default_int64_field_name} in {default_term_expr}'
check_vec = vectors[0].iloc[:, [0, 1]][0:len(default_term_expr)].to_dict('records')
term_expr = f'{ct.default_int64_field_name} in [0, 1]'
check_vec = vectors[0].iloc[:, [0, 1]][0:2].to_dict('records')
collection_w.query(term_expr, check_task=CheckTasks.check_query_results, check_items={exp_res: check_vec})
# entities, ids = init_data(connect, collection)
# assert len(ids) == ut.default_nb
# top_k = 10
# nq = 2
# query, _ = ut.gen_query_vectors(ut.default_float_vec_field_name, entities, top_k=top_k, nq=nq)
# connect.load_collection(collection)
# search_res = connect.search(collection, query)
# assert len(search_res) == nq
# assert len(search_res[0]) == top_k
# query_res = connect.query(collection, default_term_expr)
# logging.getLogger().info(query_res)
@pytest.mark.tags(ct.CaseLabel.L3)
def test_query_partition_repeatedly(self):
"""
@ -682,17 +649,6 @@ class TestQueryOperation(TestcaseBase):
res_two, _ = collection_w.query(default_term_expr, partition_names=[partition_w.name])
assert res_one == res_two
# conn = self._connect()
# collection_w = self.init_collection_wrap(name=cf.gen_unique_str(prefix))
# partition_w = self.init_partition_wrap(collection_wrap=collection_w)
# df = cf.gen_default_dataframe_data(ct.default_nb)
# partition_w.insert(df)
# conn.flush([collection_w.name])
# partition_w.load()
# res_one, _ = collection_w.query(default_term_expr, partition_names=[partition_w.name])
# res_two, _ = collection_w.query(default_term_expr, partition_names=[partition_w.name])
# assert res_one == res_two
@pytest.mark.tags(ct.CaseLabel.L3)
def test_query_another_partition(self):
"""