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)