コード例 #1
0
ファイル: pickphase_tests.py プロジェクト: XStargate/aimbat
    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])
コード例 #2
0
ファイル: pickphase_tests.py プロジェクト: XStargate/aimbat
    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'])
コード例 #3
0
ファイル: plotutils_tests.py プロジェクト: wbm06/aimbat
    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))
コード例 #4
0
    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])
コード例 #5
0
    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'])
コード例 #6
0
#	axs = {}
#	axs['Seis'] = axpp
#	dx = 0.07
#	x0 = rectseis[0] + rectseis[2] + 0.01
#	xf = x0 - dx*1
#	xq = x0 - dx*2
#	xs = x0 - dx*3
#	xn = x0 - dx*4
#	xz = x0 - dx*5
#	xp = x0 - dx*6
#	rectfron = [xf, 0.93, 0.06, 0.04]
#	rectprev = [xp, 0.93, 0.06, 0.04]
#	rectnext = [xn, 0.93, 0.06, 0.04]
#	rectzoba = [xz, 0.93, 0.06, 0.04]
#	rectsave = [xs, 0.93, 0.06, 0.04]
#	rectquit = [xq, 0.93, 0.06, 0.04]
#	axs['Fron'] = fig.add_axes(rectfron)
#	axs['Prev'] = fig.add_axes(rectprev)
#	axs['Next'] = fig.add_axes(rectnext)
#	axs['Zoba'] = fig.add_axes(rectzoba)
#	axs['Save'] = fig.add_axes(rectsave)
#	axs['Quit'] = fig.add_axes(rectquit)
#	return axs

if __name__ == "__main__":
    gsac, opts = getDataOpts()
    axs = getAxes(opts)
    ppm = PickPhaseMenu(gsac, opts, axs)

    show()