Merge 74b7b39 into 3905173

psf · Jan 24, 2022 · caae8eb · caae8eb
2 parents 3905173 + 74b7b39
commit caae8eb
Show file tree

Hide file tree

Showing 7 changed files with 48 additions and 8 deletions.
diff --git a/CHANGES.md b/CHANGES.md
@@ -38,6 +38,7 @@
   `--preview` (#2789)
 - Enable Python 3.10+ by default, without any extra need to specify
   `--target-version=py310`. (#2758)
+- Use parenthesis on method access on number integerals (float and int) (#2799)
 
 ### Packaging
 

diff --git a/src/black/linegen.py b/src/black/linegen.py
@@ -197,6 +197,14 @@ def visit_decorators(self, node: Node) -> Iterator[Line]:
             yield from self.line()
             yield from self.visit(child)
 
+    def visit_power(self, node: Node) -> Iterator[Line]:
+        for idx, leaf in enumerate(node.children[:-1]):
+            next_leaf = node.children[idx + 1]
+            if leaf.type == token.NUMBER and next_leaf.type == syms.trailer:
+                wrap_in_parentheses(node, leaf)
+
+        yield from self.visit_default(node)
+
     def visit_SEMI(self, leaf: Leaf) -> Iterator[Line]:
         """Remove a semicolon and put the other statement on a separate line."""
         yield from self.line()

diff --git a/tests/data/expression.diff b/tests/data/expression.diff
@@ -11,7 +11,7 @@
  True
  False
  1
-@@ -29,63 +29,96 @@
+@@ -29,91 +29,127 @@
  ~great
  +value
  -1
@@ -130,8 +130,11 @@
  call(**self.screen_kwargs)
  call(b, **self.screen_kwargs)
  lukasz.langa.pl
-@@ -94,26 +127,29 @@
- 1.0 .real
+ call.me(maybe)
+-1 .real
+-1.0 .real
++(1).real
++(1.0).real
  ....__class__
  list[str]
  dict[str, int]

diff --git a/tests/data/expression.py b/tests/data/expression.py
@@ -382,8 +382,8 @@ async def f():
 call(b, **self.screen_kwargs)
 lukasz.langa.pl
 call.me(maybe)
-1 .real
-1.0 .real
+(1).real
+(1.0).real
 ....__class__
 list[str]
 dict[str, int]

diff --git a/tests/data/expression_skip_magic_trailing_comma.diff b/tests/data/expression_skip_magic_trailing_comma.diff
@@ -11,7 +11,7 @@
  True
  False
  1
-@@ -29,63 +29,84 @@
+@@ -29,91 +29,110 @@
  ~great
  +value
  -1
@@ -118,8 +118,11 @@
  call(**self.screen_kwargs)
  call(b, **self.screen_kwargs)
  lukasz.langa.pl
-@@ -94,26 +115,24 @@
- 1.0 .real
+ call.me(maybe)
+-1 .real
+-1.0 .real
++(1).real
++(1.0).real
  ....__class__
  list[str]
  dict[str, int]

diff --git a/tests/data/use_braces_in_int_method_calls.py b/tests/data/use_braces_in_int_method_calls.py
@@ -0,0 +1,24 @@
+count = 5 .bit_count()
+abs = 10 .__abs__()
+is_integer = 10.5 .is_integer()
+
+if 10 .real: ...
+
+for number in range(1, 10):
+    if number.imag: ...
+
+
+# output
+count = (5).bit_count()
+abs = (10).__abs__()
+is_integer = (10.5).is_integer()
+
+if (10).real:
+    ...
+
+for number in range(1, 10):
+    if number.imag:
+        ...
+
+
+
diff --git a/tests/test_format.py b/tests/test_format.py
@@ -53,6 +53,7 @@
     "string_prefixes",
     "tricky_unicode_symbols",
     "tupleassign",
+    "use_braces_in_int_method_calls",
 ]
 
 PY310_CASES: List[str] = [