From 067cf3bcc53a6d4ad2fd8ce86dbeb82e4f282656 Mon Sep 17 00:00:00 2001 From: binbin <83755740+binbinlv@users.noreply.github.com> Date: Wed, 27 Oct 2021 21:08:34 +0800 Subject: [PATCH] Add ndarray test case (#10778) Signed-off-by: Binbin Lv --- .../python_client/testcases/test_search_20.py | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/tests/python_client/testcases/test_search_20.py b/tests/python_client/testcases/test_search_20.py index 38a3d5337d..3eb886c78f 100644 --- a/tests/python_client/testcases/test_search_20.py +++ b/tests/python_client/testcases/test_search_20.py @@ -834,6 +834,29 @@ class TestCollectionSearch(TestcaseBase): check_items={"nq": 0, "_async": _async}) + @pytest.mark.tags(CaseLabel.L2) + def test_search_with_ndarray(self, dim, auto_id, _async): + """ + target: test search with ndarray + method: search using ndarray data + expected: search successfully + """ + # 1. initialize without data + collection_w, _, _, insert_ids = self.init_collection_general(prefix, True, + auto_id=auto_id, + dim=dim)[0:4] + # 2. search collection without data + log.info("test_search_with_ndarray: Searching collection %s " + "using ndarray" % collection_w.name) + vectors = np.random.randn(default_nq, dim) + collection_w.search(vectors, default_search_field, default_search_params, + default_limit, default_search_exp, _async=_async, + check_task=CheckTasks.check_search_results, + check_items={"nq": default_nq, + "ids": insert_ids, + "limit": default_limit, + "_async": _async}) + @pytest.mark.tags(CaseLabel.L2) @pytest.mark.parametrize("search_params", [{}, {"params": {}}, {"params": {"nprobe": 10}}]) def test_search_normal_default_params(self, dim, auto_id, search_params, _async):