diff options
Diffstat (limited to 'circuitpython/extmod/ulab/tests/2d/numpy/ones.py')
| -rw-r--r-- | circuitpython/extmod/ulab/tests/2d/numpy/ones.py | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/circuitpython/extmod/ulab/tests/2d/numpy/ones.py b/circuitpython/extmod/ulab/tests/2d/numpy/ones.py new file mode 100644 index 0000000..f0aee86 --- /dev/null +++ b/circuitpython/extmod/ulab/tests/2d/numpy/ones.py @@ -0,0 +1,13 @@ +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))) + +for dtype in dtypes: + print(np.ones((3,3), dtype=dtype)) + print(np.ones((4,2), dtype=dtype)) |