circuitpython/extmod/ulab/tests/1d/numpy/convolve.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

import math

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

x = np.array((1,2,3))
y = np.array((1,10,100,1000))
result = (np.convolve(x, y))
ref_result = np.array([1, 12, 123, 1230, 2300, 3000],dtype=np.float)
cmp_result = []
for p,q in zip(list(result), list(ref_result)):
    cmp_result.append(math.isclose(p, q, rel_tol=1e-06, abs_tol=1e-06))
print(cmp_result)