from towhee import pipeline


def test_image_embedding_resnet50(repo_name='towhee/image-embedding-pipeline-template', img_path='./test_data/test.jpg', dimension=2048):
    embedding_pipeline = pipeline(repo_name)
    (embedding,) = embedding_pipeline(img_path)
    # get the output shape in README
    assert (dimension,)==embedding[0].shape