Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def test_registry_sklearn():
assert (
registry("us-west-1", scikit_learn_framework_name)
== "746614075791.dkr.ecr.us-west-1.amazonaws.com"
)
assert (
registry("us-west-2", scikit_learn_framework_name)
== "246618743249.dkr.ecr.us-west-2.amazonaws.com"
)
assert (
registry("us-east-1", scikit_learn_framework_name)
== "683313688378.dkr.ecr.us-east-1.amazonaws.com"
)
assert (
registry("us-east-2", scikit_learn_framework_name)
== "257758044811.dkr.ecr.us-east-2.amazonaws.com"
)
assert (
registry("ap-northeast-1", scikit_learn_framework_name)
== "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
)
assert (
registry("ap-northeast-2", scikit_learn_framework_name)
== "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
)
assert (
registry("ap-southeast-1", scikit_learn_framework_name)
== "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
)
assert (
registry("ap-southeast-2", scikit_learn_framework_name)
)
assert (
registry("ap-southeast-2", "sparkml-serving")
== "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
)
assert (
registry("ap-south-1", "sparkml-serving") == "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
)
assert (
registry("eu-west-1", "sparkml-serving") == "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
)
assert (
registry("eu-west-2", "sparkml-serving") == "764974769150.dkr.ecr.eu-west-2.amazonaws.com"
)
assert (
registry("eu-central-1", "sparkml-serving")
== "492215442770.dkr.ecr.eu-central-1.amazonaws.com"
)
assert (
registry("ca-central-1", "sparkml-serving")
== "341280168497.dkr.ecr.ca-central-1.amazonaws.com"
)
assert (
registry("us-gov-west-1", "sparkml-serving")
== "414596584902.dkr.ecr.us-gov-west-1.amazonaws.com"
)
assert (
registry("us-iso-east-1", "sparkml-serving")
== "833128469047.dkr.ecr.us-iso-east-1.c2s.ic.gov"
)
def test_registry_sparkml_serving():
assert (
registry("us-west-1", "sparkml-serving") == "746614075791.dkr.ecr.us-west-1.amazonaws.com"
)
assert (
registry("us-west-2", "sparkml-serving") == "246618743249.dkr.ecr.us-west-2.amazonaws.com"
)
assert (
registry("us-east-1", "sparkml-serving") == "683313688378.dkr.ecr.us-east-1.amazonaws.com"
)
assert (
registry("us-east-2", "sparkml-serving") == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
)
assert (
registry("ap-northeast-1", "sparkml-serving")
== "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
)
assert (
registry("ap-northeast-2", "sparkml-serving")
== "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
)
assert (
def test_region_invalid():
with pytest.raises(KeyError):
registry("us-west-5", "scikit-learn")
registry("ap-northeast-2", "sparkml-serving")
== "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
)
assert (
registry("ap-southeast-1", "sparkml-serving")
== "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
)
assert (
registry("ap-southeast-2", "sparkml-serving")
== "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
)
assert (
registry("ap-south-1", "sparkml-serving") == "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
)
assert (
registry("eu-west-1", "sparkml-serving") == "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
)
assert (
registry("eu-west-2", "sparkml-serving") == "764974769150.dkr.ecr.eu-west-2.amazonaws.com"
)
assert (
registry("eu-central-1", "sparkml-serving")
== "492215442770.dkr.ecr.eu-central-1.amazonaws.com"
)
assert (
registry("ca-central-1", "sparkml-serving")
== "341280168497.dkr.ecr.ca-central-1.amazonaws.com"
)
assert (
registry("us-gov-west-1", "sparkml-serving")
== "414596584902.dkr.ecr.us-gov-west-1.amazonaws.com"
)
def test_registry_sklearn():
assert (
registry("us-west-1", scikit_learn_framework_name)
== "746614075791.dkr.ecr.us-west-1.amazonaws.com"
)
assert (
registry("us-west-2", scikit_learn_framework_name)
== "246618743249.dkr.ecr.us-west-2.amazonaws.com"
)
assert (
registry("us-east-1", scikit_learn_framework_name)
== "683313688378.dkr.ecr.us-east-1.amazonaws.com"
)
assert (
registry("us-east-2", scikit_learn_framework_name)
== "257758044811.dkr.ecr.us-east-2.amazonaws.com"
)
assert (
registry("ap-northeast-1", scikit_learn_framework_name)
== "683313688378.dkr.ecr.us-east-1.amazonaws.com"
)
assert (
registry("us-east-2", scikit_learn_framework_name)
== "257758044811.dkr.ecr.us-east-2.amazonaws.com"
)
assert (
registry("ap-northeast-1", scikit_learn_framework_name)
== "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
)
assert (
registry("ap-northeast-2", scikit_learn_framework_name)
== "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
)
assert (
registry("ap-southeast-1", scikit_learn_framework_name)
== "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
)
assert (
registry("ap-southeast-2", scikit_learn_framework_name)
== "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
)
assert (
registry("ap-south-1", scikit_learn_framework_name)
== "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
)
assert (
registry("eu-west-1", scikit_learn_framework_name)
== "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
)
assert (
registry("eu-west-2", scikit_learn_framework_name)
def test_registry_sparkml_serving():
assert (
registry("us-west-1", "sparkml-serving") == "746614075791.dkr.ecr.us-west-1.amazonaws.com"
)
assert (
registry("us-west-2", "sparkml-serving") == "246618743249.dkr.ecr.us-west-2.amazonaws.com"
)
assert (
registry("us-east-1", "sparkml-serving") == "683313688378.dkr.ecr.us-east-1.amazonaws.com"
)
assert (
registry("us-east-2", "sparkml-serving") == "257758044811.dkr.ecr.us-east-2.amazonaws.com"
)
assert (
registry("ap-northeast-1", "sparkml-serving")
== "354813040037.dkr.ecr.ap-northeast-1.amazonaws.com"
)
assert (
registry("ap-northeast-2", "sparkml-serving")
def test_sparkml_model(sagemaker_session):
sparkml = SparkMLModel(sagemaker_session=sagemaker_session, model_data=MODEL_DATA, role=ROLE)
assert sparkml.image == registry(REGION, "sparkml-serving") + "/sagemaker-sparkml-serving:2.2"
== "366743142698.dkr.ecr.ap-northeast-2.amazonaws.com"
)
assert (
registry("ap-southeast-1", scikit_learn_framework_name)
== "121021644041.dkr.ecr.ap-southeast-1.amazonaws.com"
)
assert (
registry("ap-southeast-2", scikit_learn_framework_name)
== "783357654285.dkr.ecr.ap-southeast-2.amazonaws.com"
)
assert (
registry("ap-south-1", scikit_learn_framework_name)
== "720646828776.dkr.ecr.ap-south-1.amazonaws.com"
)
assert (
registry("eu-west-1", scikit_learn_framework_name)
== "141502667606.dkr.ecr.eu-west-1.amazonaws.com"
)
assert (
registry("eu-west-2", scikit_learn_framework_name)
== "764974769150.dkr.ecr.eu-west-2.amazonaws.com"
)
assert (
registry("eu-central-1", scikit_learn_framework_name)
== "492215442770.dkr.ecr.eu-central-1.amazonaws.com"
)
assert (
registry("ca-central-1", scikit_learn_framework_name)
== "341280168497.dkr.ecr.ca-central-1.amazonaws.com"
)
assert (
registry("us-gov-west-1", scikit_learn_framework_name)