File: test_jit_setup.py

package info (click to toggle)
pypy3 7.3.19%2Bdfsg-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, 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,585 bytes parent folder | download | duplicates (9)
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
class AppTestPyPyJIT:
    spaceconfig = dict(usemodules=('pypyjit',))

    def test_setup(self):
        # this just checks that the module is setting up things correctly, and
        # the resulting code makes sense on top of CPython.
        import pypyjit
        try:
            pypyjit.set_param(threshold=5, inlining=1)
            pypyjit.set_param("trace_eagerness=3,inlining=0")

            def f(x, y):
                return x*y+1

            assert f(6, 7) == 43

            def gen(x):
                i = 0
                while i < x:
                    yield i*i
                    i += 1

            assert list(gen(3)) == [0, 1, 4]
        finally:
            pypyjit.set_param('default')

    def test_no_jit(self):
        import pypyjit
        was_called = []
        def should_not_be_called(*args, **kwds):
            was_called.append((args, kwds))
        try:
            pypyjit.set_param('off')
            pypyjit.set_compile_hook(should_not_be_called)
            def f():
                pass
            for i in range(2500):
                f()
            assert not was_called
        finally:
            pypyjit.set_compile_hook(None)
            pypyjit.set_param('default')

    def test_doc(self):
        import pypyjit
        d = pypyjit.PARAMETER_DOCS
        assert type(d) is dict
        assert 'threshold' in d

    def test_interface_residual_call(self):
        import pypyjit
        def f(*args, **kwds):
            return (args, kwds)
        res = pypyjit.residual_call(f, 4, x=6)
        assert res == ((4,), {'x': 6})