summaryrefslogtreecommitdiff
path: root/tests/micropython/viper_ptr32_store_boundary_intbig.py
blob: b44f31b00af4d2e5044718fca69d555fc1836d31 (plain)
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
# Test boundary conditions for various architectures

SET_TEMPLATE = """
@micropython.viper
def set{off}(dest: ptr32):
    saved = dest
    dest[{off}] = {val}
    assert int(saved) == int(dest)
set{off}(buffer)
print(hex(get_index(buffer, {off})))
"""

BIT_THRESHOLDS = (5, 8, 11, 12)
SIZE = 4
MASK = (1 << (8 * SIZE)) - 1


@micropython.viper
def set_index(dest: ptr32, i: int, val: uint):
    saved = dest
    dest[i] = val
    assert int(saved) == int(dest)


def get_index(src, i):
    return (
        src[i * SIZE]
        + (src[(i * SIZE) + 1] << 8)
        + (src[(i * SIZE) + 2] << 16)
        + (src[(i * SIZE) + 3] << 24)
    )


buffer = bytearray(((1 << max(BIT_THRESHOLDS) + 1) // 1024) * 1024)
next = 1
val = 0
for bit in BIT_THRESHOLDS:
    print("---", bit)
    pre, idx, post = (
        (((1 << bit) - (2 * SIZE)) // SIZE),
        (((1 << bit) - (1 * SIZE)) // SIZE),
        ((1 << bit) // SIZE),
    )
    val = (val << 8) + next
    next += 1
    set_index(buffer, pre, val & MASK)
    val = (val << 8) + next
    next += 1
    set_index(buffer, idx, val & MASK)
    val = (val << 8) + next
    next += 1
    set_index(buffer, post, val & MASK)
    val = (val << 8) + next
    next += 1
    print(hex(get_index(buffer, pre)), hex(get_index(buffer, idx)), hex(get_index(buffer, post)))
    exec(SET_TEMPLATE.format(off=pre, val=val & MASK))
    val = (val << 8) + next
    next += 1
    exec(SET_TEMPLATE.format(off=idx, val=val & MASK))
    val = (val << 8) + next
    next += 1
    exec(SET_TEMPLATE.format(off=post, val=val & MASK))