def __init__(self, ishape, axes=None, wave_name='db4', level=None):
self.wave_name = wave_name
self.axes = axes
self.level = level
oshape, _ = wavelet.get_wavelet_shape(ishape, wave_name, axes, level)
super().__init__(oshape, ishape)
def __init__(self, oshape, axes=None, wave_name='db4', level=None):
self.wave_name = wave_name
self.axes = axes
self.level = level
ishape, self.coeff_slices = wavelet.get_wavelet_shape(
oshape, wave_name, axes, level)
super().__init__(oshape, ishape)
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)
