From 82edb4d2df25b289fb5b5fe364e693f50f9a170f Mon Sep 17 00:00:00 2001 From: zhuwenxing Date: Thu, 5 Jan 2023 15:33:39 +0800 Subject: [PATCH] [test]Add assert for chaos test (#21518) Signed-off-by: zhuwenxing --- .../test_all_collections_after_chaos.py | 16 +++++++++++----- tests/python_client/testcases/test_e2e.py | 16 ++++++++++------ 2 files changed, 21 insertions(+), 11 deletions(-) diff --git a/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py b/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py index 022f9a8474..137c5cd63e 100644 --- a/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py +++ b/tests/python_client/chaos/testcases/test_all_collections_after_chaos.py @@ -97,17 +97,23 @@ class TestAllCollection(TestcaseBase): log.info(f"assert load: {tt}") # search - search_vectors = cf.gen_vectors(1, ct.default_dim) + nq = 5 + topk = 5 + search_vectors = cf.gen_vectors(nq, ct.default_dim) t0 = time.time() - res_1, _ = collection_w.search(data=search_vectors, - anns_field=ct.default_float_vec_field_name, - param=search_params, limit=1) + res, _ = collection_w.search(data=search_vectors, + anns_field=ct.default_float_vec_field_name, + param=search_params, limit=topk) tt = time.time() - t0 log.info(f"assert search: {tt}") + assert len(res) == nq + assert len(res[0]) <= topk # query - term_expr = f'{ct.default_int64_field_name} in [1001,1201,4999,2999]' + term_expr = f'{ct.default_int64_field_name} in [1, 2, 3, 4]' t0 = time.time() res, _ = collection_w.query(term_expr) tt = time.time() - t0 log.info(f"assert query result {len(res)}: {tt}") + assert len(res) >= 4 + diff --git a/tests/python_client/testcases/test_e2e.py b/tests/python_client/testcases/test_e2e.py index 91a808f6d8..83fc2be6ef 100644 --- a/tests/python_client/testcases/test_e2e.py +++ b/tests/python_client/testcases/test_e2e.py @@ -79,17 +79,21 @@ class TestE2e(TestcaseBase): collection_w.load() tt = time.time() - t0 log.info(f"assert load: {tt}") - search_vectors = cf.gen_vectors(1, ct.default_dim) + nq = 5 + topk = 5 + search_vectors = cf.gen_vectors(nq, ct.default_dim) t0 = time.time() - res_1, _ = collection_w.search(data=search_vectors, - anns_field=ct.default_float_vec_field_name, - param=search_params, limit=1) + res, _ = collection_w.search(data=search_vectors, + anns_field=ct.default_float_vec_field_name, + param=search_params, limit=topk) tt = time.time() - t0 log.info(f"assert search: {tt}") - + assert len(res) == nq + assert len(res[0]) <= topk # query - term_expr = f'{ct.default_int64_field_name} in [1001,1201,4999,2999]' + term_expr = f'{ct.default_int64_field_name} in [1, 2, 3, 4]' t0 = time.time() res, _ = collection_w.query(term_expr) tt = time.time() - t0 log.info(f"assert query result {len(res)}: {tt}") + assert len(res) >= 4