aboutsummaryrefslogtreecommitdiff
path: root/circuitpython/extmod/ulab/tests/1d/complex/complex_exp.py
blob: 979b5b8e1d16146107f104c0525a837515809079 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# this test is meaningful only, when the firmware supports complex arrays

try:
    from ulab import numpy as np
except:
    import numpy as np

dtypes = (np.uint8, np.int8, np.uint16, np.int16, np.float, np.complex)

for dtype in dtypes:
    a = np.array(range(4), dtype=dtype)
    print('\narray:\n', a)
    print('\nexponential:\n', np.exp(a))

b = np.array([0, 1j, 2+2j, 3-3j], dtype=np.complex)
print('\narray:\n', b)
print('\nexponential:\n', np.exp(b))