milvus-io / milvus

    # TODO: enable

    @pytest.mark.timeout(BUILD_TIMEOUT)

    @pytest.mark.level(2)

    def _test_create_index_multiprocessing(self, connect, collection, 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

        '''

        status, ids = connect.insert(collection, vectors)

        def build(connect):

            status = connect.create_index(collection, IndexType.IVFLAT, {"nlist": NLIST})

            assert status.OK()

        process_num = 8

        processes = []

        for i in range(process_num):

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

            p = Process(target=build, args=(m,))

            processes.append(p)

            p.start()

            time.sleep(0.2)

        for p in processes:

            p.join()

        query_vec = [vectors[0]]

        top_k = 1

        search_param = {"nprobe": nprobe}

        status, result = connect.search(collection, top_k, query_vec, params=search_param)

        assert len(result) == 1

        assert len(result[0]) == top_k

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

    # TODO: enable

    @pytest.mark.timeout(BUILD_TIMEOUT)

        assert len(result) == len(query_vecs)

        logging.getLogger().info(result)

    @pytest.mark.timeout(BUILD_TIMEOUT)

    @pytest.mark.level(2)

    def test_create_index_multithread(self, connect, collection, 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

        '''

        status, ids = connect.insert(collection, vectors)

        def build(connect):

            status = connect.create_index(collection, IndexType.IVFLAT, {"nlist": NLIST})

            assert status.OK()

        threads_num = 8

        threads = []

        for i in range(threads_num):

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

            t = threading.Thread(target=build, args=(m,))

            threads.append(t)

            t.start()

            time.sleep(0.2)

        for t in threads:

            t.join()

        query_vec = [vectors[0]]

        top_k = 1

        search_param = {"nprobe": nprobe}

        status, result = connect.search(collection, top_k, query_vec, params=search_param)

        assert len(result) == 1

        assert len(result[0]) == top_k

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

    @pytest.mark.level(2)

    @pytest.mark.timeout(BUILD_TIMEOUT)

        assert len(result) == len(query_vecs)

    # TODO: enable

    @pytest.mark.timeout(BUILD_TIMEOUT)

    @pytest.mark.level(2)

    def _test_create_index_multiprocessing(self, connect, ip_collection, 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

        '''

        status, ids = connect.insert(ip_collection, vectors)

        def build(connect):

            status = connect.create_index(ip_collection, IndexType.IVFLAT, {"nlist": NLIST})

            assert status.OK()

        process_num = 8

        processes = []

        for i in range(process_num):

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

            p = Process(target=build, args=(m,))

            processes.append(p)

            p.start()

            time.sleep(0.2)

        for p in processes:

            p.join()

        query_vec = [vectors[0]]

        top_k = 1

        search_param = {"nprobe": nprobe}

        status, result = connect.search(ip_collection, top_k, query_vec, params=search_param)

        assert len(result) == 1

        assert len(result[0]) == top_k

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

    # TODO: enable

    @pytest.mark.timeout(BUILD_TIMEOUT)