Python Flow.save_config Examples

Example #1

File: test_flow_yaml_parser.py Project: JHP4911/JINA

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'])

Example #2

File: test_flow_yaml_parser.py Project: yk/jina

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'])

Example #3

File: test_type_parse.py Project: jina-ai/jina

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

Example #4

File: test_flow_yaml_parser.py Project: yuanl/jina

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