mirror of https://github.com/pyodide/pyodide.git
20 lines
476 B
Python
20 lines
476 B
Python
|
def test_pywt(selenium):
|
||
|
|
selenium.load_package("pywavelets")
|
||
|
|
selenium.run(
|
||
|
|
r"""
|
||
|
|
import pywt
|
||
|
|
import numpy as np
|
||
|
|
|
||
|
|
def checkit(a,v):
|
||
|
|
assert (np.rint(a) == v).all()
|
||
|
|
|
||
|
|
x = [3, 7, 1, 1, -2, 5, 4, 6]
|
||
|
|
cA, cD = pywt.dwt(x, 'db2')
|
||
|
|
w = pywt.Wavelet('sym3')
|
||
|
|
checkit(pywt.idwt(cA, cD, 'db2'),x)
|
||
|
|
cA, cD = pywt.dwt(x, wavelet=w, mode='periodization')
|
||
|
|
checkit(pywt.idwt(cA, cD, 'sym3', 'symmetric'),[1,1,-2,5])
|
||
|
|
checkit(pywt.idwt(cA, cD, 'sym3', 'periodization'),x)
|
||
|
|
"""
|
||
|
|
)
|