def test_flow_yaml_dump(): f = Flow(optimize_level=FlowOptimizeLevel.IGNORE_GATEWAY, no_gateway=True) f.save_config('test1.yml') fl = Flow.load_config('test1.yml') assert f.args.optimize_level == fl.args.optimize_level rm_files(['test1.yml'])
def test_flow_yaml_dump(): f = Flow(logserver_config=str(cur_dir.parent / 'yaml' / 'test-server-config.yml'), optimize_level=FlowOptimizeLevel.IGNORE_GATEWAY, no_gateway=True) f.save_config('test1.yml') fl = Flow.load_config('test1.yml') assert f.args.logserver_config == fl.args.logserver_config assert f.args.optimize_level == fl.args.optimize_level rm_files(['test1.yml'])
def test_jtype(tmpdir): flow_path = os.path.join(tmpdir, 'flow.yml') exec_path = os.path.join(tmpdir, 'exec.yml') f = Flow() f.save_config(flow_path) with open(flow_path, 'r') as file: conf = yaml.safe_load(file) assert 'jtype' in conf assert conf['jtype'] == 'Flow' e = BaseExecutor() e.save_config(exec_path) with open(exec_path, 'r') as file: conf = yaml.safe_load(file) assert 'jtype' in conf assert conf['jtype'] == 'BaseExecutor' assert type(BaseExecutor.load_config(exec_path)) == BaseExecutor assert type(Flow.load_config(flow_path)) == Flow
def test_flow_yaml_dump(tmpdir): f = Flow() f.save_config(os.path.join(str(tmpdir), 'test1.yml')) fl = Flow.load_config(os.path.join(str(tmpdir), 'test1.yml')) assert f.args.inspect == fl.args.inspect