Пример #1
0
def test_cli(tmpdir, config, uses_output_dir):
    args = ['--uses', 'tests/integration/optimizers/optimizer_conf.yml']
    output_dir = os.path.join(tmpdir, 'out')
    if uses_output_dir:
        args.extend(['--output-dir', output_dir])
    run_optimizer_cli(set_optimizer_parser().parse_args(args))
    if uses_output_dir:
        assert yaml.load(open(os.path.join(
            output_dir, 'best_parameters.yml'))) == BEST_PARAMETERS
Пример #2
0
def test_cli(tmpdir, config, uses_output_dir):
    print(os.environ['JINA_OPTIMIZER_PARAMETER_FILE'])
    args = ['--uses', os.path.join(cur_dir, 'optimizer_conf.yml')]
    output_dir = os.path.join(tmpdir, 'out')
    if uses_output_dir:
        args.extend(['--output-dir', output_dir])
    run_optimizer_cli(set_optimizer_parser().parse_args(args))
    if uses_output_dir:
        assert (yaml.load(open(os.path.join(
            output_dir, 'best_parameters.yml'))) == BEST_PARAMETERS)