Coverage for tests/test_tutorial/test_parameter_types/test_enum/test_tutorial001.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 tutorial001 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" in result.output 9 ctx1abcdefghi

16 assert "[simple|conv|lstm]" in result.output 9 ctx1abcdefghi

17 assert "default: simple" in result.output 9 ctx1abcdefghi

18 

19 

20def test_main(): 9 ctx1abcdefghi

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

22 assert result.exit_code == 0 9 ctx1abcdefghi

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

24 

25 

26def test_main_default(): 9 ctx1abcdefghi

27 result = runner.invoke(app) 9 ctx1abcdefghi

28 assert result.exit_code == 0 9 ctx1abcdefghi

29 assert "Training neural network of type: simple" in result.output 9 ctx1abcdefghi

30 

31 

32def test_invalid_case(): 9 ctx1abcdefghi

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

34 assert result.exit_code != 0 9 ctx1abcdefghi

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

36 assert "'CONV' is not one of" in result.output 9 ctx1abcdefghi

37 assert "simple" in result.output 9 ctx1abcdefghi

38 assert "conv" in result.output 9 ctx1abcdefghi

39 assert "lstm" in result.output 9 ctx1abcdefghi

40 

41 

42def test_invalid_other(): 9 ctx1abcdefghi

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

44 assert result.exit_code != 0 9 ctx1abcdefghi

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

46 assert "'capsule' is not one of" in result.output 9 ctx1abcdefghi

47 assert "simple" in result.output 9 ctx1abcdefghi

48 assert "conv" in result.output 9 ctx1abcdefghi

49 assert "lstm" in result.output 9 ctx1abcdefghi

50 

51 

52def test_script(): 9 ctx1abcdefghi

53 result = subprocess.run( 9 ctx1abcdefghi

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

55 capture_output=True, 

56 encoding="utf-8", 

57 ) 

58 assert "Usage" in result.stdout 9 ctx1abcdefghi