micropython/tests/micropython/viper_subscr_multi.py

# test viper with multiple subscripts in a single expression


@micropython.viper
def f1(b: ptr8):
    b[0] += b[1]


b = bytearray(b"\x01\x02")
f1(b)
print(b)


@micropython.viper
def f2(b: ptr8, i: int):
    b[0] += b[i]


b = bytearray(b"\x01\x02")
f2(b, 1)
print(b)


@micropython.viper
def f3(b: ptr8) -> int:
    return b[0] << 24 | b[1] << 16 | b[2] << 8 | b[3]


print(hex(f3(b"\x01\x02\x03\x04")))