summaryrefslogtreecommitdiff
path: root/tests/micropython/viper_addr.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/micropython/viper_addr.py')
-rw-r--r--tests/micropython/viper_addr.py20
1 files changed, 12 insertions, 8 deletions
diff --git a/tests/micropython/viper_addr.py b/tests/micropython/viper_addr.py
index 0d8efb90b..84bc6c002 100644
--- a/tests/micropython/viper_addr.py
+++ b/tests/micropython/viper_addr.py
@@ -1,39 +1,43 @@
# test passing addresses to viper
+
@micropython.viper
-def get_addr(x:ptr) -> ptr:
+def get_addr(x: ptr) -> ptr:
return x
+
@micropython.viper
-def memset(dest:ptr8, c:int, n:int):
+def memset(dest: ptr8, c: int, n: int):
for i in range(n):
dest[i] = c
+
@micropython.viper
-def memsum(src:ptr8, n:int) -> int:
+def memsum(src: ptr8, n: int) -> int:
s = 0
for i in range(n):
s += src[i]
return s
+
# create array and get its address
-ar = bytearray('0000')
+ar = bytearray("0000")
addr = get_addr(ar)
print(type(ar))
print(type(addr))
print(ar)
# pass array as an object
-memset(ar, ord('1'), len(ar))
+memset(ar, ord("1"), len(ar))
print(ar)
# pass direct pointer to array buffer
-memset(addr, ord('2'), len(ar))
+memset(addr, ord("2"), len(ar))
print(ar)
# pass direct pointer to array buffer, with offset
-memset(addr + 2, ord('3'), len(ar) - 2)
+memset(addr + 2, ord("3"), len(ar) - 2)
print(ar)
# pass a read-only bytes object in
-print(memsum(b'\x01\x02\x03\x04', 4))
+print(memsum(b"\x01\x02\x03\x04", 4))