def test_template_gen(self): """Test template generation.""" from eqcorrscan.core.template_gen import from_sac import glob import os import obspy samp_rate = 20 length = 8 for event in ['2014p611252', 'No_head']: test_files = os.path.join(os.path.abspath(os.path. dirname(__file__)), 'test_data', 'SAC', event, '*') sac_files = glob.glob(test_files) # We currently do not support SAC template generation below version # 1.0.0 as before this, SACIO did not fill the reference time, # which is needed for defining pick times. This is usually the # trace start time, but isn't always... if int(obspy.__version__.split('.')[0]) >= 1: template = from_sac(sac_files, lowcut=2.0, highcut=8.0, samp_rate=samp_rate, filt_order=4, length=length, swin='all', prepick=0.1, debug=0, plot=False) for tr in template: self.assertEqual(len(tr.data), length * samp_rate) else: with self.assertRaises(NotImplementedError): template = from_sac(sac_files, lowcut=2.0, highcut=8.0, samp_rate=samp_rate, filt_order=4, length=length, swin='all', prepick=0.1, debug=0, plot=False)
def test_sac_template_gen(self): """Test template generation.""" from eqcorrscan.core.template_gen import from_sac import glob import os import obspy samp_rate = 20 length = 8 for event in ['2014p611252', 'No_head']: test_files = os.path.join(os.path.abspath(os.path. dirname(__file__)), 'test_data', 'SAC', event, '*') sac_files = glob.glob(test_files) # We currently do not support SAC template generation below version # 1.0.0 as before this, SACIO did not fill the reference time, # which is needed for defining pick times. This is usually the # trace start time, but isn't always... if int(obspy.__version__.split('.')[0]) >= 1: template = from_sac(sac_files, lowcut=2.0, highcut=8.0, samp_rate=samp_rate, filt_order=4, length=length, swin='all', prepick=0.1, debug=0, plot=False) for tr in template: self.assertEqual(len(tr.data), length * samp_rate) else: with self.assertRaises(NotImplementedError): template = from_sac(sac_files, lowcut=2.0, highcut=8.0, samp_rate=samp_rate, filt_order=4, length=length, swin='all', prepick=0.1, debug=0, plot=False)
def test_sac_template_gen(self): """Test template generation.""" samp_rate = 20 length = 8 for event in ['2014p611252', 'No_head']: test_files = os.path.join( os.path.abspath(os.path.dirname(__file__)), 'test_data', 'SAC', event, '*') # Test with various input types filelist = glob.glob(test_files) streamlist = [read(f) for f in glob.glob(test_files)] stream = read(test_files) for sac_files in [filelist, streamlist, stream]: templates = from_sac(sac_files, lowcut=2.0, highcut=8.0, samp_rate=samp_rate, filt_order=4, length=length, swin='all', prepick=0.1, debug=0, plot=False) self.assertEqual(len(templates), 1) template = templates[0] self.assertEqual(len(template), len(sactoevent(stream).picks)) for tr in template: self.assertEqual(len(tr.data), length * samp_rate)