milvus-io / milvus

    ******************************************************************

    """

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_ids(self, connect, ip_collection):

        '''

        target: test add vectors in collection, use customize ids

        method: create collection and add vectors in it, check the ids returned and the collection length after vectors added

        expected: the length of ids and the collection row count

        '''

        nq = 5

        top_k = 1

        vectors = gen_vectors(nq, dim)

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

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

        assert status.OK()

        connect.flush([ip_collection])

        assert len(ids) == nq

        # check search result

        status, result = connect.search(ip_collection, top_k, vectors)

        logging.getLogger().info(result)

        assert len(result) == nq

        for i in range(nq):

            assert result[i][0].id == i

    @pytest.mark.timeout(ADD_TIMEOUT)

    ******************************************************************

    """

    @pytest.mark.timeout(ADD_TIMEOUT)

    def test_insert_ids(self, connect, collection):

        '''

        target: test add vectors in collection, use customize ids

        method: create collection and add vectors in it, check the ids returned and the collection length after vectors added

        expected: the length of ids and the collection row count

        '''

        nq = 5

        top_k = 1

        vectors = gen_vectors(nq, dim)

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

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

        connect.flush([collection])

        assert status.OK()

        assert len(ids) == nq

        status, result = connect.search(

            collection, top_k, query_records=vectors)

        logging.getLogger().info(result)

        assert len(result) == nq

        for i in range(nq):

            assert result[i][0].id == i