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

1import importlib 8 ctx1abcdefgh

2import subprocess 8 ctx1abcdefgh

3import sys 8 ctx1abcdefgh

4from types import ModuleType 8 ctx1abcdefgh

5 

6import pytest 8 ctx1abcdefgh

7from typer.testing import CliRunner 8 ctx1abcdefgh

8 

9runner = CliRunner() 8 ctx1abcdefgh

10 

11 

12@pytest.fixture( 8 ctx1abcdefgh

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial002_py39"), 

16 pytest.param("tutorial002_an_py39"), 

17 ], 

18) 

19def get_mod(request: pytest.FixtureRequest) -> ModuleType: 8 ctx1abcdefgh

20 module_name = f"docs_src.parameter_types.enum.{request.param}" 8 ctx1abcdefgh

21 mod = importlib.import_module(module_name) 8 ctx1abcdefgh

22 return mod 8 ctx1abcdefgh

23 

24 

25def test_upper(mod: ModuleType): 8 ctx1abcdefgh

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

27 assert result.exit_code == 0 8 ctx1abcdefgh

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

29 

30 

31def test_mix(mod: ModuleType): 8 ctx1abcdefgh

32 result = runner.invoke(mod.app, ["--network", "LsTm"]) 8 ctx1abcdefgh

33 assert result.exit_code == 0 8 ctx1abcdefgh

34 assert "Training neural network of type: lstm" in result.output 8 ctx1abcdefgh

35 

36 

37def test_script(mod: ModuleType): 8 ctx1abcdefgh

38 result = subprocess.run( 8 ctx1abcdefgh

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

40 capture_output=True, 

41 encoding="utf-8", 

42 ) 

43 assert "Usage" in result.stdout 8 ctx1abcdefgh