catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Tech debt (#3653)

Browse source
This commit is contained in:
Bert Blommers 2021-02-02 16:31:26 +00:00 • committed by GitHub
commit 737636f9df
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
20 changed files with 11 additions and 57 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_sagemaker/test_sagemaker_models.py
View file

@ -45,6 +45,14 @@ def test_describe_model():
assert model.get("ModelName").should.equal("blah")
@mock_sagemaker
def test_describe_model_not_found():
client = boto3.client("sagemaker", region_name="us-east-1")
with pytest.raises(ClientError) as err:
client.describe_model(ModelName="unknown")
assert err.value.response["Error"]["Message"].should.contain("Could not find model")
@mock_sagemaker
def test_create_model():
client = boto3.client("sagemaker", region_name="us-east-1")