def test_upstream_calibration_controls(self):
QTest.mouseClick(self.widget.us_etalon_rb,
QtCore.Qt.LeftButton,
pos=QtCore.QPoint(
2,
self.widget.us_etalon_rb.height() / 2))
self.assertTrue(self.widget.us_etalon_rb.isChecked())
self.assertEqual(
self.model.us_temperature_model.calibration_parameter.modus, 1)
QTest.mouseClick(self.widget.us_temperature_rb,
QtCore.Qt.LeftButton,
pos=QtCore.QPoint(
2,
self.widget.us_etalon_rb.height() / 2))
self.assertEqual(
self.model.us_temperature_model.calibration_parameter.modus, 0)
self.widget.us_temperature_txt.setText('')
QTest.keyClicks(self.widget.us_temperature_txt, '1800')
QTest.keyPress(self.widget.us_temperature_txt, QtCore.Qt.Key_Enter)
self.assertEqual(
self.model.us_temperature_model.calibration_parameter.temperature,
1800)
filename = os.path.join(temperature_fitting_path, '15A_lamp.txt')
self.controller.load_us_etalon_file(filename)
self.assertEqual(
self.model.us_temperature_model.calibration_parameter.
etalon_file_name, filename)
self.assertEqual(str(self.widget.us_etalon_filename_lbl.text()),
os.path.basename(filename))
def test_markers(self):
self.add_one_pick()
pv = self.pile_viewer
self.assertEqual(pv.viewer.get_active_event(), None)
conf = config.config('snuffler')
# test kinds and phases
kinds = range(5)
fkey_map = pyrocko_pile_viewer.fkey_map
for k in kinds:
for fkey, fkey_int in fkey_map.items():
fkey_int += 1
QTest.keyPress(pv, fkey)
QTest.keyPress(pv, str(k))
if fkey_int != 10:
want = conf.phase_key_mapping.get("F%s" % fkey_int,
'Undefined')
else:
want = None
m = pv.viewer.get_markers()[0]
self.assertEqual(m.kind, k)
if want:
self.assertEqual(m.get_phasename(), want)
def test_changing_roi(self):
self.controller.load_data_file(os.path.join(unittest_files_path, 'temper_009.spe'))
x, y = self.model.spectrum.data
self.widget.roi_widget.roi_gbs[0].x_min_txt.setText("")
QTest.keyClicks(self.widget.roi_widget.roi_gbs[0].x_min_txt, "0")
QTest.keyPress(self.widget.roi_widget.roi_gbs[0].x_min_txt, QtCore.Qt.Key_Enter)
new_x, new_y = self.model.spectrum.data
self.assertNotEqual(len(x), len(new_x))
def test_click_non_dialogs(self):
'''Click through many menu option combinations that do not require
further interaction. Activate options in pairs of two.
'''
pv = self.pile_viewer
non_dialog_actions = [
'Indivdual Scale',
'Common Scale',
'Common Scale per Station',
'Common Scale per Component',
'Scaling based on Minimum and Maximum',
'Scaling based on Mean +- 2 x Std. Deviation',
'Scaling based on Mean +- 4 x Std. Deviation',
'Sort by Names',
'Sort by Distance',
'Sort by Azimuth',
'Sort by Distance in 12 Azimuthal Blocks',
'Sort by Backazimuth',
'3D distances',
'Subsort by Network, Station, Location, Channel',
'Subsort by Network, Station, Channel, Location',
'Subsort by Station, Network, Channel, Location',
'Subsort by Location, Network, Station, Channel',
'Subsort by Channel, Network, Station, Location',
'Subsort by Network, Station, Channel (Grouped by Location)',
'Subsort by Station, Network, Channel (Grouped by Location)',
]
options = [
'Antialiasing', 'Liberal Fetch Optimization', 'Clip Traces',
'Show Boxes', 'Color Traces', 'Show Scale Ranges',
'Show Scale Axes', 'Show Zero Lines', 'Fix Scale Ranges',
'Allow Downsampling', 'Allow Degapping', 'FFT Filtering',
'Bandpass is Lowpass + Highpass', 'Watch Files'
]
# create an event marker and activate it
self.add_one_pick()
QTest.keyPress(self.pile_viewer, 'A')
QTest.keyPress(self.pile_viewer, 'e')
event = model.Event()
markers = pv.viewer.get_markers()
self.assertEqual(len(markers), 1)
markers[0]._event = event
right_click_menu = self.pile_viewer.viewer.menu
for action_name in non_dialog_actions:
for oa in options:
for ob in options:
self.click_menu_item(right_click_menu, action_name)
self.click_menu_item(right_click_menu, oa)
self.click_menu_item(right_click_menu, ob)
options.remove(oa)
def test_spectrum_bkg_txt_fields_change_linear_regions(self):
self.spectrum_model.load_spectrum(os.path.join(data_path, 'spectrum_001.xy'))
self.widget.bkg_spectrum_gb.setChecked(True)
self.widget.bkg_spectrum_inspect_btn.toggle()
self.widget.bkg_spectrum_x_min_txt.setText('5')
QTest.keyPress(self.widget.bkg_spectrum_x_min_txt, QtCore.Qt.Key_Enter)
self.widget.bkg_spectrum_x_max_txt.setText('11')
QTest.keyPress(self.widget.bkg_spectrum_x_max_txt, QtCore.Qt.Key_Enter)
x_min, x_max = self.widget.spectrum_view.linear_region_item.getRegion()
self.assertEqual(x_min, 5)
self.assertEqual(x_max, 11)
def test_event_marker(self):
pv = self.pile_viewer
self.add_one_pick()
# select all markers
QTest.keyPress(pv, 'a', Qt.ShiftModifier)
# convert to EventMarker
QTest.keyPress(pv, 'e')
for m in pv.viewer.get_markers():
self.assertTrue(isinstance(m, gui_util.EventMarker))
def test_manual(self):
self._window.show()
self.qpart.text = 'aaaaa\naaaaaXXXXX\n'
base._processPendingEvents(self.app)
self.qpart.cursorPosition = (2, 0)
QTest.keyClicks(self.qpart, "a")
QTest.keyPress(self.qpart, Qt.Key_Space, Qt.ControlModifier, 100)
QTest.keyClicks(self.qpart, "a")
QTest.keyClick(self.qpart, Qt.Key_Tab)
self.assertEqual(self.qpart.text, 'aaaaa\naaaaaXXXXX\naaaaa')
def test_spectrum_bkg_txt_fields_change_linear_regions(self):
self.spectrum_model.load_pattern(
os.path.join(data_path, 'spectrum_001.xy'))
self.widget.bkg_spectrum_gb.setChecked(True)
self.widget.bkg_spectrum_inspect_btn.toggle()
self.widget.bkg_spectrum_x_min_txt.setText('5')
QTest.keyPress(self.widget.bkg_spectrum_x_min_txt, QtCore.Qt.Key_Enter)
self.widget.bkg_spectrum_x_max_txt.setText('11')
QTest.keyPress(self.widget.bkg_spectrum_x_max_txt, QtCore.Qt.Key_Enter)
x_min, x_max = self.widget.pattern_widget.linear_region_item.getRegion(
)
self.assertEqual(x_min, 5)
self.assertEqual(x_max, 11)
def test_too_long_list(self):
self._window.show()
self.qpart.text = '\n'.join(['asdf' + str(i) \
for i in range(100)]) + '\n'
base._processPendingEvents(self.app)
self.qpart.cursorPosition = (100, 0)
QTest.keyClicks(self.qpart, "asdf")
self.assertIsNotNone(self.qpart._completer._widget)
self.qpart.text = '\n'.join(['asdf' + str(i) \
for i in range(1000)]) + '\n'
base._processPendingEvents(self.app)
self.qpart.cursorPosition = (1000, 0)
QTest.keyClicks(self.qpart, "asdf")
self.assertIsNone(self.qpart._completer._widget)
QTest.keyPress(self.qpart, Qt.Key_Space, Qt.ControlModifier, 100)
self.assertIsNotNone(self.qpart._completer._widget)
def testRealizaMovimientoConControles(self):
class ActorConControles(pilasengine.actores.actor.Actor):
def actualizar(self):
control = self.pilas.escena_actual().control
if control.derecha:
self.x += 1
elif control.izquierda:
self.x -= 1
actor = ActorConControles(self.pilas)
self.assertEqual(0, actor.x, "Comienza en el punto (0, 0)")
# Simula la pulsación de la tecla Derecha
widget = self.pilas.obtener_widget()
QTest.keyPress(widget, QtCore.Qt.Key_Right)
# Simula que pasó un solo tick (si hago 60 de estos simulará 1 segundo).
self.pilas.simular_actualizacion_logica()
self.assertEqual(1, actor.x, "Luego de pulsar DERECHA se mueve")
# importante: suelta la tecla que comezó a pulsar antes.
QTest.keyRelease(widget, QtCore.Qt.Key_Right)
# Pulsa dos veces hacia la izquierda:
QTest.keyPress(widget, QtCore.Qt.Key_Left)
self.pilas.simular_actualizacion_logica()
QTest.keyPress(widget, QtCore.Qt.Key_Left)
self.pilas.simular_actualizacion_logica()
self.assertEqual(-1, actor.x, "Luego de pulsar dos veces \
IZQUIERDA pasa a x=-1")
def testRealizaMovimientoConControles(self):
class ActorConControles(pilasengine.actores.actor.Actor):
def actualizar(self):
control = self.pilas.escena_actual().control
if control.derecha:
self.x += 1
elif control.izquierda:
self.x -= 1
actor = ActorConControles(self.pilas)
self.assertEqual(0, actor.x, "Comienza en el punto (0, 0)")
# Simula la pulsación de la tecla Derecha
widget = self.pilas.obtener_widget()
QTest.keyPress(widget, QtCore.Qt.Key_Right)
# Simula que pasó un solo tick (si hago 60 de estos simulará 1 segundo).
self.pilas.simular_actualizacion_logica()
self.assertEqual(1, actor.x, "Luego de pulsar DERECHA se mueve")
# importante: suelta la tecla que comezó a pulsar antes.
QTest.keyRelease(widget, QtCore.Qt.Key_Right)
# Pulsa dos veces hacia la izquierda:
QTest.keyPress(widget, QtCore.Qt.Key_Left)
self.pilas.simular_actualizacion_logica()
QTest.keyPress(widget, QtCore.Qt.Key_Left)
self.pilas.simular_actualizacion_logica()
self.assertEqual(
-1, actor.x, "Luego de pulsar dos veces \
IZQUIERDA pasa a x=-1")
def enter_value_into_text_field(self, text_field, value):
text_field.setText('')
QTest.keyClicks(text_field, str(value))
QTest.keyPress(text_field, QtCore.Qt.Key_Enter)
QtGui.QApplication.processEvents()
def u_bahn_():
# cube
QTest.keyPress(win.viewer_3d, Qt.Key_B, Qt.ControlModifier)
qtest.mouse_move([350, 320])
QTest.keyPress(win.viewer_3d, Qt.Key_C, Qt.NoModifier, 100)
QTest.keyPress(win.command_dock_widget.line_edit, Qt.Key_Z, Qt.NoModifier)
QTest.keyPress(win.command_dock_widget.line_edit, Qt.Key_Minus,
Qt.NoModifier)
QTest.keyPress(win.command_dock_widget.line_edit, Qt.Key_1, Qt.NoModifier)
QTest.keyPress(win.command_dock_widget.line_edit, Qt.Key_Return,
Qt.NoModifier)
qtest.mouse_move([430, 320])
QTest.keyPress(win.command_dock_widget.line_edit, Qt.Key_Return,
Qt.NoModifier)
qtest.click_at([430, 330])
# sphere
qtest.click_at([120, 10])
qtest.click_at([120, 130])
qtest.click_at([390, 320])
qtest.click_at([430, 320])
# intersection
qtest.click_at([200, 10])
qtest.click_at([200, 55])
qtest.click_at([390, 320])
qtest.click_at([370, 330])
# delete
qtest.click_at([200, 10])
qtest.click_at([200, 30])
qtest.click_at([390, 320])
qtest.click_at([360, 320])
QTest.keyPress(win.viewer_3d, Qt.Key_Escape, Qt.NoModifier)
def cube_diagonal_():
QTest.keyPress(win.viewer_3d, Qt.Key_P, Qt.AltModifier)
qtest.click_at([250, 180])
qtest.click_at([210, 440])
qtest.click_at([650, 400])
QTest.keyPress(win, Qt.Key_X, Qt.AltModifier)
qtest.click_at([400, 50])
qtest.click_at([400, 400])
QTest.keyPress(win, Qt.Key_Space, Qt.NoModifier)
qtest.click_at([700, 100])
QTest.keyPress(win, Qt.Key_G, Qt.ControlModifier)
qtest.click_at([215, 440])
QTest.keyPress(win, Qt.Key_Space)
QTest.keyPress(win, Qt.Key_Space)
qtest.click_at([660, 90])
def enter_value_into_text_field(self, text_field, value):
text_field.setText('')
QTest.keyClicks(text_field, str(value))
QTest.keyPress(text_field, QtCore.Qt.Key_Enter)
QtGui.QApplication.processEvents()
def testCal(self):
self.lineEdit.setText("1+1")
QTest.keyPress(self.lineEdit,Qt.Key_Enter,Qt.NoModifier, -1)
self.assertEquals(self.browser.toPlainText().count(),5, self.browser.toPlainText().count())
def clear_all_markers(self):
pv = self.pile_viewer
QTest.keyPress(pv, 'a', Qt.ShiftModifier, 10)
QTest.keyPress(pv, Qt.Key_Backspace)
self.assertEqual(len(pv.viewer.get_markers()), 0)
def tearDownClass(cls):
'''
Quit snuffler.
'''
QTest.keyPress(cls.pile_viewer, 'q')
