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