Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/black/numerics.py: 100%

1""" 

2Formatting numeric literals. 

3""" 

4from blib2to3.pytree import Leaf 

5 

6 

7def format_hex(text: str) -> str: 

8 """ 

9 Formats a hexadecimal string like "0x12B3" 

10 """ 

11 before, after = text[:2], text[2:] 

12 return f"{before}{after.upper()}" 

13 

14 

15def format_scientific_notation(text: str) -> str: 

16 """Formats a numeric string utilizing scentific notation""" 

17 before, after = text.split("e") 

18 sign = "" 

19 if after.startswith("-"): 

20 after = after[1:] 

21 sign = "-" 

22 elif after.startswith("+"): 

23 after = after[1:] 

24 before = format_float_or_int_string(before) 

25 return f"{before}e{sign}{after}" 

26 

27 

28def format_complex_number(text: str) -> str: 

29 """Formats a complex string like `10j`""" 

30 number = text[:-1] 

31 suffix = text[-1] 

32 return f"{format_float_or_int_string(number)}{suffix}" 

33 

34 

35def format_float_or_int_string(text: str) -> str: 

36 """Formats a float string like "1.0".""" 

37 if "." not in text: 

38 return text 

39 

40 before, after = text.split(".") 

41 return f"{before or 0}.{after or 0}" 

42 

43 

44def normalize_numeric_literal(leaf: Leaf) -> None: 

45 """Normalizes numeric (float, int, and complex) literals. 

46 

47 All letters used in the representation are normalized to lowercase.""" 

48 text = leaf.value.lower() 

49 if text.startswith(("0o", "0b")): 

50 # Leave octal and binary literals alone. 

51 pass 

52 elif text.startswith("0x"): 

53 text = format_hex(text) 

54 elif "e" in text: 

55 text = format_scientific_notation(text) 

56 elif text.endswith("j"): 

57 text = format_complex_number(text) 

58 else: 

59 text = format_float_or_int_string(text) 

60 leaf.value = text