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

1import subprocess 8 ctx1abcdefgh

2import sys 8 ctx1abcdefgh

3 

4from typer.testing import CliRunner 8 ctx1abcdefgh

5 

6from docs_src.parameter_types.enum import tutorial001_py39 as mod 8 ctx1abcdefgh

7 

8runner = CliRunner() 8 ctx1abcdefgh

9app = mod.app 8 ctx1abcdefgh

10 

11 

12def test_help(): 8 ctx1abcdefgh

13 result = runner.invoke(app, ["--help"]) 8 ctx1abcdefgh

14 assert result.exit_code == 0 8 ctx1abcdefgh

15 assert "--network" in result.output 8 ctx1abcdefgh

16 assert "[simple|conv|lstm]" in result.output 8 ctx1abcdefgh

17 assert "default: simple" in result.output 8 ctx1abcdefgh

18 

19 

20def test_main(): 8 ctx1abcdefgh

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

22 assert result.exit_code == 0 8 ctx1abcdefgh

23 assert "Training neural network of type: conv" in result.output 8 ctx1abcdefgh

24 

25 

26def test_main_default(): 8 ctx1abcdefgh

27 result = runner.invoke(app) 8 ctx1abcdefgh

28 assert result.exit_code == 0 8 ctx1abcdefgh

29 assert "Training neural network of type: simple" in result.output 8 ctx1abcdefgh

30 

31 

32def test_invalid_case(): 8 ctx1abcdefgh

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

34 assert result.exit_code != 0 8 ctx1abcdefgh

35 assert "Invalid value for '--network'" in result.output 8 ctx1abcdefgh

36 assert "'CONV' is not one of" in result.output 8 ctx1abcdefgh

37 assert "simple" in result.output 8 ctx1abcdefgh

38 assert "conv" in result.output 8 ctx1abcdefgh

39 assert "lstm" in result.output 8 ctx1abcdefgh

40 

41 

42def test_invalid_other(): 8 ctx1abcdefgh

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

44 assert result.exit_code != 0 8 ctx1abcdefgh

45 assert "Invalid value for '--network'" in result.output 8 ctx1abcdefgh

46 assert "'capsule' is not one of" in result.output 8 ctx1abcdefgh

47 assert "simple" in result.output 8 ctx1abcdefgh

48 assert "conv" in result.output 8 ctx1abcdefgh

49 assert "lstm" in result.output 8 ctx1abcdefgh

50 

51 

52def test_script(): 8 ctx1abcdefgh

53 result = subprocess.run( 8 ctx1abcdefgh

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

55 capture_output=True, 

56 encoding="utf-8", 

57 ) 

58 assert "Usage" in result.stdout 8 ctx1abcdefgh