File: test_weakref.py

package info (click to toggle)
pypy3 7.3.19%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 212,236 kB
  • sloc: python: 2,098,316; ansic: 540,565; sh: 21,462; asm: 14,419; cpp: 4,451; makefile: 4,209; objc: 761; xml: 530; exp: 499; javascript: 314; pascal: 244; lisp: 45; csh: 12; awk: 4
file content (55 lines) | stat: -rw-r--r-- 1,702 bytes parent folder | download | duplicates (2)
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
import os
import sys
import textwrap
import subprocess

def test_WeakValueDictionary_len(tmpdir):
    src = textwrap.dedent("""
        from weakref import WeakValueDictionary
        class Foo:
            pass
        N = 1000
        D = WeakValueDictionary()
        for i in range(N):
            D[i] = Foo()

        for i in range(10):
            x = len(D)
        print('OK')
    """)
    testfile = tmpdir.join('testfile.py')
    testfile.write(src)
    #
    # by setting a very small PYPY_GC_NURSERY value, we force running a minor
    # collection inside WeakValueDictionary.__len__. We just check that the
    # snippet above completes correctly, instead of raising "dictionary
    # changed size during iteration"
    env = {'PYPY_GC_NURSERY': '1k'}
    env['PATH'] = os.environ.get("PATH")
    subprocess.run([sys.executable, str(testfile)], env=env, check=True)


def test_WeakKeyDictionary_len(tmpdir):
    src = textwrap.dedent("""
        from weakref import WeakKeyDictionary
        class Foo:
            pass
        N = 1000
        D = WeakKeyDictionary()
        for i in range(N):
            D[Foo()] = i

        for i in range(10):
            x = len(D)
        print('OK')
    """)
    testfile = tmpdir.join('testfile.py')
    testfile.write(src)
    #
    # by setting a very small PYPY_GC_NURSERY value, we force running a minor
    # collection inside WeakValueDictionary.__len__. We just check that the
    # snippet above completes correctly, instead of raising "dictionary
    # changed size during iteration"
    env = {'PYPY_GC_NURSERY': '1k'}
    env['PATH'] = os.environ.get("PATH")
    subprocess.run([sys.executable, str(testfile)], env=env, check=True)