summaryrefslogtreecommitdiffstats
path: root/Lib/test/test_peepholer.py
diff options
context:
space:
mode:
Diffstat (limited to 'Lib/test/test_peepholer.py')
-rw-r--r--Lib/test/test_peepholer.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/Lib/test/test_peepholer.py b/Lib/test/test_peepholer.py
index 5696500..d1d5874 100644
--- a/Lib/test/test_peepholer.py
+++ b/Lib/test/test_peepholer.py
@@ -146,6 +146,24 @@ class TestTranforms(unittest.TestCase):
asm = dis_single('a="x"*1000')
self.assertTrue('(1000)' in asm)
+ def test_binary_subscr_on_unicode(self):
+ # valid code get optimized
+ asm = dis_single('"foo"[0]')
+ self.assertIn("('f')", asm)
+ self.assertNotIn('BINARY_SUBSCR', asm)
+ asm = dis_single('"\u0061\uffff"[1]')
+ self.assertIn("('\\uffff')", asm)
+ self.assertNotIn('BINARY_SUBSCR', asm)
+
+ # invalid code doesn't get optimized
+ # out of range
+ asm = dis_single('"fuu"[10]')
+ self.assertIn('BINARY_SUBSCR', asm)
+ # non-BMP char (see #5057)
+ asm = dis_single('"\U00012345"[0]')
+ self.assertIn('BINARY_SUBSCR', asm)
+
+
def test_folding_of_unaryops_on_constants(self):
for line, elem in (
('-0.5', '(-0.5)'), # unary negative