milvus-io / milvus

        assert status.OK()

        assert len(ids) == nq

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_tag_A(self, connect, collection):

        '''

        target: test add vectors in collection created before

        method: create partition and add vectors in it

        expected: the collection row count equals to nq

        '''

        nq = 5

        vectors = gen_vectors(nq, dim)

        status = connect.create_partition(collection, tag)

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

        assert status.OK()

        assert len(ids) == nq

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_tag_not_existed(self, connect, collection):

        assert status.OK()

        assert len(ids) == nq

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_tag(self, connect, collection):

        '''

        target: test add vectors in collection created before

        method: create collection and add vectors in it, with the partition_tag param

        expected: the collection row count equals to nq

        '''

        nq = 5

        vectors = gen_vectors(nq, dim)

        status = connect.create_partition(collection, tag)

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

        assert status.OK()

        assert len(ids) == nq

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_tag_A(self, connect, collection):