JezK
Edit File: test_weakref.py
import gc import weakref import unittest import greenlet from . import TestCase class WeakRefTests(TestCase): def test_dead_weakref(self): def _dead_greenlet(): g = greenlet.greenlet(lambda: None) g.switch() return g o = weakref.ref(_dead_greenlet()) gc.collect() self.assertEqual(o(), None) def test_inactive_weakref(self): o = weakref.ref(greenlet.greenlet()) gc.collect() self.assertEqual(o(), None) def test_dealloc_weakref(self): seen = [] def worker(): try: greenlet.getcurrent().parent.switch() finally: seen.append(g()) g = greenlet.greenlet(worker) g.switch() g2 = greenlet.greenlet(lambda: None, g) g = weakref.ref(g2) g2 = None self.assertEqual(seen, [None])