circuitpython/tests/micropython/viper_misc.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

152 lines
2.5 KiB
Python
Raw Normal View History

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