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