118 lines
2.5 KiB
Python
118 lines
2.5 KiB
Python
import micropython
|
|
|
|
# viper function taking and returning ints
|
|
@micropython.viper
|
|
def viper_int(x:int, y:int) -> int:
|
|
return x + y + 3
|
|
print(viper_int(1, 2))
|
|
|
|
# viper function taking and returning objects
|
|
@micropython.viper
|
|
def viper_object(x:object, y:object) -> object:
|
|
return x + y
|
|
print(viper_object(1, 2))
|
|
|
|
# return None as non-object (should return 0)
|
|
@micropython.viper
|
|
def viper_ret_none() -> int:
|
|
return None
|
|
print(viper_ret_none())
|
|
|
|
# 3 args
|
|
@micropython.viper
|
|
def viper_3args(a:int, b:int, c:int) -> int:
|
|
return a + b + c
|
|
print(viper_3args(1, 2, 3))
|
|
|
|
# 4 args
|
|
@micropython.viper
|
|
def viper_4args(a:int, b:int, c:int, d:int) -> int:
|
|
return a + b + c + d
|
|
# viper call with 4 args not yet supported
|
|
#print(viper_4args(1, 2, 3, 4))
|
|
|
|
# a local (should have automatic type int)
|
|
@micropython.viper
|
|
def viper_local(x:int) -> int:
|
|
y = 4
|
|
return x + y
|
|
print(viper_local(3))
|
|
|
|
# without type annotation, types should default to object
|
|
@micropython.viper
|
|
def viper_no_annotation(x, y):
|
|
return x * y
|
|
print(viper_no_annotation(4, 5))
|
|
|
|
# unsigned ints
|
|
@micropython.viper
|
|
def viper_uint() -> uint:
|
|
return uint(-1)
|
|
import sys
|
|
print(viper_uint() == (sys.maxsize << 1 | 1))
|
|
|
|
# a for loop
|
|
@micropython.viper
|
|
def viper_for(a:int, b:int) -> int:
|
|
total = 0
|
|
for x in range(a, b):
|
|
total += x
|
|
return total
|
|
print(viper_for(10, 10000))
|
|
|
|
# accessing a global
|
|
@micropython.viper
|
|
def viper_access_global():
|
|
global gl
|
|
gl = 1
|
|
return gl
|
|
print(viper_access_global(), gl)
|
|
|
|
# calling print with object and int types
|
|
@micropython.viper
|
|
def viper_print(x, y:int):
|
|
print(x, y + 1)
|
|
viper_print(1, 2)
|
|
|
|
# convert constants to objects in tuple
|
|
@micropython.viper
|
|
def viper_tuple_consts(x):
|
|
return (x, 1, False, True)
|
|
print(viper_tuple_consts(0))
|
|
|
|
# making a tuple from an object and an int
|
|
@micropython.viper
|
|
def viper_tuple(x, y:int):
|
|
return (x, y + 1)
|
|
print(viper_tuple(1, 2))
|
|
|
|
# making a list from an object and an int
|
|
@micropython.viper
|
|
def viper_list(x, y:int):
|
|
return [x, y + 1]
|
|
print(viper_list(1, 2))
|
|
|
|
# making a set from an object and an int
|
|
@micropython.viper
|
|
def viper_set(x, y:int):
|
|
return {x, y + 1}
|
|
print(sorted(list(viper_set(1, 2))))
|
|
|
|
# raising an exception
|
|
@micropython.viper
|
|
def viper_raise(x:int):
|
|
raise OSError(x)
|
|
try:
|
|
viper_raise(1)
|
|
except OSError as e:
|
|
print(repr(e))
|
|
|
|
# calling GC after defining the function
|
|
@micropython.viper
|
|
def viper_gc() -> int:
|
|
return 1
|
|
print(viper_gc())
|
|
import gc
|
|
gc.collect()
|
|
print(viper_gc())
|