aboutsummaryrefslogtreecommitdiff
path: root/circuitpython/extmod/ulab/tests/2d/numpy/buffer.py
diff options
context:
space:
mode:
Diffstat (limited to 'circuitpython/extmod/ulab/tests/2d/numpy/buffer.py')
-rw-r--r--circuitpython/extmod/ulab/tests/2d/numpy/buffer.py17
1 files changed, 17 insertions, 0 deletions
diff --git a/circuitpython/extmod/ulab/tests/2d/numpy/buffer.py b/circuitpython/extmod/ulab/tests/2d/numpy/buffer.py
new file mode 100644
index 0000000..5cce5b9
--- /dev/null
+++ b/circuitpython/extmod/ulab/tests/2d/numpy/buffer.py
@@ -0,0 +1,17 @@
+try:
+ from ulab import numpy as np
+except:
+ import numpy as np
+
+def print_as_buffer(a):
+ print(len(memoryview(a)), list(memoryview(a)))
+print_as_buffer(np.ones(3))
+print_as_buffer(np.zeros(3))
+print_as_buffer(np.eye(4))
+print_as_buffer(np.ones(1, dtype=np.int8))
+print_as_buffer(np.ones(2, dtype=np.uint8))
+print_as_buffer(np.ones(3, dtype=np.int16))
+print_as_buffer(np.ones(4, dtype=np.uint16))
+print_as_buffer(np.ones(5, dtype=np.float))
+print_as_buffer(np.linspace(0, 1, 9))
+