def test_run_error():
with pytest.raises(_errors.ExecuteUserScriptError) as e:
_modules.run('wrong module')
message = str(e.value)
assert 'ExecuteUserScriptError:' in message
assert ' No module named wrong module' in message
def test_run_no_wait(log_script_invocation, create, executable):
_modules.run('pytest', ['--version'], {'PYPATH': '/opt/ml/code'}, wait=False, capture_error=True)
expected_cmd = [executable(), '-m', 'pytest', '--version']
log_script_invocation.assert_called_with(expected_cmd, {'PYPATH': '/opt/ml/code'})
create.assert_called_with(expected_cmd, _errors.ExecuteUserScriptError,
capture_error=True)
def test_run(log_script_invocation, check_error, executable):
_modules.run('pytest', ['--version'])
expected_cmd = [executable(), '-m', 'pytest', '--version']
log_script_invocation.assert_called_with(expected_cmd, {})
check_error.assert_called_with(expected_cmd, _errors.ExecuteUserScriptError,
capture_error=False)
def test_run(log_script_invocation, check_error, executable):
_modules.run("pytest", ["--version"])
expected_cmd = [executable(), "-m", "pytest", "--version"]
log_script_invocation.assert_called_with(expected_cmd, {})
check_error.assert_called_with(expected_cmd,
_errors.ExecuteUserScriptError,
capture_error=False)
def test_run_error(capture_error):
with pytest.raises(_errors.ExecuteUserScriptError) as e:
_modules.run("wrong module", capture_error=capture_error)
message = str(e.value)
assert "ExecuteUserScriptError:" in message
if capture_error:
assert " No module named wrong module" in message
def test_run_no_wait(log_script_invocation, create, executable):
_modules.run(
"pytest", ["--version"], {"PYPATH": "/opt/ml/code"}, wait=False, capture_error=True
)
expected_cmd = [executable(), "-m", "pytest", "--version"]
log_script_invocation.assert_called_with(expected_cmd, {"PYPATH": "/opt/ml/code"})
create.assert_called_with(expected_cmd, _errors.ExecuteUserScriptError, capture_error=True)
def test_run():
_modules.run('pytest', ['--version'])
