def test_sagemaker_exception_in_ground_truth(self):
    mock_client = MagicMock()
    mock_exception = ClientError({'Error': {'Message': 'SageMaker broke'}}, 'ground_truth')
    mock_client.create_labeling_job.side_effect = mock_exception
    mock_args = self.parser.parse_args(required_args)
    with self.assertRaises(Exception):
        _utils.get_labeling_job_outputs(mock_client, vars(mock_args))
def test_sagemaker_exception_in_ground_truth(self):
    mock_client = MagicMock()
    mock_exception = ClientError({'Error': {'Message': 'SageMaker broke'}}, 'ground_truth')
    mock_client.create_labeling_job.side_effect = mock_exception
    mock_args = self.parser.parse_args(required_args)
    with self.assertRaises(Exception):
        _utils.get_labeling_job_outputs(mock_client, vars(mock_args))