cpython/Lib/test/test_stackrefs.py at 474f650ced253ab97bd0666382ad2614a1d10ab0 · 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
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

    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:
            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, (1, -1))
                results.add((refcount, flags))
            self.assertEqual(len(results), 1)

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

        for obj in mortal_objs:
            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, 0)
                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, 1)
                results.add((refcount, flags))
            self.assertEqual(len(results), 1)


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