未验证 提交 e5498b05 编写于 作者: B binbin 提交者: GitHub

Reduce insert data number for maximum dim search test case (#9737)

Signed-off-by: NBinbin Lv <binbin.lv@zilliz.com>
上级 e432fd19
......@@ -1081,26 +1081,27 @@ class TestCollectionSearch(TestcaseBase):
@pytest.mark.tags(CaseLabel.L2)
#@pytest.mark.skip(reason="skip temporarily for debug")
def test_search_max_dim(self, nq, auto_id, _async):
def test_search_max_dim(self, auto_id, _async):
"""
target: test search with max configuration
method: create connection, collection, insert and search with max dim
expected: search successfully with limit(topK)
"""
# 1. initialize with data
collection_w, _, _, insert_ids = self.init_collection_general(prefix, True, default_nb,
collection_w, _, _, insert_ids = self.init_collection_general(prefix, True, 100,
auto_id=auto_id,
dim=max_dim)
# 2. search
nq = 2
log.info("test_search_max_dim: searching collection %s" % collection_w.name)
vectors = [[random.random() for _ in range(max_dim)] for _ in range(nq)]
collection_w.search(vectors[:nq], default_search_field,
default_search_params, 2,
default_search_params, nq,
default_search_exp, _async=_async,
check_task=CheckTasks.check_search_results,
check_items={"nq": nq,
"ids": insert_ids,
"limit": 2,
"limit": nq,
"_async": _async})
@pytest.mark.tags(CaseLabel.L2)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册