Add parentheses around numeric literals, to avoid turning 3 .bit_length() into 3.bit_length().

This commit is contained in:
Mark Dickinson 2010-06-29 08:52:36 +00:00
parent 82c8d93357
commit 3eb0290346
2 changed files with 14 additions and 10 deletions

View File

@ -84,6 +84,9 @@ class UnparseTestCase(unittest.TestCase):
self.check_roundtrip("not True or False")
self.check_roundtrip("True or not False")
def test_integer_parens(self):
self.check_roundtrip("3 .__abs__()")
def test_chained_comparisons(self):
self.check_roundtrip("1 < 4 <= 5")
self.check_roundtrip("a is b is c is not d")

View File

@ -302,16 +302,17 @@ class Unparser:
self.write("`")
def _Num(self, t):
# There are no negative numeric literals in Python; however,
# some optimizations produce a negative Num in the AST. Add
# parentheses to avoid turning (-1)**2 into -1**2.
strnum = repr(t.n)
if strnum.startswith("-"):
self.write("(")
self.write(strnum)
self.write(")")
else:
self.write(strnum)
# Add parentheses around numeric literals to avoid:
#
# (1) turning (-1)**2 into -1**2, and
# (2) turning 3 .__abs__() into 3.__abs__()
#
# For (1), note that Python doesn't actually have negative
# numeric literals, but (at least in Python 2.x) there's a CST
# transformation that can produce negative Nums in the AST.
self.write("(")
self.write(repr(t.n))
self.write(")")
def _List(self, t):
self.write("[")