Exemplo n.º 1
0
def test_exec_mo_binary(mocker):
    subprocess_run = mocker.patch('subprocess.run')
    mocker.patch('os.chdir')

    args = prepare_args('ModelOptimizer', value_options={'--foo': 'bar'})
    exec_mo_binary(args)

    subprocess_run.assert_called_once_with(args, check=False, timeout=None)
Exemplo n.º 2
0
def test_prepare_args():
    args = prepare_args('foo', ['a', 'b'], {'bar': 123, 'x': 'baz'})
    assert args[0] == sys.executable
    assert args[1] == 'foo'
    assert '--a' in args
    assert '--b' in args
    assert '--bar' in args
    assert '--x' in args

    assert args[args.index('--bar') + 1] == '123'
    assert args[args.index('--x') + 1] == 'baz'