def test_run_simple_wf():
wfdoc = workflow.load_wfdoc(SIMPLE_WF)
wf = workflow.SimpleWorkflow(
output_dir=tempfile.mkdtemp())
output = wf.run(wfdoc)
print('OUTPUT=', output)
assert 'tas_mon_HadGEM2-ES_rcp85_r1i1p1_20850116-21201216.nc' in output[0]
def test_build_tree():
wfdoc = workflow.load_wfdoc(TREE_WF)
tree = workflow.build_tree(wfdoc)
assert list(tree.edges) == [
("root", "output"),
("output", "average_tas"),
("average_tas", "subset_tas"),
]
def test_run_tree_wf_3():
wfdoc = workflow.load_wfdoc(TREE_WF)
wf = workflow.TreeWorkflow(data_root_dir=CMIP5_ARCHIVE_ROOT,
output_dir=tempfile.mkdtemp())
output = wf.run(wfdoc)
assert 'output.nc' in output[0]
def test_build_tree():
wfdoc = workflow.load_wfdoc(TREE_WF)
tree = workflow.build_tree(wfdoc)
assert list(tree.edges) == [('root', 'output'), ('output', 'average_tas'),
('average_tas', 'subset_tas')]
def test_replace_inputs():
wfdoc = workflow.load_wfdoc(TREE_WF)
steps = workflow.replace_inputs(wfdoc)
assert steps['subset_tas']['in']['data_ref'] == \
["cmip5.output1.MOHC.HadGEM2-ES.rcp85.mon.atmos.Amon.r1i1p1.latest.tas"]
def test_validate_tree_wf():
wfdoc = workflow.load_wfdoc(TREE_WF)
wf = workflow.TreeWorkflow(data_root_dir=CMIP5_ARCHIVE_ROOT,
output_dir=tempfile.mkdtemp())
assert wf.validate(wfdoc) is True
def test_run_simple_wf():
wfdoc = workflow.load_wfdoc(SIMPLE_WF)
wf = workflow.SimpleWorkflow(data_root_dir=CMIP5_ARCHIVE_ROOT,
output_dir=tempfile.mkdtemp())
output = wf.run(wfdoc)
assert 'output.nc' in output[0]
def test_replace_inputs():
wfdoc = workflow.load_wfdoc(TREE_WF)
steps = workflow.replace_inputs(wfdoc)
assert steps["subset_tas"]["in"]["collection"] == [
"cmip5.output1.MOHC.HadGEM2-ES.rcp85.mon.atmos.Amon.r1i1p1.latest.tas"
]
def test_validate_tree_wf(tmp_path):
wfdoc = workflow.load_wfdoc(TREE_WF)
wf = workflow.Workflow(output_dir=tmp_path)
assert wf.validate(wfdoc) is True
def test_run_tree_wf_3():
wfdoc = workflow.load_wfdoc(TREE_WF)
wf = workflow.TreeWorkflow(
output_dir=tempfile.mkdtemp())
output = wf.run(wfdoc)
assert 'tas_mon_HadGEM2-ES_rcp85_r1i1p1_20850116-21201216.nc' in output[0]
def test_validate_tree_wf():
wfdoc = workflow.load_wfdoc(TREE_WF)
wf = workflow.TreeWorkflow(
output_dir=tempfile.mkdtemp())
assert wf.validate(wfdoc) is True
def test_validate_simple_wf():
wfdoc = workflow.load_wfdoc(SIMPLE_WF)
wf = workflow.SimpleWorkflow(
output_dir=tempfile.mkdtemp())
assert wf.validate(wfdoc) is True
