def test_delete_dataset(make_stubber, error_code):
lookoutvision_client = boto3.client('lookoutvision')
lookoutvision_stubber = make_stubber(lookoutvision_client)
project_name = 'test-project_name'
dataset_type = 'train'
lookoutvision_stubber.stub_delete_dataset(project_name,
dataset_type,
error_code=error_code)
if error_code is None:
Datasets.delete_dataset(lookoutvision_client, project_name,
dataset_type)
else:
with pytest.raises(ClientError) as exc_info:
Datasets.delete_dataset(lookoutvision_client, project_name,
dataset_type)
assert exc_info.value.response['Error']['Code'] == error_code
