def test_fwt_iwt(self):
for n in range(5, 11):
input = np.zeros(n, dtype=np.float)
input[0] = 1
_, coeff_slices = wavelet.get_wavelet_shape([n])
npt.assert_allclose(
wavelet.iwt(wavelet.fwt(input), [n], coeff_slices), input)
def _apply(self, input):
return wavelet.iwt(input,
self.oshape,
self.coeff_slices,
wave_name=self.wave_name,
axes=self.axes,
level=self.level)
def _apply(self, input):
device = backend.get_device(input)
with device:
return wavelet.iwt(input,
self.oshape,
wave_name=self.wave_name,
axes=self.axes,
level=self.level)
