from image_embedding_operator_template import ImageEmbeddingOperatorTemplate


def test_image_embedding_operator(model_name='resnet50', img_path='./test_data/test.jpg', dimension=2048):
    op = ResnetImageEmbedding(model_name)
    embedding = op(img_path)
    # get the output shape in README
    assert (dimension,)==embedding[0].shape