Coverage for tests / utils.py: 100%

1import re 8 ctx1abcdefgh

2import sys 8 ctx1abcdefgh

3from os import getenv 8 ctx1abcdefgh

4 

5import pytest 8 ctx1abcdefgh

6from typer._completion_shared import _get_shell_name 8 ctx1abcdefgh

7 

8needs_py310 = pytest.mark.skipif( 8 ctx1abcdefgh

9 sys.version_info < (3, 10), reason="requires python3.10+" 

10) 

11 

12needs_linux = pytest.mark.skipif( 8 ctx1abcdefgh

13 not sys.platform.startswith("linux"), reason="Test requires Linux" 

14) 

15 

16shell = _get_shell_name() 8 ctx1abcdefgh

17needs_bash = pytest.mark.skipif( 8 ctx1abcdefgh

18 shell is None or "bash" not in shell, reason="Test requires Bash" 

19) 

20 

21requires_completion_permission = pytest.mark.skipif( 8 ctx1abcdefgh

22 not getenv("_TYPER_RUN_INSTALL_COMPLETION_TESTS", False), 

23 reason="Test requires permission to run completion installation tests", 

24) 

25 

26 

27def strip_double_spaces(text: str) -> str: 8 ctx1abcdefgh

28 return re.sub(r" {2,}", " ", text) 8 ctx1abcdefgh

29 

30 

31def normalize_rich_output( 8 ctx1abcdefgh

32 text: str, replace_with: str = "*", squash_whitespaces: bool = True 

33) -> str: 

34 """ 

35 Replace all rich formatting characters with a simple placeholder. 

36 """ 

37 text = re.sub(r"\x1b\[[0-9;]*m", replace_with, text) 8 ctx1abcdefgh

38 text = re.sub(r"[\u2500-\u257F]", replace_with, text) 8 ctx1abcdefgh

39 text = re.sub(r"[\U0001F300-\U0001F6FF]", replace_with, text) 8 ctx1abcdefgh

40 text = re.sub(f"{re.escape(replace_with)}+", replace_with, text) 8 ctx1abcdefgh

41 if squash_whitespaces: 8 ctx1abcdefgh

42 text = strip_double_spaces(text) 8 ctx1abcdefgh

43 return text 8 ctx1abcdefgh