File: test_template.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 (18 lines) | stat: -rw-r--r-- 498 bytes parent folder | download | duplicates (9) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
 
from rpython.tool.sourcetools import compile_template


some_global = 5

def test_template():
    init_value = 50

    def template(n):
        args = ['arg%d' % i for i in range(n)]
        yield     'def add(%s):' % (', '.join(args),)
        yield     '    total = init_value + some_global'
        for i in range(n):
            yield '    total += arg%d' % i
        yield     '    return total'

    add = compile_template(template(5), 'add')
    assert add(10000, 1000, 100, 10, 1) == 11166