Skip to content

Commit 32cbaca

Browse files
committed
fix invalid message test
1 parent 1d4168e commit 32cbaca

2 files changed

Lines changed: 7 additions & 6 deletions

File tree

tests/test_tutorial/test_parameter_types/test_enum/test_tutorial004.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,16 +27,17 @@ def test_main():
2727
def test_invalid():
2828
result = runner.invoke(app, ["--network", "capsule"])
2929
assert result.exit_code != 0
30+
assert "Invalid value for '--network'" in result.output
3031
assert (
31-
"Invalid value for '--network': invalid choice: capsule. (choose from"
32-
in result.output
33-
or "Invalid value for '--network': 'capsule' is not one of" in result.output
32+
"invalid choice: capsule. (choose from" in result.output
33+
or "'capsule' is not one of" in result.output
3434
)
3535
assert "simple" in result.output
3636
assert "conv" in result.output
3737
assert "lstm" in result.output
3838

3939

40+
@needs_py38
4041
def test_script():
4142
result = subprocess.run(
4243
[sys.executable, "-m", "coverage", "run", mod.__file__, "--help"],

tests/test_tutorial/test_parameter_types/test_enum/test_tutorial004_an.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,10 @@ def test_main():
2727
def test_invalid():
2828
result = runner.invoke(app, ["--network", "capsule"])
2929
assert result.exit_code != 0
30+
assert "Invalid value for '--network'" in result.output
3031
assert (
31-
"Invalid value for '--network': invalid choice: capsule. (choose from"
32-
in result.output
33-
or "Invalid value for '--network': 'capsule' is not one of" in result.output
32+
"invalid choice: capsule. (choose from" in result.output
33+
or "'capsule' is not one of" in result.output
3434
)
3535
assert "simple" in result.output
3636
assert "conv" in result.output

0 commit comments

Comments
 (0)