class TestSageMakerBaseOperator(unittest.TestCase): def setUp(self): self.sagemaker = SageMakerBaseOperator( task_id='test_sagemaker_operator', aws_conn_id='sagemaker_test_id', config=config ) def test_parse_integer(self): self.sagemaker.integer_fields = [['key1'], ['key2', 'key3'], ['key2', 'key4'], ['key5', 'key6']] self.sagemaker.parse_config_integers() self.assertEqual(self.sagemaker.config, parsed_config)
def setUp(self): self.sagemaker = SageMakerBaseOperator( task_id='test_sagemaker_operator', aws_conn_id='sagemaker_test_id', config=config)