def test_load_config(transformer):
expected_config = {
"input_config": {
"DataSource": {"S3DataSource": {"S3DataType": S3_DATA_TYPE, "S3Uri": DATA}}
},
"output_config": {"S3OutputPath": OUTPUT_PATH},
"resource_config": {
"InstanceCount": INSTANCE_COUNT,
"InstanceType": INSTANCE_TYPE,
"VolumeKmsKeyId": KMS_KEY_ID,
},
}
actual_config = _TransformJob._load_config(DATA, S3_DATA_TYPE, None, None, None, transformer)
assert actual_config == expected_config
def test_load_config(transformer):
expected_config = {
'input_config': {
'DataSource': {
'S3DataSource': {
'S3DataType': S3_DATA_TYPE,
'S3Uri': DATA,
},
},
},
'output_config': {
'S3OutputPath': OUTPUT_PATH,
},
'resource_config': {
'InstanceCount': INSTANCE_COUNT,
'InstanceType': INSTANCE_TYPE,
},
}
actual_config = _TransformJob._load_config(DATA, S3_DATA_TYPE, None, None, None, transformer)
assert actual_config == expected_config
