milvus-io / milvus

        res_count = connect.count_entities(id_collection)

        assert res_count == nb

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_the_same_ids(self, connect, id_collection, insert_count):

        '''

        target: test insert vectors in collection, use customize the same ids

        method: create collection and insert vectors in it, check the ids returned and the collection length after vectors inserted

        expected: the length of ids and the collection row count

        '''

        nb = insert_count

        ids = [1 for i in range(nb)]

        res_ids = connect.bulk_insert(id_collection, gen_entities(nb), ids)

        connect.flush([id_collection])

        assert len(res_ids) == nb

        assert res_ids == ids

        res_count = connect.count_entities(id_collection)

        assert res_count == nb

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_ids_fields(self, connect, get_filter_field, get_vector_field):

    def insert_count(self, request):

        yield request.param

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_ids(self, connect, id_collection, insert_count):

        '''

        target: test insert vectors in collection, use customize ids

        method: create collection and insert vectors in it, check the ids returned and the collection length after vectors inserted

        expected: the length of ids and the collection row count

        '''

        nb = insert_count

        ids = [i for i in range(nb)]

        res_ids = connect.bulk_insert(id_collection, gen_entities(nb), ids)

        connect.flush([id_collection])

        assert len(res_ids) == nb

        assert res_ids == ids

        res_count = connect.count_entities(id_collection)

        assert res_count == nb

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_the_same_ids(self, connect, id_collection, insert_count):