Coverage for tests / test_tutorial / test_parameter_types / test_enum / test_tutorial004.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("tutorial004_py310"), 

16 pytest.param("tutorial004_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_help(mod: ModuleType): 7 ctx1abcdefg

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

27 assert result.exit_code == 0 7 ctx1abcdefg

28 assert "--network [simple|conv|lstm]" in result.output.replace(" ", "") 7 ctx1abcdefg

29 

30 

31def test_main(mod): 7 ctx1abcdefg

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

33 assert result.exit_code == 0 7 ctx1abcdefg

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

35 

36 

37def test_invalid(mod: ModuleType): 7 ctx1abcdefg

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

39 assert result.exit_code != 0 7 ctx1abcdefg

40 assert "Invalid value for '--network'" in result.output 7 ctx1abcdefg

41 assert ( 7 ctx1abcdefg

42 "invalid choice: capsule. (choose from" in result.output 

43 or "'capsule' is not one of" in result.output 

44 ) 

45 assert "simple" in result.output 7 ctx1abcdefg

46 assert "conv" in result.output 7 ctx1abcdefg

47 assert "lstm" in result.output 7 ctx1abcdefg

48 

49 

50def test_script(mod: ModuleType): 7 ctx1abcdefg

51 result = subprocess.run( 7 ctx1abcdefg

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

53 capture_output=True, 

54 encoding="utf-8", 

55 ) 

56 assert "Usage" in result.stdout 7 ctx1abcdefg