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

1import subprocess 9 ctx1abcdefghi

2import sys 9 ctx1abcdefghi

3 

4import typer 9 ctx1abcdefghi

5from typer.testing import CliRunner 9 ctx1abcdefghi

6 

7from docs_src.parameter_types.enum import tutorial001 as mod 9 ctx1abcdefghi

8 

9runner = CliRunner() 9 ctx1abcdefghi

10 

11app = typer.Typer() 9 ctx1abcdefghi

12app.command()(mod.main) 9 ctx1abcdefghi

13 

14 

15def test_help(): 9 ctx1abcdefghi

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

17 assert result.exit_code == 0 9 ctx1abcdefghi

18 assert "--network" in result.output 9 ctx1abcdefghi

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

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

21 

22 

23def test_main(): 9 ctx1abcdefghi

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

25 assert result.exit_code == 0 9 ctx1abcdefghi

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

27 

28 

29def test_invalid_case(): 9 ctx1abcdefghi

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

31 assert result.exit_code != 0 9 ctx1abcdefghi

32 assert "Invalid value for '--network': 'CONV' is not one of" in result.output 9 ctx1abcdefghi

33 assert "simple" in result.output 9 ctx1abcdefghi

34 assert "conv" in result.output 9 ctx1abcdefghi

35 assert "lstm" in result.output 9 ctx1abcdefghi

36 

37 

38def test_invalid_other(): 9 ctx1abcdefghi

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

40 assert result.exit_code != 0 9 ctx1abcdefghi

41 assert "Invalid value for '--network': 'capsule' is not one of" in result.output 9 ctx1abcdefghi

42 assert "simple" in result.output 9 ctx1abcdefghi

43 assert "conv" in result.output 9 ctx1abcdefghi

44 assert "lstm" in result.output 9 ctx1abcdefghi

45 

46 

47def test_script(): 9 ctx1abcdefghi

48 result = subprocess.run( 9 ctx1abcdefghi

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

50 capture_output=True, 

51 encoding="utf-8", 

52 ) 

53 assert "Usage" in result.stdout 9 ctx1abcdefghi