Python SageMakerHook.check_s3_url 예제들

예제 #1

 def test_check_s3_url(self, mock_check_prefix, mock_check_bucket, mock_check_key, mock_client):
     mock_client.return_value = None
     hook = SageMakerHook()
     mock_check_bucket.side_effect = [False, True, True, True]
     mock_check_key.side_effect = [False, True, False]
     mock_check_prefix.side_effect = [False, True, True]
     with pytest.raises(AirflowException):
         hook.check_s3_url(data_url)
     with pytest.raises(AirflowException):
         hook.check_s3_url(data_url)
     assert hook.check_s3_url(data_url) is True
     assert hook.check_s3_url(data_url) is True

예제 #2

파일: test_sagemaker.py 프로젝트: lgov/airflow

 def test_check_s3_url(self, mock_check_prefix, mock_check_bucket, mock_check_key, mock_client):
     mock_client.return_value = None
     hook = SageMakerHook()
     mock_check_bucket.side_effect = [False, True, True, True]
     mock_check_key.side_effect = [False, True, False]
     mock_check_prefix.side_effect = [False, True, True]
     self.assertRaises(AirflowException, hook.check_s3_url, data_url)
     self.assertRaises(AirflowException, hook.check_s3_url, data_url)
     self.assertEqual(hook.check_s3_url(data_url), True)
     self.assertEqual(hook.check_s3_url(data_url), True)