def testFileParam(self):
file_param = job_model.FileParam('my_name',
'my_value',
'my_docker_path',
'my_remote_uri',
recursive=True,
file_provider=job_model.P_GCS)
self.assertEqual('my_name', file_param.name)
self.assertEqual('my_docker_path', file_param.docker_path)
self.assertEqual('my_remote_uri', file_param.uri)
self.assertTrue(file_param.recursive)
('glog', 'gs://bucket/my.log', 'gs://bucket/my.log', job_model.P_GCS),
])
def test_logging_param_maker(self, unused_name, uri, expected_out,
provider):
del unused_name
param = param_util.build_logging_param(
uri, util_class=param_util.OutputFileParamUtil)
self.assertEqual(param.uri, expected_out)
self.assertEqual(param.file_provider, provider)
TASK_DESCRIPTORS = [
job_model.TaskDescriptor(
None, {
'inputs': [
job_model.FileParam(
'IN', uri='gs://in/*', file_provider=job_model.P_GCS)
],
'outputs': []
}, None),
job_model.TaskDescriptor(
None, {
'inputs': [],
'outputs': [
job_model.FileParam(
'OUT', uri='gs://out/*', file_provider=job_model.P_GCS)
]
}, None),
]
class TestSubmitValidator(unittest.TestCase):