summaryrefslogtreecommitdiffstatshomepage
path: root/tests/basics/fun_code_full.py
blob: 5eb23150df09cf5fe625fceaf033fa3149c28eb0 (plain) (blame)
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
# Test function.__code__ attributes not available with MICROPY_PY_BUILTINS_CODE <= MICROPY_PY_BUILTINS_CODE_BASIC

try:
    (lambda: 0).__code__.co_code
except AttributeError:
    print("SKIP")
    raise SystemExit

try:
    import warnings
    warnings.simplefilter("ignore") # ignore deprecation warning about co_lnotab
except ImportError:
    pass

def f(x, y):
    a = x + y
    b = x - y
    return a * b

code = f.__code__

print(type(code.co_code)) # both bytes (but mpy and cpy have different instruction sets)
print(code.co_consts) # (not necessarily the same set, but in this function they are)
print(code.co_filename.rsplit('/')[-1]) # same terminal filename but might be different paths on other ports
print(type(code.co_firstlineno)) # both ints (but mpy points to first line inside, cpy points to declaration)
print(code.co_name)
print(iter(code.co_names) is not None) # both iterable (but mpy returns dict with names as keys, cpy only the names; and not necessarily the same set)
print(type(code.co_lnotab)) # both bytes

co_lines = code.co_lines()

l = list(co_lines)
first_start = l[0][0]
last_end = l[-1][1]
print(first_start) # co_lines should start at the start of the bytecode
print(len(code.co_code) - last_end) # and end at the end of the bytecode

prev_end = 0
for start, end, line_no in l:
    if end != prev_end:
        print("non-contiguous")
        break # the offset ranges should be contiguous
    prev_end = end
else:
    print("contiguous")