def test_get_dft(self): """ """ cal_dft = self.s.get_dft('m/s2') self.assertEqual(type(cal_dft), DFT) cal_fas = cal_dft.fas tgt_fas = read_fas(os.path.join(DATA_DIR, '004676xa.fas')) fs = os.path.join(OUTPUT_DIR, 'recordings.seismogram.dft.png') plot_fass([tgt_fas, cal_fas], labels=['tgt', 'cal'], colors=['r', 'b'], widths=[3, 1], title='dft calculation', filespec=fs)
def test_get_dft(self): """ """ cal_dft = self.smg.get_dft('m/s2') self.assertEqual(type(cal_dft), DFT) cal_fas = cal_dft.fas tgt_fas = read_fas(os.path.join(DATA_DIR, '004676xa.fas')) fs = os.path.join(FIG_DIR, 'recordings.seismogram.dft.png') plot.fass( [tgt_fas, cal_fas], labels=['tgt', 'cal'], colors=['r', 'b'], widths=[2, 0.5], title='dft-fas calculation', png_filespec=fs )
def test_read_fas(self): """ """ fas = read_fas(os.path.join(DATA_DIR, '004676xa.fas')) f_freqs = [1.22070E-02, 2.44141E-02, 3.66211E-02] l_freqs = [9.99756E+01, 9.99878E+01, 1.00000E+02] f_ampls = [2.29340E-06, 3.22889E-06, 2.14292E-05] l_ampls = [2.76323E-06, 2.65286E-06, 2.41304E-06] self.assertListEqual(list(fas.frequencies[:+3]), f_freqs) self.assertListEqual(list(fas.frequencies[-3:]), l_freqs) self.assertListEqual(list(fas.amplitudes[:+3]), f_ampls) self.assertListEqual(list(fas.amplitudes[-3:]), l_ampls) self.assertEqual(fas.unit, UNITS['acc'])