def test_main(self):
    ground_truth._utils = MagicMock()
    ground_truth._utils.add_default_client_arguments = _utils.add_default_client_arguments
    ground_truth._utils.get_labeling_job_outputs.return_value = ('s3://fake-bucket/output', 'arn:aws:sagemaker:us-east-1:999999999999:labeling-job')
    with patch('builtins.open', mock_open()) as file_open:
        ground_truth.main(required_args)
    ground_truth._utils.create_labeling_job.assert_called()
    ground_truth._utils.wait_for_labeling_job.assert_called()
    ground_truth._utils.get_labeling_job_outputs.assert_called()
    file_open.assert_has_calls([call('/tmp/output_manifest_location.txt', 'w'), call('/tmp/active_learning_model_arn.txt', 'w')], any_order=True)
    file_open().write.assert_has_calls([call('s3://fake-bucket/output'), call('arn:aws:sagemaker:us-east-1:999999999999:labeling-job')], any_order=False)
def test_main(self):
    ground_truth._utils = MagicMock()
    ground_truth._utils.add_default_client_arguments = _utils.add_default_client_arguments
    ground_truth._utils.get_labeling_job_outputs.return_value = ('s3://fake-bucket/output', 'arn:aws:sagemaker:us-east-1:999999999999:labeling-job')
    with patch('builtins.open', mock_open()) as file_open:
        ground_truth.main(required_args)
    ground_truth._utils.create_labeling_job.assert_called()
    ground_truth._utils.wait_for_labeling_job.assert_called()
    ground_truth._utils.get_labeling_job_outputs.assert_called()
    file_open.assert_has_calls([call('/tmp/output_manifest_location.txt', 'w'), call('/tmp/active_learning_model_arn.txt', 'w')], any_order=True)
    file_open().write.assert_has_calls([call('s3://fake-bucket/output'), call('arn:aws:sagemaker:us-east-1:999999999999:labeling-job')], any_order=False)