May 23, 2025 · May 23, 2025 · May 23, 2025 · May 23, 2025 · May 23, 2025 · Jun 3, 2025
diff --git a/mypyc/primitives/registry.py b/mypyc/primitives/registry.py
 import mypyc.primitives.list_ops
 import mypyc.primitives.misc_ops
 import mypyc.primitives.str_ops
 import mypyc.primitives.tuple_ops  # noqa: F401
 import mypyc.primitives.tuple_ops
 import mypyc.primitives.weakref_ops  # noqa: F401
diff --git a/mypyc/primitives/weakref_ops.py b/mypyc/primitives/weakref_ops.py
 from mypyc.ir.ops import ERR_MAGIC
 from mypyc.ir.rtypes import object_rprimitive
 from mypyc.primitives.registry import function_op

 # Weakref operations

 new_ref_op = function_op(
    name="weakref.ReferenceType",
    arg_types=[object_rprimitive, object_rprimitive],
    return_type=object_rprimitive,
    c_function_name="PyWeakref_NewRef",
    error_kind=ERR_MAGIC,
 )
diff --git a/mypyc/test-data/irbuild-weakref.test b/mypyc/test-data/irbuild-weakref.test
 [case testWeakrefRef]
 import weakref
 from typing import Any, Callable
 def f(x: object) -> object:
    return weakref.ref(x)

 [out]
 def f(x):
    x, r0 :: object
 L0:
    r0 = PyWeakref_NewRef(x, NULL)
    return r0

 [case testWeakrefRefCallback]
 import weakref
 from typing import Any, Callable
 def f(x: object, cb: Callable[[object], Any]) -> object:
    return weakref.ref(x, cb)

 [out]
 def f(x, cb):
    x, cb, r0 :: object
 L0:
    r0 = PyWeakref_NewRef(x, cb)
    return r0

 [case testFromWeakrefRef]
 from typing import Any, Callable
 from weakref import ref
 def f(x: object) -> object:
    return ref(x)

 [out]
 def f(x):
    x, r0 :: object
 L0:
    r0 = PyWeakref_NewRef(x, NULL)
    return r0

 [case testFromWeakrefRefCallback]
 from typing import Any, Callable
 from weakref import ref
 def f(x: object, cb: Callable[[object], Any]) -> object:
    return ref(x, cb)

 [out]
 def f(x, cb):
    x, cb, r0 :: object
 L0:
    r0 = PyWeakref_NewRef(x, cb)
    return r0
diff --git a/mypyc/test-data/run-weakref.test b/mypyc/test-data/run-weakref.test
 # Test cases for weakrefs (compile and run)

 [case testWeakrefRef]
 import asyncio
 import pytest  # type: ignore [import-not-found]
 from weakref import ref

 def test_weakref_ref():
    # some random weakreffable object
    obj = asyncio.new_event_loop()
    r = ref(obj)
    assert r() is obj
    obj = None
    assert r() is None, r()

 def test_weakref_ref_with_callback():
    # some random weakreffable object
    obj = asyncio.new_event_loop()
    r = ref(obj, lambda x: x)
    assert r() is obj
    obj = None
    assert r() is None, r()
diff --git a/mypyc/test/test_irbuild.py b/mypyc/test/test_irbuild.py
    "irbuild-constant-fold.test",
    "irbuild-glue-methods.test",
    "irbuild-math.test",
    "irbuild-weakref.test",
 ]

 if sys.version_info >= (3, 10):
diff --git a/mypyc/test/test_run.py b/mypyc/test/test_run.py
    "run-dunders-special.test",
    "run-singledispatch.test",
    "run-attrs.test",
    "run-weakref.test",
    "run-python37.test",
    "run-python38.test",
 ]
Original file line number	Diff line number	Diff line change
Expand Up		@@ -371,4 +371,5 @@ def load_address_op(name: str, type: RType, src: str) -> LoadAddressDescription:
		import mypyc.primitives.list_ops
		import mypyc.primitives.misc_ops
		import mypyc.primitives.str_ops
		import mypyc.primitives.tuple_ops # noqa: F401
		import mypyc.primitives.tuple_ops
		import mypyc.primitives.weakref_ops # noqa: F401
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,13 @@
		from mypyc.ir.ops import ERR_MAGIC
		from mypyc.ir.rtypes import object_rprimitive
		from mypyc.primitives.registry import function_op

		# Weakref operations

		new_ref_op = function_op(
		name="weakref.ReferenceType",
		arg_types=[object_rprimitive, object_rprimitive],
		return_type=object_rprimitive,
		c_function_name="PyWeakref_NewRef",
		error_kind=ERR_MAGIC,
		)
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,51 @@
		[case testWeakrefRef]
		import weakref
		from typing import Any, Callable
		def f(x: object) -> object:
		return weakref.ref(x)

		[out]
		def f(x):
		x, r0 :: object
		L0:
		r0 = PyWeakref_NewRef(x, NULL)
		return r0

		[case testWeakrefRefCallback]
		import weakref
		from typing import Any, Callable
		def f(x: object, cb: Callable[[object], Any]) -> object:
		return weakref.ref(x, cb)

		[out]
		def f(x, cb):
		x, cb, r0 :: object
		L0:
		r0 = PyWeakref_NewRef(x, cb)
		return r0

		[case testFromWeakrefRef]
		from typing import Any, Callable
		from weakref import ref
		def f(x: object) -> object:
		return ref(x)

		[out]
		def f(x):
		x, r0 :: object
		L0:
		r0 = PyWeakref_NewRef(x, NULL)
		return r0

		[case testFromWeakrefRefCallback]
		from typing import Any, Callable
		from weakref import ref
		def f(x: object, cb: Callable[[object], Any]) -> object:
		return ref(x, cb)

		[out]
		def f(x, cb):
		x, cb, r0 :: object
		L0:
		r0 = PyWeakref_NewRef(x, cb)
		return r0
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,22 @@
		# Test cases for weakrefs (compile and run)

		[case testWeakrefRef]
		import asyncio
		import pytest # type: ignore [import-not-found]
		from weakref import ref

		def test_weakref_ref():
		# some random weakreffable object
		obj = asyncio.new_event_loop()
		r = ref(obj)
		assert r() is obj
		obj = None
		assert r() is None, r()

		def test_weakref_ref_with_callback():
		# some random weakreffable object
		obj = asyncio.new_event_loop()
		r = ref(obj, lambda x: x)
		assert r() is obj
		obj = None
		assert r() is None, r()
Original file line number	Diff line number	Diff line change
Expand Up		@@ -53,6 +53,7 @@
		"irbuild-constant-fold.test",
		"irbuild-glue-methods.test",
		"irbuild-math.test",
		"irbuild-weakref.test",
		]

		if sys.version_info >= (3, 10):
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -69,6 +69,7 @@
		"run-dunders-special.test",
		"run-singledispatch.test",
		"run-attrs.test",
		"run-weakref.test",
		"run-python37.test",
		"run-python38.test",
		]
Expand Down