milvus-io / milvus

            with pytest.raises(Exception) as e:

                status, result = connect.search(collection, top_k, query_vecs)

    @pytest.mark.level(2)

    def test_search_with_invalid_top_k_ip(self, connect, ip_collection, get_top_k):

        '''

        target: test search fuction, with the wrong top_k

        method: search with top_k

        expected: raise an error, and the connection is normal

        '''

        top_k = get_top_k

        logging.getLogger().info(top_k)

        nprobe = 1

        query_vecs = gen_vectors(1, dim)

        if isinstance(top_k, int):

            status, result = connect.search(ip_collection, top_k, query_vecs)

            assert not status.OK()

        else:

            with pytest.raises(Exception) as e:

                status, result = connect.search(ip_collection, top_k, query_vecs)

    """

    Test search collection with invalid nprobe

    def get_top_k(self, request):

        yield request.param

    @pytest.mark.level(1)

    def test_search_with_invalid_top_k(self, connect, collection, get_top_k):

        '''

        target: test search fuction, with the wrong top_k

        method: search with top_k

        expected: raise an error, and the connection is normal

        '''

        top_k = get_top_k

        logging.getLogger().info(top_k)

        nprobe = 1

        query_vecs = gen_vectors(1, dim)

        if isinstance(top_k, int):

            status, result = connect.search(collection, top_k, query_vecs)

            assert not status.OK()

        else:

            with pytest.raises(Exception) as e:

                status, result = connect.search(collection, top_k, query_vecs)

    @pytest.mark.level(2)

    def test_search_with_invalid_top_k_ip(self, connect, ip_collection, get_top_k):