Coverage for tests/test_tutorial/test_parameter_types/test_enum/test_tutorial004_an.py: 100%

1import subprocess 9 ctx1abcdefghi

2import sys 9 ctx1abcdefghi

3 

4from typer.testing import CliRunner 9 ctx1abcdefghi

5 

6from docs_src.parameter_types.enum import tutorial004_an as mod 9 ctx1abcdefghi

7 

8runner = CliRunner() 9 ctx1abcdefghi

9app = mod.app 9 ctx1abcdefghi

10 

11 

12def test_help(): 9 ctx1abcdefghi

13 result = runner.invoke(app, ["--help"]) 9 ctx1abcdefghi

14 assert result.exit_code == 0 9 ctx1abcdefghi

15 assert "--network [simple|conv|lstm]" in result.output.replace(" ", "") 9 ctx1abcdefghi

16 

17 

18def test_main(): 9 ctx1abcdefghi

19 result = runner.invoke(app, ["--network", "conv"]) 9 ctx1abcdefghi

20 assert result.exit_code == 0 9 ctx1abcdefghi

21 assert "Training neural network of type: conv" in result.output 9 ctx1abcdefghi

22 

23 

24def test_invalid(): 9 ctx1abcdefghi

25 result = runner.invoke(app, ["--network", "capsule"]) 9 ctx1abcdefghi

26 assert result.exit_code != 0 9 ctx1abcdefghi

27 assert "Invalid value for '--network'" in result.output 9 ctx1abcdefghi

28 assert ( 9 ctx1abcdefghi

29 "invalid choice: capsule. (choose from" in result.output 

30 or "'capsule' is not one of" in result.output 

31 ) 

32 assert "simple" in result.output 9 ctx1abcdefghi

33 assert "conv" in result.output 9 ctx1abcdefghi

34 assert "lstm" in result.output 9 ctx1abcdefghi

35 

36 

37def test_script(): 9 ctx1abcdefghi

38 result = subprocess.run( 9 ctx1abcdefghi

39 [sys.executable, "-m", "coverage", "run", mod.__file__, "--help"], 

40 capture_output=True, 

41 encoding="utf-8", 

42 ) 

43 assert "Usage" in result.stdout 9 ctx1abcdefghi