def test__save_headers_filterParams(self):
# copy the file each time so we can destroy it
shutil.copy2('20120124.00520523.bhz.pkl', 'temp-test-bhz.pkl')
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac1, opts1 = getDataOpts()
axs1 = getAxes(opts1)
setattr(opts1,'labelqual',True)
ppm1 = PickPhaseMenu(gsac1, opts1, axs1)
originalSaclist = gsac1.saclist
# click the save button
click_save_event = matplotlib.backend_bases.MouseEvent('button_press_event', ppm1.axpp.figure.canvas, 63, 481)
ppm1.save(click_save_event)
#click the save parameters chosen event
click_params_event = matplotlib.backend_bases.MouseEvent('button_press_event', ppm1.axpp.figure.canvas, 415, 54)
ppm1.save_headers_filterParams(click_params_event)
# reload to make sure data has been overwritten
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac2, opts2 = getDataOpts()
for i in xrange(len(gsac2.saclist)):
self.assertNotEqual(gsac1.saclist[i], gsac2.saclist[i])
def test__save_headers_filterParams(self):
# copy the file each time so we can destroy it
shutil.copy2('20120124.00520523.bhz.pkl', 'temp-test-bhz.pkl')
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac1, opts1 = getDataOpts()
axs1 = getAxes(opts1)
setattr(opts1,'labelqual',True)
ppm1 = PickPhaseMenu(gsac1, opts1, axs1)
# randomly get stuff
rand_order = random.randint(1, 4)
rand_lowFreq = random.uniform(0.1,0.2)
rand_highFreq = random.uniform(0.8,1.2)
opts1.filterParameters['lowFreq'] = rand_lowFreq
opts1.filterParameters['highFreq'] = rand_highFreq
opts1.filterParameters['order'] = rand_order
# click the save button
click_save_event = matplotlib.backend_bases.MouseEvent('button_press_event', ppm1.axpp.figure.canvas, 63, 481)
ppm1.save(click_save_event)
#click the save parameters chosen event
click_params_event = matplotlib.backend_bases.MouseEvent('button_press_event', ppm1.axpp.figure.canvas, 256, 48)
ppm1.save_headers_filterParams(click_params_event)
# reload to make sure parameters exist now
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac2, opts2 = getDataOpts()
self.assertEqual(rand_lowFreq, opts2.filterParameters['lowFreq'])
self.assertEqual(rand_highFreq, opts2.filterParameters['highFreq'])
self.assertEqual(rand_order, opts2.filterParameters['order'])
def test__save_headers_filterParams(self):
# copy the file each time so we can destroy it
shutil.copy2('20120124.00520523.bhz.pkl', 'temp-test-bhz.pkl')
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac1, opts1 = getDataOpts()
axs1 = getAxes(opts1)
setattr(opts1, 'labelqual', True)
ppm1 = PickPhaseMenu(gsac1, opts1, axs1)
originalSaclist = gsac1.saclist
# click the save button
click_save_event = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppm1.axpp.figure.canvas, 63, 481)
ppm1.save(click_save_event)
#click the save parameters chosen event
click_params_event = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppm1.axpp.figure.canvas, 415, 54)
ppm1.save_headers_filterParams(click_params_event)
# reload to make sure data has been overwritten
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac2, opts2 = getDataOpts()
for i in xrange(len(gsac2.saclist)):
self.assertNotEqual(gsac1.saclist[i], gsac2.saclist[i])
def test_sortButtonWorks(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
# get files before sorting
unsortedFiles = []
for sacdh in gsac.selist:
unsortedFiles.append(sacdh.filename)
# click the sort button
event_clickSortBtn = matplotlib.backend_bases.MouseEvent('button_press_event', ppmm.axstk.figure.canvas, 62, 295)
ppmm.sorting(event_clickSortBtn)
# click sort filename button
event_clickSortFilenameBtn = matplotlib.backend_bases.MouseEvent('button_press_event', ppmm.figsort.canvas, 151, 700)
ppmm.sort_file(event_clickSortFilenameBtn)
# get files after sorting
sortedFiles = []
for sacdh in gsac.selist:
sortedFiles.append(sacdh.filename)
self.assertNotEqual(unsortedFiles, sortedFiles)
def test_getDataOpts(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
self.assertEqual(len(gsac.delist), 7)
self.assertEqual(len(gsac.selist), 117)
self.assertEqual(
len(gsac.delist) + len(gsac.selist), len(gsac.saclist))
def test_sortButtonWorks(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
# get files before sorting
unsortedFiles = []
for sacdh in gsac.selist:
unsortedFiles.append(sacdh.filename)
# click the sort button
event_clickSortBtn = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppmm.axstk.figure.canvas, 62, 295)
ppmm.sorting(event_clickSortBtn)
# click sort filename button
event_clickSortFilenameBtn = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppmm.figsort.canvas, 151, 700)
ppmm.sort_file(event_clickSortFilenameBtn)
# get files after sorting
sortedFiles = []
for sacdh in gsac.selist:
sortedFiles.append(sacdh.filename)
self.assertNotEqual(unsortedFiles, sortedFiles)
def runBefore(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
fake_event = matplotlib.backend_bases.MouseEvent('button_press_event', ppmm.axstk.figure.canvas, 56, 224)
ppmm.filtering(fake_event)
return ppmm
def test_timeSelector_ignore(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
opts.labelqual = True
axs = getAxes(opts)
ppm = PickPhaseMenu(gsac, opts, axs)
ppm.on_select(-3,4.3)
self.assertEqual(ppm.axpp.get_xlim(),(-3,4.3))
def runBefore(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
fake_event = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppmm.axstk.figure.canvas, 56, 224)
ppmm.filtering(fake_event)
return ppmm
def test_filter_spreadButter(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
# click the filter button
event_clickFilterBtn = matplotlib.backend_bases.MouseEvent('button_press_event', ppmm.axstk.figure.canvas, 71,223)
ppmm.filtering(event_clickFilterBtn)
# click apply filter button
event_clickApplyFilterBtn = matplotlib.backend_bases.MouseEvent('button_press_event', ppmm.figfilter.canvas, 646,829)
ppmm.applyFilter(event_clickApplyFilterBtn)
def test_sortFigExists(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
self.assertFalse(hasattr(ppmm,'figsort'))
fake_event = matplotlib.backend_bases.MouseEvent('button_press_event', ppmm.axstk.figure.canvas, 62, 295)
ppmm.sorting(fake_event)
self.assertIsNotNone(ppmm.figsort)
self.assertIsNotNone(ppmm.sortAxs)
def main(): gsac, opts = getDataOpts() axs = getAxes(opts) ppmm = PickPhaseMenuMore(gsac, opts, axs) fmt = 'png' fmt = 'pdf' if opts.savefig: if opts.pklfile is None: fignm = 'ttpick.' + fmt else: fignm = opts.pklfile + '.' + fmt plt.savefig(fignm, format=fmt) else: plt.show()
def test_sortFigExists(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
self.assertFalse(hasattr(ppmm, 'figsort'))
fake_event = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppmm.axstk.figure.canvas, 62, 295)
ppmm.sorting(fake_event)
self.assertIsNotNone(ppmm.figsort)
self.assertIsNotNone(ppmm.sortAxs)
def main():
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
fmt = 'png'
fmt = 'pdf'
if opts.savefig:
if opts.pklfile is None:
fignm = 'ttpick.' + fmt
else:
fignm = opts.pklfile + '.' + fmt
plt.savefig(fignm, format=fmt)
else:
plt.show()
def test_filter_spreadButter(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
axs = getAxes(opts)
ppmm = PickPhaseMenuMore(gsac, opts, axs)
# click the filter button
event_clickFilterBtn = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppmm.axstk.figure.canvas, 71, 223)
ppmm.filtering(event_clickFilterBtn)
# click apply filter button
event_clickApplyFilterBtn = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppmm.figfilter.canvas, 646, 829)
ppmm.applyFilter(event_clickApplyFilterBtn)
def test__save_headers_filterParams(self):
# copy the file each time so we can destroy it
shutil.copy2('20120124.00520523.bhz.pkl', 'temp-test-bhz.pkl')
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac1, opts1 = getDataOpts()
axs1 = getAxes(opts1)
setattr(opts1, 'labelqual', True)
ppm1 = PickPhaseMenu(gsac1, opts1, axs1)
# randomly get stuff
rand_order = random.randint(1, 4)
rand_lowFreq = random.uniform(0.1, 0.2)
rand_highFreq = random.uniform(0.8, 1.2)
opts1.filterParameters['lowFreq'] = rand_lowFreq
opts1.filterParameters['highFreq'] = rand_highFreq
opts1.filterParameters['order'] = rand_order
# click the save button
click_save_event = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppm1.axpp.figure.canvas, 63, 481)
ppm1.save(click_save_event)
#click the save parameters chosen event
click_params_event = matplotlib.backend_bases.MouseEvent(
'button_press_event', ppm1.axpp.figure.canvas, 256, 48)
ppm1.save_headers_filterParams(click_params_event)
# reload to make sure parameters exist now
sys.argv[1:] = ['temp-test-bhz.pkl']
gsac2, opts2 = getDataOpts()
self.assertEqual(rand_lowFreq, opts2.filterParameters['lowFreq'])
self.assertEqual(rand_highFreq, opts2.filterParameters['highFreq'])
self.assertEqual(rand_order, opts2.filterParameters['order'])
def test_sortSeismogramsFilename(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
# before sorting
unsortedFiles = []
for sacdh in gsac.saclist:
unsortedFiles.append(sacdh.filename)
# after sorting
sortedFiles = []
opts.sortby = 'i'
sortSeis(gsac, opts)
for sacdh in gsac.saclist:
sortedFiles.append(sacdh.filename)
sortedFiles = sortedFiles.sort()
self.assertEqual(sortedFiles, unsortedFiles.sort())
def test_sortSeismogramsFilename(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
# before sorting
unsortedFiles = []
for sacdh in gsac.saclist:
unsortedFiles.append(sacdh.filename)
# after sorting
sortedFiles = []
opts.sortby = 'i';
sortSeis(gsac, opts)
for sacdh in gsac.saclist:
sortedFiles.append(sacdh.filename)
sortedFiles = sortedFiles.sort()
self.assertEqual(sortedFiles, unsortedFiles.sort())
def test_readSACfilesRight(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
# event year correct
self.assertEqual(gsac.event[0], 2012)
# event month correct
self.assertEqual(gsac.event[1], 1)
# event day correct
self.assertEqual(gsac.event[2], 9)
# event lat correct
self.assertEqual(gsac.event[6], -10.616999626159668)
# event lon correct
self.assertEqual(gsac.event[7], 165.16000366210938)
# event depth correct
self.assertEqual(gsac.event[8], 28.0)
# event magnitude correct
self.assertEqual(gsac.event[9], 6.400000095367432)
def test_readSACfilesRight(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
# event year correct
self.assertEqual(gsac.event[0], 2012)
# event month correct
self.assertEqual(gsac.event[1], 1)
# event day correct
self.assertEqual(gsac.event[2], 9)
# event lat correct
self.assertEqual(gsac.event[6], -10.616999626159668)
# event lon correct
self.assertEqual(gsac.event[7], 165.16000366210938)
# event depth correct
self.assertEqual(gsac.event[8], 28.0)
# event magnitude correct
self.assertEqual(gsac.event[9], 6.400000095367432)
def test_getDataOpts(self):
sys.argv[1:] = [test_filename]
gsac, opts = getDataOpts()
self.assertEqual(len(gsac.delist),7)
self.assertEqual(len(gsac.selist),117)
self.assertEqual(len(gsac.delist)+len(gsac.selist),len(gsac.saclist))
