Apr 8, 2025 · Apr 8, 2025 · Apr 8, 2025 · Apr 8, 2025
diff --git a/Lib/test/test_capi/test_opt.py b/Lib/test/test_capi/test_opt.py
        self.assertIn("_CONTAINS_OP_SET", uops)
        self.assertNotIn("_TO_BOOL_BOOL", uops)

    def test_to_bool_bool_contains_op_dict(self):
        """
        Test that _TO_BOOL_BOOL is removed from code like:

        res = foo in some_dict
        if res:
            ....

        """
        def testfunc(n):
            x = 0
            s = {1: 1, 2: 2, 3: 3}
            for _ in range(n):
                a = 2
                in_dict = a in s
                if in_dict:
                    x += 1
            return x

        res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
        self.assertEqual(res, TIER2_THRESHOLD)
        self.assertIsNotNone(ex)
        uops = get_opnames(ex)
        self.assertIn("_CONTAINS_OP_DICT", uops)
        self.assertNotIn("_TO_BOOL_BOOL", uops)


    def test_remove_guard_for_known_type_str(self):
        def f(n):
            for i in range(n):
diff --git a/Misc/NEWS.d/next/Core_and_Builtins/2025-04-08-21-20-12.gh-issue-131798.Ft9tIF.rst b/Misc/NEWS.d/next/Core_and_Builtins/2025-04-08-21-20-12.gh-issue-131798.Ft9tIF.rst
 Allow the JIT to remove an extra ``_TO_BOOL_BOOL`` instruction after
 ``_CONTAINS_OP_DICT`` by setting the return type to bool.
diff --git a/Python/optimizer_bytecodes.c b/Python/optimizer_bytecodes.c
        res = sym_new_type(ctx, &PyBool_Type);
    }

    op(_CONTAINS_OP_DICT, (left, right -- res)) {
        res = sym_new_type(ctx, &PyBool_Type);
    }

    op(_LOAD_CONST, (-- value)) {
        PyObject *val = PyTuple_GET_ITEM(co->co_consts, this_instr->oparg);
        int opcode = _Py_IsImmortal(val) ? _LOAD_CONST_INLINE_BORROW : _LOAD_CONST_INLINE;
diff --git a/Python/optimizer_cases.c.h b/Python/optimizer_cases.c.h
Original file line number	Diff line number	Diff line change
Expand Up		@@ -1629,6 +1629,33 @@ def testfunc(n):
		self.assertIn("_CONTAINS_OP_SET", uops)
		self.assertNotIn("_TO_BOOL_BOOL", uops)

		def test_to_bool_bool_contains_op_dict(self):
		"""
		Test that _TO_BOOL_BOOL is removed from code like:

		res = foo in some_dict
		if res:
		....

		"""
		def testfunc(n):
		x = 0
		s = {1: 1, 2: 2, 3: 3}
		for _ in range(n):
		a = 2
		in_dict = a in s
		if in_dict:
		x += 1
		return x

		res, ex = self._run_with_optimizer(testfunc, TIER2_THRESHOLD)
		self.assertEqual(res, TIER2_THRESHOLD)
		self.assertIsNotNone(ex)
		uops = get_opnames(ex)
		self.assertIn("_CONTAINS_OP_DICT", uops)
		self.assertNotIn("_TO_BOOL_BOOL", uops)


		def test_remove_guard_for_known_type_str(self):
		def f(n):
		for i in range(n):
Expand Down
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		Allow the JIT to remove an extra ``_TO_BOOL_BOOL`` instruction after
		``_CONTAINS_OP_DICT`` by setting the return type to bool.
Original file line number	Diff line number	Diff line change
Expand Up		@@ -481,6 +481,10 @@ dummy_func(void) {
		res = sym_new_type(ctx, &PyBool_Type);
		}

		op(_CONTAINS_OP_DICT, (left, right -- res)) {
		res = sym_new_type(ctx, &PyBool_Type);
		}

		op(_LOAD_CONST, (-- value)) {
		PyObject *val = PyTuple_GET_ITEM(co->co_consts, this_instr->oparg);
		int opcode = _Py_IsImmortal(val) ? _LOAD_CONST_INLINE_BORROW : _LOAD_CONST_INLINE;
Expand Down