aboutsummaryrefslogtreecommitdiff
path: root/circuitpython/extmod/ulab/tests/2d/numpy/eye.py
diff options
context:
space:
mode:
authorRaghuram Subramani <raghus2247@gmail.com>2022-06-19 19:47:51 +0530
committerRaghuram Subramani <raghus2247@gmail.com>2022-06-19 19:47:51 +0530
commit4fd287655a72b9aea14cdac715ad5b90ed082ed2 (patch)
tree65d393bc0e699dd12d05b29ba568e04cea666207 /circuitpython/extmod/ulab/tests/2d/numpy/eye.py
parent0150f70ce9c39e9e6dd878766c0620c85e47bed0 (diff)
add circuitpython code
Diffstat (limited to 'circuitpython/extmod/ulab/tests/2d/numpy/eye.py')
-rw-r--r--circuitpython/extmod/ulab/tests/2d/numpy/eye.py30
1 files changed, 30 insertions, 0 deletions
diff --git a/circuitpython/extmod/ulab/tests/2d/numpy/eye.py b/circuitpython/extmod/ulab/tests/2d/numpy/eye.py
new file mode 100644
index 0000000..630eed4
--- /dev/null
+++ b/circuitpython/extmod/ulab/tests/2d/numpy/eye.py
@@ -0,0 +1,30 @@
+try:
+ from ulab import numpy as np
+except:
+ import numpy as np
+
+dtypes = (np.uint8, np.int8, np.uint16, np.int16, np.float)
+
+print(np.ones(3))
+print(np.ones((3,3)))
+
+print(np.eye(3))
+print(np.eye(3, M=4))
+print(np.eye(3, M=4, k=0))
+print(np.eye(3, M=4, k=-1))
+print(np.eye(3, M=4, k=-2))
+print(np.eye(3, M=4, k=-3))
+print(np.eye(3, M=4, k=1))
+print(np.eye(3, M=4, k=2))
+print(np.eye(3, M=4, k=3))
+print(np.eye(4, M=4))
+print(np.eye(4, M=3, k=0))
+print(np.eye(4, M=3, k=-1))
+print(np.eye(4, M=3, k=-2))
+print(np.eye(4, M=3, k=-3))
+print(np.eye(4, M=3, k=1))
+print(np.eye(4, M=3, k=2))
+print(np.eye(4, M=3, k=3))
+
+for dtype in dtypes:
+ print(np.eye(3, dtype=dtype)) \ No newline at end of file