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

1import subprocess 7 ctx1abcdefg

2import sys 7 ctx1abcdefg

3 

4from typer.testing import CliRunner 7 ctx1abcdefg

5 

6from docs_src.parameter_types.enum import tutorial001_py310 as mod 7 ctx1abcdefg

7 

8runner = CliRunner() 7 ctx1abcdefg

9app = mod.app 7 ctx1abcdefg

10 

11 

12def test_help(): 7 ctx1abcdefg

13 result = runner.invoke(app, ["--help"]) 7 ctx1abcdefg

14 assert result.exit_code == 0 7 ctx1abcdefg

15 assert "--network" in result.output 7 ctx1abcdefg

16 assert "[simple|conv|lstm]" in result.output 7 ctx1abcdefg

17 assert "default: simple" in result.output 7 ctx1abcdefg

18 

19 

20def test_main(): 7 ctx1abcdefg

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

22 assert result.exit_code == 0 7 ctx1abcdefg

23 assert "Training neural network of type: conv" in result.output 7 ctx1abcdefg

24 

25 

26def test_main_default(): 7 ctx1abcdefg

27 result = runner.invoke(app) 7 ctx1abcdefg

28 assert result.exit_code == 0 7 ctx1abcdefg

29 assert "Training neural network of type: simple" in result.output 7 ctx1abcdefg

30 

31 

32def test_invalid_case(): 7 ctx1abcdefg

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

34 assert result.exit_code != 0 7 ctx1abcdefg

35 assert "Invalid value for '--network'" in result.output 7 ctx1abcdefg

36 assert "'CONV' is not one of" in result.output 7 ctx1abcdefg

37 assert "simple" in result.output 7 ctx1abcdefg

38 assert "conv" in result.output 7 ctx1abcdefg

39 assert "lstm" in result.output 7 ctx1abcdefg

40 

41 

42def test_invalid_other(): 7 ctx1abcdefg

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

44 assert result.exit_code != 0 7 ctx1abcdefg

45 assert "Invalid value for '--network'" in result.output 7 ctx1abcdefg

46 assert "'capsule' is not one of" in result.output 7 ctx1abcdefg

47 assert "simple" in result.output 7 ctx1abcdefg

48 assert "conv" in result.output 7 ctx1abcdefg

49 assert "lstm" in result.output 7 ctx1abcdefg

50 

51 

52def test_script(): 7 ctx1abcdefg

53 result = subprocess.run( 7 ctx1abcdefg

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

55 capture_output=True, 

56 encoding="utf-8", 

57 ) 

58 assert "Usage" in result.stdout 7 ctx1abcdefg