Don't omit whitespace when the factor is not a math operator

[etc/vim.git] / black.py

diff --git a/black.py b/black.py
index 1c07de713fde713f5814275cc0a6e07f8c18db26..0dd763073622d2ec1d0234b46cf5cd4e42bfe607 100644 (file)
--- a/black.py
+++ b/black.py
@@ -340,6 +340,7 @@ MATH_OPERATORS = {
     token.AMPER,
     token.PERCENT,
     token.CIRCUMFLEX,
     token.AMPER,
     token.PERCENT,
     token.CIRCUMFLEX,

+    token.TILDE,

     token.LEFTSHIFT,
     token.RIGHTSHIFT,
     token.DOUBLESTAR,
     token.LEFTSHIFT,
     token.RIGHTSHIFT,
     token.DOUBLESTAR,


@@ -888,7 +889,11 @@ def whitespace(leaf: Leaf) -> str:  # noqa C901
             if prevp.parent and prevp.parent.type in {syms.subscript, syms.sliceop}:
                 return NO
 
             if prevp.parent and prevp.parent.type in {syms.subscript, syms.sliceop}:
                 return NO
 

-        elif prevp.parent and prevp.parent.type in {syms.factor, syms.star_expr}:
+        elif (
+            prevp.parent
+            and prevp.parent.type in {syms.factor, syms.star_expr}
+            and prevp.type in MATH_OPERATORS
+        ):

             return NO
 
     elif prev.type in OPENING_BRACKETS:
             return NO
 
     elif prev.type in OPENING_BRACKETS: