cpython/Lib/test/test_stackrefs.py at 72ec2532fe8c31c05cdbc74ea1e21ff350a75239 · python/cpython · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
import unittest
import sys
from test.support import cpython_only
try:
    import _testinternalcapi
except ImportError:
    _testinternalcapi = None

@cpython_only
class TestDefinition(unittest.TestCase):

    BIG_REFCOUNT = 1000
    FLAGS_REFCOUNT = 1
    FLAGS_NO_REFCOUNT = 0
    FLAGS_INVALID = -1

    def test_equivalence(self):
        def run_with_refcount_check(self, func, obj):
            refcount = sys.getrefcount(obj)
            res = func(obj)
            self.assertEqual(sys.getrefcount(obj), refcount)
            return res

        funcs_with_incref = [
            _testinternalcapi.stackref_from_object_new,
            _testinternalcapi.stackref_from_object_steal_with_incref,
            _testinternalcapi.stackref_make_heap_safe,
            _testinternalcapi.stackref_make_heap_safe_with_borrow,
            _testinternalcapi.stackref_strong_reference,
        ]

        funcs_with_borrow = [
            _testinternalcapi.stackref_from_object_borrow,
            _testinternalcapi.stackref_dup_borrowed_with_close,
        ]

        immortal_objs = (None, True, False, 42, '1')

        for obj in immortal_objs:
            self.assertTrue(sys._is_immortal(obj))
            results = set()
            for func in funcs_with_incref + funcs_with_borrow:
                refcount, flags = run_with_refcount_check(self, func, obj)
                self.assertGreater(refcount, self.BIG_REFCOUNT)
                self.assertIn(flags, (self.FLAGS_REFCOUNT, self.FLAGS_INVALID))
                results.add((refcount, flags))
            self.assertEqual(len(results), 1)

        mortal_objs = (object(), range(10), iter([1, 2, 3]))

        for obj in mortal_objs:
            self.assertFalse(sys._is_immortal(obj))
            results = set()
            for func in funcs_with_incref:
                refcount, flags = run_with_refcount_check(self, func, obj)
                self.assertLess(refcount, self.BIG_REFCOUNT)
                self.assertEqual(flags, self.FLAGS_NO_REFCOUNT)
                results.add((refcount, flags))
            self.assertEqual(len(results), 1)

            results = set()
            for func in funcs_with_borrow:
                refcount, flags = run_with_refcount_check(self, func, obj)
                self.assertLess(refcount, self.BIG_REFCOUNT)
                self.assertEqual(flags, self.FLAGS_REFCOUNT)
                results.add((refcount, flags))
            self.assertEqual(len(results), 1)


if __name__ == "__main__":
    unittest.main()