Coverage for tests/test_rich_markup_mode.py: 100%

1import typer 9 ctx1abcdefghi

2import typer.completion 9 ctx1abcdefghi

3from typer.testing import CliRunner 9 ctx1abcdefghi

4 

5runner = CliRunner() 9 ctx1abcdefghi

6rounded = ["╭", "─", "┬", "╮", "│", "├", "┼", "┤", "╰", "┴", "╯"] 9 ctx1abcdefghi

7 

8 

9def test_rich_markup_mode_none(): 9 ctx1abcdefghi

10 app = typer.Typer(rich_markup_mode=None) 9 ctx1abcdefghi

11 

12 @app.command() 9 ctx1abcdefghi

13 def main(arg: str): 9 ctx1abcdefghi

14 """Main function""" 

15 print(f"Hello {arg}") 9 ctx1abcdefghi

16 

17 assert app.rich_markup_mode is None 9 ctx1abcdefghi

18 

19 result = runner.invoke(app, ["World"]) 9 ctx1abcdefghi

20 assert "Hello World" in result.stdout 9 ctx1abcdefghi

21 

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

23 assert all(c not in result.stdout for c in rounded) 9 ctx1abcdefghi

24 

25 

26def test_rich_markup_mode_rich(): 9 ctx1abcdefghi

27 app = typer.Typer(rich_markup_mode="rich") 9 ctx1abcdefghi

28 

29 @app.command() 9 ctx1abcdefghi

30 def main(arg: str): 9 ctx1abcdefghi

31 """Main function""" 

32 print(f"Hello {arg}") 9 ctx1abcdefghi

33 

34 assert app.rich_markup_mode == "rich" 9 ctx1abcdefghi

35 

36 result = runner.invoke(app, ["World"]) 9 ctx1abcdefghi

37 assert "Hello World" in result.stdout 9 ctx1abcdefghi

38 

39 result = runner.invoke(app, ["--help"]) 9 ctx1abcdefghi

40 assert any(c in result.stdout for c in rounded) 9 ctx1abcdefghi