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

1import importlib 7 ctx1abcdefg

2import subprocess 7 ctx1abcdefg

3import sys 7 ctx1abcdefg

4from types import ModuleType 7 ctx1abcdefg

5 

6import pytest 7 ctx1abcdefg

7from typer.testing import CliRunner 7 ctx1abcdefg

8 

9runner = CliRunner() 7 ctx1abcdefg

10 

11 

12@pytest.fixture( 7 ctx1abcdefg

13 name="mod", 

14 params=[ 

15 pytest.param("tutorial002_py310"), 

16 pytest.param("tutorial002_an_py310"), 

17 ], 

18) 

19def get_mod(request: pytest.FixtureRequest) -> ModuleType: 7 ctx1abcdefg

20 module_name = f"docs_src.parameter_types.enum.{request.param}" 7 ctx1abcdefg

21 mod = importlib.import_module(module_name) 7 ctx1abcdefg

22 return mod 7 ctx1abcdefg

23 

24 

25def test_upper(mod: ModuleType): 7 ctx1abcdefg

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

27 assert result.exit_code == 0 7 ctx1abcdefg

28 assert "Training neural network of type: conv" in result.output 7 ctx1abcdefg

29 

30 

31def test_mix(mod: ModuleType): 7 ctx1abcdefg

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

33 assert result.exit_code == 0 7 ctx1abcdefg

34 assert "Training neural network of type: lstm" in result.output 7 ctx1abcdefg

35 

36 

37def test_script(mod: ModuleType): 7 ctx1abcdefg

38 result = subprocess.run( 7 ctx1abcdefg

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

40 capture_output=True, 

41 encoding="utf-8", 

42 ) 

43 assert "Usage" in result.stdout 7 ctx1abcdefg