def test_image(sagemaker_session):
    object2vec = Object2Vec(sagemaker_session=sagemaker_session, **ALL_REQ_ARGS)
    assert object2vec.train_image() == registry(REGION, "object2vec") + "/object2vec:1"