Coverage for tests/test_tutorial/test_parameter_types/test_enum/test_tutorial002_an.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 tutorial002_an 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_upper(): 9 ctx1abcdefghi

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

17 assert result.exit_code == 0 9 ctx1abcdefghi

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

19 

20 

21def test_mix(): 9 ctx1abcdefghi

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

23 assert result.exit_code == 0 9 ctx1abcdefghi

24 assert "Training neural network of type: lstm" in result.output 9 ctx1abcdefghi

25 

26 

27def test_script(): 9 ctx1abcdefghi

28 result = subprocess.run( 9 ctx1abcdefghi

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

30 capture_output=True, 

31 encoding="utf-8", 

32 ) 

33 assert "Usage" in result.stdout 9 ctx1abcdefghi