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