Method test_constant_folding_unaryop

Lib/test/test_peepholer.py:278–321

(self)

                self.check_lnotab(code)

    def test_constant_folding_unaryop(self):
        intrinsic_positive = 5
        tests = [
            ('-0', 'UNARY_NEGATIVE', None, True, 'LOAD_SMALL_INT', 0),
            ('-0.0', 'UNARY_NEGATIVE', None, True, 'LOAD_CONST', -0.0),
            ('-(1.0-1.0)', 'UNARY_NEGATIVE', None, True, 'LOAD_CONST', -0.0),
            ('-0.5', 'UNARY_NEGATIVE', None, True, 'LOAD_CONST', -0.5),
            ('---1', 'UNARY_NEGATIVE', None, True, 'LOAD_CONST', -1),
            ('---""', 'UNARY_NEGATIVE', None, False, None, None),
            ('~~~1', 'UNARY_INVERT', None, True, 'LOAD_CONST', -2),
            ('~~~""', 'UNARY_INVERT', None, False, None, None),
            ('not not True', 'UNARY_NOT', None, True, 'LOAD_CONST', True),
            ('not not x', 'UNARY_NOT', None, True, 'LOAD_NAME', 'x'),  # this should be optimized regardless of constant or not
            ('+++1', 'CALL_INTRINSIC_1', intrinsic_positive, True, 'LOAD_SMALL_INT', 1),
            ('---x', 'UNARY_NEGATIVE', None, False, None, None),
            ('~~~x', 'UNARY_INVERT', None, False, None, None),
            ('+++x', 'CALL_INTRINSIC_1', intrinsic_positive, False, None, None),
            ('~True', 'UNARY_INVERT', None, False, None, None),
        ]

        for (
            expr,
            original_opcode,
            original_argval,
            is_optimized,
            optimized_opcode,
            optimized_argval,
        ) in tests:
            with self.subTest(expr=expr, is_optimized=is_optimized):
                code = compile(expr, "", "single")
                if is_optimized:
                    self.assertNotInBytecode(code, original_opcode, argval=original_argval)
                    self.assertInBytecode(code, optimized_opcode, argval=optimized_argval)
                else:
                    self.assertInBytecode(code, original_opcode, argval=original_argval)
                self.check_lnotab(code)

        # Check that -0.0 works after marshaling
        def negzero():
            return -(1.0-1.0)

        for instr in dis.get_instructions(negzero):
            self.assertNotStartsWith(instr.opname, 'UNARY_')
        self.check_lnotab(negzero)

    def test_constant_folding_binop(self):
        tests = [