def test_match_spectra_sizes_3d_with_taper(self): s1 = hs.signals.Signal1D(np.zeros((10, 20, 2000))) s1.data[:] = 1.0 s2 = hs.signals.Signal1D(np.zeros((10, 20, 1000))) s2.data[:] = 1.0 o1, o2 = squ.match_spectra_sizes(s1, s2, taper=True, size=100) assert np.allclose(o1.data.shape, o2.data.shape)
def test_match_spectra_sizes_1d(self): s1 = hs.signals.Signal1D(np.zeros(2000)) s1.data[:] = 1.0 s2 = hs.signals.Signal1D(np.zeros(1000)) s2.data[:] = 1.0 o1, o2 = squ.match_spectra_sizes(s1, s2, taper=False) assert np.allclose(o1.data.shape, o2.data.shape)