def test_apod_with_fft(self):
spec = fromBruker(self.filename, False, False)
spec = fft1d(spec, 'fft')
spec = apod(spec, w=lambda s: EM(s, 0.))
test_data = ngp.fft( ngp.em(self.data) )
ts.assert_array_equal(spec, test_data, 'Apodization with zero filling and FFT failed.')
self.assertEqual("apodization" in spec.history._stepnames, True, 'Apodization not added to Spec history')
self.assertEqual(spec.history._stepnames, ['apodization', 'FFT'], 'Spec history not in the correct order (See fapplyBefore)')
def test_zf_with_fft(self):
spec = fromBruker(self.filename, False, False)
spec = fft1d(spec, 'fft')
spec = zf1d(spec, size = 2**15)
test_data = fft(zf_size(self.data, 2**15))
self.assertEqual(spec.shape[-1], 2**15, 'Spec size is not correct')
self.assertEqual("ZF" in spec.history._stepnames, True, 'ZF not added to Spec history')
self.assertEqual(spec.history._stepnames, ['ZF','FFT'], 'Spec history not in the correct order (See fapplyBefore)')
ts.assert_array_equal(spec, test_data, 'Simple Zero filling falied (see zf_size)')
self.assertEqual(spec.udic[0]['size'], 2**15, 'udic size not set correctly')
def freq_domain(self):
if self.udic[0]['freq']:
return self
else:
return fft(self)
