jepler/micropython-ulab
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity
micropython-ulab/tests/2d/numpy/operators.py
Zoltán Vörös 47ad73ab57
Floordiv (#593) 
* implement floor division

* fix 3D, 4D loops

* add missing array declaration in 3D, and 4D

* Add test cases for floor division and fix it for ints (#599)

* Add test cases for floor division

* Fix define name in comment

* Fix floor division of ints

---------

Co-authored-by: Maciej Sokołowski <matemaciek@gmail.com>
2023-04-23 21:14:59 +02:00

181 lines
7.4 KiB
Python
Raw Blame History

try:
from ulab import numpy as np
except ImportError:
import numpy as np
print(len(np.array([1, 2, 3, 4, 5], dtype=np.uint8)))
print(len(np.array([[1, 2, 3],[4, 5, 6]])))
print(~np.array([0, -1, -100], dtype=np.uint8))
print(~np.array([0, -1, -100], dtype=np.uint16))
print(~np.array([0, -1, -100], dtype=np.int8))
print(~np.array([0, -1, -100], dtype=np.int16))
print(abs(np.array([0, -1, -100], dtype=np.uint8)))
print(abs(np.array([0, -1, -100], dtype=np.uint16)))
print(abs(np.array([0, -1, -100], dtype=np.int8)))
print(abs(np.array([0, -1, -100], dtype=np.int16)))
print(abs(np.array([0, -1, -100], dtype=np.float)))
print(-(np.array([0, -1, -100], dtype=np.uint8)))
print(-(np.array([0, -1, -100], dtype=np.uint16)))
print(-(np.array([0, -1, -100], dtype=np.int8)))
print(-(np.array([0, -1, -100], dtype=np.int16)))
print(-(np.array([0, -1, -100], dtype=np.float)))
print(+(np.array([0, -1, -100], dtype=np.uint8)))
print(+(np.array([0, -1, -100], dtype=np.uint16)))
print(+(np.array([0, -1, -100], dtype=np.int8)))
print(+(np.array([0, -1, -100], dtype=np.int16)))
print(+(np.array([0, -1, -100], dtype=np.float)))
print(np.array([1,2,3], dtype=np.float) > np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) > np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) > np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) < np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) < np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) < np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) >= np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) >= np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) >= np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) <= np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) <= np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) <= np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) > 4)
print(np.array([1,2,3], dtype=np.float) > 4.0)
print(np.array([1,2,3], dtype=np.float) < 4)
print(np.array([1,2,3], dtype=np.float) < 4.0)
print(np.array([1,2,3], dtype=np.float) == np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) == np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) == np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) != np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) != np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) != np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) == 4)
print(np.array([1,2,3], dtype=np.float) == 4.0)
print(np.array([1,2,3], dtype=np.float) != 4)
print(np.array([1,2,3], dtype=np.float) != 4.0)
print(np.array([1,2,3], dtype=np.float) - np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) - np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) - np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) + np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) + np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) + np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) * np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) * np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) * np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) ** np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) ** np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) ** np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) / np.array([4,5,6], dtype=np.float))
print(np.array([1,2,3], dtype=np.float) / np.array([4,5,6], dtype=np.uint16))
print(np.array([1,2,3], dtype=np.float) / np.array([4,5,6], dtype=np.int16))
print(np.array([10,20,30], dtype=np.float) // np.array([4,5,6], dtype=np.float))
print(np.array([10,20,30], dtype=np.float) // np.array([4,5,6], dtype=np.uint16))
print(np.array([10,20,30], dtype=np.float) // np.array([4,5,6], dtype=np.int16))
print(np.array([1,2,3], dtype=np.float) - 4)
print(np.array([1,2,3], dtype=np.float) - 4.0)
print(np.array([1,2,3], dtype=np.float) + 4)
print(np.array([1,2,3], dtype=np.float) + 4.0)
print(np.array([1,2,3], dtype=np.float) * 4)
print(np.array([1,2,3], dtype=np.float) * 4.0)
print(np.array([1,2,3], dtype=np.float) ** 4)
print(np.array([1,2,3], dtype=np.float) ** 4.0)
print(np.array([1,2,3], dtype=np.float) / 4)
print(np.array([1,2,3], dtype=np.float) / 4.0)
print(np.array([10,20,30], dtype=np.float) // 4)
print(np.array([10,20,30], dtype=np.float) // 4.0)
print(np.array([10,20,30], dtype=np.int8) // 4)
print(np.array([10,20,30], dtype=np.int8) // 4.0)
print(np.array([10,20,30], dtype=np.uint16) // 4)
print(np.array([10,20,30], dtype=np.uint16) // 4.0)
print(np.array([10,20,30], dtype=np.int16) // 4)
print(np.array([10,20,30], dtype=np.int16) // 4.0)
a = np.array([1,2,3], dtype=np.float)
a -= np.array([4,5,6], dtype=np.float)
print(a)
a = np.array([1,2,3], dtype=np.float)
a -= np.array([4,5,6], dtype=np.uint16)
print(a)
a = np.array([1,2,3], dtype=np.float)
a -= np.array([4,5,6], dtype=np.int16)
print(a)
a = np.array([1,2,3], dtype=np.float)
a += np.array([4,5,6], dtype=np.float)
print(a)
a = np.array([1,2,3], dtype=np.float)
a += np.array([4,5,6], dtype=np.uint16)
print(a)
a = np.array([1,2,3], dtype=np.float)
a += np.array([4,5,6], dtype=np.int16)
print(a)
a = np.array([1,2,3], dtype=np.float)
a *= np.array([4,5,6], dtype=np.float)
print(a)
a = np.array([1,2,3], dtype=np.float)
a *= np.array([4,5,6], dtype=np.uint16)
print(a)
a = np.array([1,2,3], dtype=np.float)
a *= np.array([4,5,6], dtype=np.int16)
print(a)
a = np.array([1,2,3], dtype=np.float)
#a /= np.array([4,5,6])
print(a)
a = np.array([1,2,3], dtype=np.float)
a **= np.array([4,5,6], dtype=np.float)
print(a)
a = np.array([1,2,3], dtype=np.float)
a **= np.array([4,5,6], dtype=np.uint16)
print(a)
a = np.array([1,2,3], dtype=np.float)
a **= np.array([4,5,6], dtype=np.int16)
print(a)
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.uint8))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.int8))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.int8))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.int16))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.int16))
print(np.array([1,2,3],dtype=np.uint16) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.int16))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.uint16))
print(np.array([1,2,3],dtype=np.uint8) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.int8) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.uint16) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.float))
print(np.array([1,2,3],dtype=np.int16) + np.array([4,5,6],dtype=np.float))
a = np.array([1, 2, 3, 4, 5], dtype=np.uint8)
for i, _a in enumerate(a):
print("element %d in a:"%i, _a)
Reference in a new issue View git blame Copy permalink