milvus-io / milvus

        assert res == 0

    # TODO: enable

    @pytest.mark.level(2)

    @pytest.mark.timeout(20)

    def _test_collection_rows_count_multiprocessing(self, connect, collection, args):

        '''

        target: test collection rows_count is correct or not with multiprocess

        method: create collection and add vectors in it,

            assert the value returned by count_entities method is equal to length of vectors

        expected: the count is equal to the length of vectors

        '''

        nq = 2

        vectors = gen_vectors(nq, dim)

        res = connect.insert(collection_name=collection, records=vectors)

        time.sleep(add_time_interval)

        def rows_count(milvus):

            status, res = milvus.count_entities(collection)

            logging.getLogger().info(status)

            assert res == nq

        process_num = 8

        processes = []

        for i in range(process_num):

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

            p = Process(target=rows_count, args=(milvus, ))

            processes.append(p)

            p.start()

            logging.getLogger().info(p)

        for p in processes:

            p.join()

    def test_collection_rows_count_multi_collections(self, connect):

        '''

        assert res == 0

    # TODO: enable

    @pytest.mark.timeout(60)

    def _test_collection_rows_count_multiprocessing(self, connect, ip_collection, args):

        '''

        target: test collection rows_count is correct or not with multiprocess

        method: create collection and add vectors in it,

            assert the value returned by count_entities method is equal to length of vectors

        expected: the count is equal to the length of vectors

        '''

        nq = 2

        vectors = gen_vectors(nq, dim)

        res = connect.insert(collection_name=ip_collection, records=vectors)

        time.sleep(add_time_interval)

        def rows_count(milvus):

            status, res = milvus.count_entities(ip_collection)

            logging.getLogger().info(status)

            assert res == nq

        process_num = 8

        processes = []

        for i in range(process_num):

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

            p = Process(target=rows_count, args=(milvus,))

            processes.append(p)

            p.start()

            logging.getLogger().info(p)

        for p in processes:

            p.join()

    def test_collection_rows_count_multi_collections(self, connect):

        '''