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

1import subprocess 8 ctx1abcdefgh

2import sys 8 ctx1abcdefgh

3 

4import typer 8 ctx1abcdefgh

5from typer.testing import CliRunner 8 ctx1abcdefgh

6 

7from docs_src.parameter_types.enum import tutorial001 as mod 8 ctx1abcdefgh

8 

9runner = CliRunner() 8 ctx1abcdefgh

10 

11app = typer.Typer() 8 ctx1abcdefgh

12app.command()(mod.main) 8 ctx1abcdefgh

13 

14 

15def test_help(): 8 ctx1abcdefgh

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

17 assert result.exit_code == 0 8 ctx1abcdefgh

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

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

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

21 

22 

23def test_main(): 8 ctx1abcdefgh

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

25 assert result.exit_code == 0 8 ctx1abcdefgh

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

27 

28 

29def test_invalid_case(): 8 ctx1abcdefgh

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

31 assert result.exit_code != 0 8 ctx1abcdefgh

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

33 assert "simple" in result.output 8 ctx1abcdefgh

34 assert "conv" in result.output 8 ctx1abcdefgh

35 assert "lstm" in result.output 8 ctx1abcdefgh

36 

37 

38def test_invalid_other(): 8 ctx1abcdefgh

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

40 assert result.exit_code != 0 8 ctx1abcdefgh

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

42 assert "simple" in result.output 8 ctx1abcdefgh

43 assert "conv" in result.output 8 ctx1abcdefgh

44 assert "lstm" in result.output 8 ctx1abcdefgh

45 

46 

47def test_script(): 8 ctx1abcdefgh

48 result = subprocess.run( 8 ctx1abcdefgh

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

50 capture_output=True, 

51 encoding="utf-8", 

52 ) 

53 assert "Usage" in result.stdout 8 ctx1abcdefgh