milvus-io / milvus

        assert result[0][0].distance == 0.0

    # TODO: enable

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def _test_create_index_multiprocessing_multicollection(self, connect, args):

        '''

        target: test create index interface with multiprocess

        method: create collection and add vectors in it, create index

        expected: return code equals to 0, and search success

        '''

        process_num = 8

        loop_num = 8

        processes = []

        collection = []

        j = 0

        while j < (process_num*loop_num):

            collection_name = gen_unique_str("test_create_index_multiprocessing")

            collection.append(collection_name)

            param = {'collection_name': collection_name,

                    'dimension': dim,

                    'index_type': IndexType.FLAT,

                    'store_raw_vector': False}

            connect.create_collection(param)

            j = j + 1

        def create_index():

            i = 0

            while i < loop_num:

                # assert connect.has_collection(collection[ids*process_num+i])

                status, ids = connect.insert(collection[ids*process_num+i], vectors)

                status = connect.create_index(collection[ids*process_num+i], IndexType.IVFLAT, {"nlist": NLIST})

                assert status.OK()

                query_vec = [vectors[0]]

                top_k = 1

                search_param = {"nprobe": nprobe}

                status, result = connect.search(collection[ids*process_num+i], top_k, query_vec, params=search_param)

                assert len(result) == 1

                assert len(result[0]) == top_k

                assert result[0][0].distance == 0.0

                i = i + 1

        for i in range(process_num):

            m = get_milvus(host=args["ip"], port=args["port"], handler=args["handler"])

            ids = i

            p = Process(target=create_index, args=(m,ids))

            processes.append(p)

            p.start()

            time.sleep(0.2)

        for p in processes:

            p.join()

    def test_create_index_collection_not_existed(self, connect):

        '''

        assert len(result[0]) == top_k

        assert result[0][0].distance == 0.0

    @pytest.mark.level(2)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    def test_create_index_multithread_multicollection(self, connect, args):

        '''

        target: test create index interface with multiprocess

        method: create collection and add vectors in it, create index

        expected: return code equals to 0, and search success

        '''

        threads_num = 8

        loop_num = 8

        threads = []

        collection = []

        j = 0

        while j < (threads_num*loop_num):

            collection_name = gen_unique_str("test_create_index_multiprocessing")

            collection.append(collection_name)

            param = {'collection_name': collection_name,

                     'dimension': dim,

                     'index_type': IndexType.FLAT,

                     'store_raw_vector': False}

            connect.create_collection(param)

            j = j + 1

        def create_index():

            i = 0

            while i < loop_num:

                # assert connect.has_collection(collection[ids*process_num+i])

                status, ids = connect.insert(collection[ids*threads_num+i], vectors)

                status = connect.create_index(collection[ids*threads_num+i], IndexType.IVFLAT, {"nlist": NLIST})

                assert status.OK()

                query_vec = [vectors[0]]

                top_k = 1

                search_param = {"nprobe": nprobe}

                status, result = connect.search(collection[ids*threads_num+i], top_k, query_vec, params=search_param)

                assert len(result) == 1

                assert len(result[0]) == top_k

                assert result[0][0].distance == 0.0

                i = i + 1

        for i in range(threads_num):

            m = get_milvus(host=args["ip"], port=args["port"], handler=args["handler"])

            ids = i

            t = threading.Thread(target=create_index, args=(m, ids))

            threads.append(t)

            t.start()

            time.sleep(0.2)

        for t in threads:

            t.join()

    @pytest.mark.timeout(BUILD_TIMEOUT)

    @pytest.mark.level(2)