milvus-io / milvus

            assert res[1]._distances[0] > epsilon

    # TODO:

    @pytest.mark.level(2)

    def _test_search_index_partitions_B(self, connect, collection, get_simple_index, get_top_k):

        '''

        target: test basic search fuction, all the search params is corrent, test all index params, and build

        method: search collection with the given vectors and tags, check the result

        expected: the length of the result is top_k

        '''

        top_k = get_top_k

        nq = 2

        tag = "tag"

        new_tag = "new_tag"

        index_type = get_simple_index["index_type"]

        if index_type == "IVF_PQ":

            pytest.skip("Skip PQ")

        connect.create_partition(collection, tag)

        connect.create_partition(collection, new_tag)

        entities, ids = init_data(connect, collection, partition_tags=tag)

        new_entities, new_ids = init_data(connect, collection, nb=6001, partition_tags=new_tag)

        connect.create_index(collection, field_name, get_simple_index)

        search_param = get_search_param(index_type)

        query, vecs = gen_query_vectors(field_name, new_entities, top_k, nq, search_params=search_param)

        if top_k > top_k_limit:

            with pytest.raises(Exception) as e:

                res = connect.search(collection, query)

        else:

            res = connect.search(collection, query, partition_tags=["(.*)tag"])

            assert not check_id_result(res[0], ids[0])

            assert check_id_result(res[1], new_ids[0])

            assert res[0]._distances[0] > epsilon

            assert res[1]._distances[0] < epsilon

            res = connect.search(collection, query, partition_tags=["new(.*)"])

            assert res[0]._distances[0] > epsilon

            assert res[1]._distances[0] < epsilon

    # 

    # test for ip metric

            assert len(res) == nq

            assert len(res[0]) == 0

    @pytest.mark.level(2)

    def test_search_index_partitions(self, connect, collection, get_simple_index, get_top_k):

        '''

        target: test basic search fuction, all the search params is corrent, test all index params, and build

        method: search collection with the given vectors and tags, check the result

        expected: the length of the result is top_k

        '''

        top_k = get_top_k

        nq = 2

        new_tag = "new_tag"

        index_type = get_simple_index["index_type"]

        if index_type == "IVF_PQ":

            pytest.skip("Skip PQ")

        connect.create_partition(collection, tag)

        connect.create_partition(collection, new_tag)

        entities, ids = init_data(connect, collection, partition_tags=tag)

        new_entities, new_ids = init_data(connect, collection, nb=6001, partition_tags=new_tag)

        connect.create_index(collection, field_name, get_simple_index)

        search_param = get_search_param(index_type)

        query, vecs = gen_query_vectors(field_name, entities, top_k, nq, search_params=search_param)

        if top_k > top_k_limit:

            with pytest.raises(Exception) as e:

                res = connect.search(collection, query)

        else:

            res = connect.search(collection, query)

            assert check_id_result(res[0], ids[0])

            assert not check_id_result(res[1], new_ids[0])

            assert res[0]._distances[0] < epsilon

            assert res[1]._distances[0] < epsilon

            res = connect.search(collection, query, partition_tags=["new_tag"])

            assert res[0]._distances[0] > epsilon

            assert res[1]._distances[0] > epsilon

    # TODO:

    @pytest.mark.level(2)