Method test_constant_folding_binop

Lib/test/test_peepholer.py:323–434

(self)

        self.check_lnotab(negzero)

    def test_constant_folding_binop(self):
        tests = [
            ('1 + 2', 'NB_ADD', True, 'LOAD_SMALL_INT', 3),
            ('1 + 2 + 3', 'NB_ADD', True, 'LOAD_SMALL_INT', 6),
            ('1 + ""', 'NB_ADD', False, None, None),
            ('1 - 2', 'NB_SUBTRACT', True, 'LOAD_CONST', -1),
            ('1 - 2 - 3', 'NB_SUBTRACT', True, 'LOAD_CONST', -4),
            ('1 - ""', 'NB_SUBTRACT', False, None, None),
            ('2 * 2', 'NB_MULTIPLY', True, 'LOAD_SMALL_INT', 4),
            ('2 * 2 * 2', 'NB_MULTIPLY', True, 'LOAD_SMALL_INT', 8),
            ('2 / 2', 'NB_TRUE_DIVIDE', True, 'LOAD_CONST', 1.0),
            ('2 / 2 / 2', 'NB_TRUE_DIVIDE', True, 'LOAD_CONST', 0.5),
            ('2 / ""', 'NB_TRUE_DIVIDE', False, None, None),
            ('2 // 2', 'NB_FLOOR_DIVIDE', True, 'LOAD_SMALL_INT', 1),
            ('2 // 2 // 2', 'NB_FLOOR_DIVIDE', True, 'LOAD_SMALL_INT', 0),
            ('2 // ""', 'NB_FLOOR_DIVIDE', False, None, None),
            ('2 % 2', 'NB_REMAINDER', True, 'LOAD_SMALL_INT', 0),
            ('2 % 2 % 2', 'NB_REMAINDER', True, 'LOAD_SMALL_INT', 0),
            ('2 % ()', 'NB_REMAINDER', False, None, None),
            ('2 ** 2', 'NB_POWER', True, 'LOAD_SMALL_INT', 4),
            ('2 ** 2 ** 2', 'NB_POWER', True, 'LOAD_SMALL_INT', 16),
            ('2 ** ""', 'NB_POWER', False, None, None),
            ('2 << 2', 'NB_LSHIFT', True, 'LOAD_SMALL_INT', 8),
            ('2 << 2 << 2', 'NB_LSHIFT', True, 'LOAD_SMALL_INT', 32),
            ('2 << ""', 'NB_LSHIFT', False, None, None),
            ('2 >> 2', 'NB_RSHIFT', True, 'LOAD_SMALL_INT', 0),
            ('2 >> 2 >> 2', 'NB_RSHIFT', True, 'LOAD_SMALL_INT', 0),
            ('2 >> ""', 'NB_RSHIFT', False, None, None),
            ('2 | 2', 'NB_OR', True, 'LOAD_SMALL_INT', 2),
            ('2 | 2 | 2', 'NB_OR', True, 'LOAD_SMALL_INT', 2),
            ('2 | ""', 'NB_OR', False, None, None),
            ('2 & 2', 'NB_AND', True, 'LOAD_SMALL_INT', 2),
            ('2 & 2 & 2', 'NB_AND', True, 'LOAD_SMALL_INT', 2),
            ('2 & ""', 'NB_AND', False, None, None),
            ('2 ^ 2', 'NB_XOR', True, 'LOAD_SMALL_INT', 0),
            ('2 ^ 2 ^ 2', 'NB_XOR', True, 'LOAD_SMALL_INT', 2),
            ('2 ^ ""', 'NB_XOR', False, None, None),
            ('(1, )[0]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 1),
            ('(1, )[-1]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 1),
            ('(1 + 2, )[0]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 3),
            ('(1, (1, 2))[1][1]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 2),
            ('(1, 2)[2-1]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 2),
            ('(1, (1, 2))[1][2-1]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 2),
            ('(1, (1, 2))[1:6][0][2-1]', 'NB_SUBSCR', True, 'LOAD_SMALL_INT', 2),
            ('"a"[0]', 'NB_SUBSCR', True, 'LOAD_CONST', 'a'),
            ('("a" + "b")[1]', 'NB_SUBSCR', True, 'LOAD_CONST', 'b'),
            ('("a" + "b", )[0][1]', 'NB_SUBSCR', True, 'LOAD_CONST', 'b'),
            ('("a" * 10)[9]', 'NB_SUBSCR', True, 'LOAD_CONST', 'a'),
            ('(1, )[1]', 'NB_SUBSCR', False, None, None),
            ('(1, )[-2]', 'NB_SUBSCR', False, None, None),
            ('"a"[1]', 'NB_SUBSCR', False, None, None),
            ('"a"[-2]', 'NB_SUBSCR', False, None, None),
            ('("a" + "b")[2]', 'NB_SUBSCR', False, None, None),
            ('("a" + "b", )[0][2]', 'NB_SUBSCR', False, None, None),
            ('("a" + "b", )[1][0]', 'NB_SUBSCR', False, None, None),
            ('("a" * 10)[10]', 'NB_SUBSCR', False, None, None),
            ('(1, (1, 2))[2:6][0][2-1]', 'NB_SUBSCR', False, None, None),
        ]