def test_tool_filter(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
widget = view.addLaser(Laser(linear_data(["a", "b", "c"])))
tool = DriftTool(view.activeWidget())
view.addTab("Tool", tool)
qtbot.waitExposed(tool)
tool.combo_element.setCurrentText("a")
tool.combo_element.activated.emit(0)
tool.spinbox_degree.setValue(1)
tool.apply()
assert np.all(np.isclose(widget.laser.data["a"], widget.laser.data["a"][0][0]))
tool.combo_element.setCurrentText("b")
tool.combo_element.activated.emit(0)
assert not np.all(np.isclose(widget.laser.data["b"], widget.laser.data["c"][0][0]))
tool.check_apply_all.setChecked(True)
tool.apply()
assert np.all(np.isclose(widget.laser.data["b"], widget.laser.data["c"][0][0]))
def test_laser_widget(qtbot: QtBot):
x = rand_data(["A1", "B2"])
y = x["A1"].copy()
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
view.addLaser(Laser(x))
widget = view.activeWidget()
widget.applyConfig(Config(1.0, 1.0, 1.0))
assert widget.laser.config.spotsize == 1.0
widget.applyCalibration({"B2": Calibration(2.0, 2.0)})
assert widget.laser.calibration["B2"].intercept == 2.0
widget.updateNames({"A1": "A1", "B2": "2B"})
assert np.all(viewspace.uniqueElements() == ["2B", "A1"])
widget.transform(flip="horizontal")
assert np.all(widget.laser.get("A1") == np.flip(y, axis=1))
widget.transform(flip="horizontal")
widget.transform(flip="vertical")
assert np.all(widget.laser.get("A1") == np.flip(y, axis=0))
widget.transform(flip="vertical")
assert np.all(widget.laser.get("A1") == y)
widget.transform(rotate="right")
assert np.all(widget.laser.get("A1") == np.rot90(y, k=1, axes=(1, 0)))
widget.transform(rotate="left")
assert np.all(widget.laser.get("A1") == y)
def test_standards_tool(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
data = linear_data(["A1", "B2"])
data["B2"][6] = 1.0
view.addLaser(Laser(data))
tool = StandardsTool(view.activeWidget())
view.addTab("Tool", tool)
qtbot.waitExposed(tool)
# Units
tool.lineedit_units.setText("unit")
tool.lineedit_units.editingFinished.emit()
tool.combo_weighting.setCurrentIndex(2)
# Table
tool.spinbox_levels.setValue(5)
assert not tool.isComplete()
assert not tool.button_plot.isEnabled()
for i in range(0, tool.table.model().rowCount()):
index = tool.table.model().index(i, 0)
tool.table.model().setData(index, i)
# Change element, check weighting
tool.combo_element.setCurrentIndex(1)
assert not tool.table.isComplete()
for i in range(0, tool.table.model().rowCount()):
index = tool.table.model().index(i, 0)
tool.table.model().setData(index, i)
assert tool.isComplete()
assert tool.button_plot.isEnabled()
assert tool.combo_weighting.currentIndex() == 0
assert tool.lineedit_units.text() == ""
tool.lineedit_units.setText("none")
assert tool.calibration["B2"].gradient == 1.75
# Check weighting updates results
tool.combo_weighting.setCurrentText("y")
assert tool.calibration["B2"].weighting == "y"
assert np.isclose(tool.calibration["B2"].gradient, 1.954022988)
# Change element back, check weighting, unit, table restored
tool.combo_element.setCurrentIndex(0)
assert tool.isComplete()
assert tool.lineedit_units.text() == "unit"
assert tool.combo_weighting.currentIndex() == 2
# Test SD weighting
tool.combo_weighting.setCurrentIndex(1)
assert np.all(tool.calibration["A1"].weights == 4.0)
tool.combo_element.setCurrentIndex(1)
dlg = tool.showCurve()
qtbot.waitExposed(dlg)
dlg.close()
def test_tool_calculator(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
widget = view.addLaser(Laser(rand_data(["a", "b"])))
tool = CalculatorTool(view.activeWidget())
view.addTab("Tool", tool)
qtbot.waitExposed(tool)
assert tool.lineedit_name.text() == "calc0"
tool.apply()
assert "calc0" in widget.laser.elements
assert tool.lineedit_name.text() == "calc1"
# overwrite
tool.lineedit_name.setText("a")
tool.apply()
assert len(widget.laser.elements) == 3
# Inserters
assert tool.formula.toPlainText() == "a"
tool.insertFunction(1)
assert tool.formula.toPlainText() == "abs(a"
tool.insertVariable(2)
assert tool.formula.toPlainText() == "abs(ba"
# Test output of previewData and output lineedit
x = np.array(np.random.random((10, 10)), dtype=[("a", float)])
tool.formula.setPlainText("mean(a)")
assert tool.previewData(x) is None
assert tool.output.text() == f"{np.mean(x['a']):.10g}"
# Array access in output
tool.formula.setPlainText("a[0]")
assert tool.previewData(x) is None
assert tool.output.text() == f"{list(map('{:.4g}'.format, x['a'][0]))}"
# Simple op
tool.formula.setPlainText("a + 1.0")
assert np.all(tool.previewData(x) == x["a"] + 1.0)
assert tool.isComplete()
# Invalid input
tool.formula.setPlainText("fail")
assert tool.previewData(x) is None
assert not tool.isComplete()
def test_laser_widget_actions(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
view.addLaser(
Laser(rand_data(["a", "b"]),
info={"File Path": "/home/pewpew/real.npz"}))
widget = view.activeWidget()
dlg = widget.actionCalibration()
dlg.close()
dlg = widget.actionConfig()
dlg.close()
widget.actionDuplicate()
widget.actionCopyImage()
dlg = widget.actionExport()
dlg.close()
dlg = widget.actionSave()
dlg.close()
dlg = widget.actionStatistics()
dlg.close()
dlg = widget.actionSelectDialog()
dlg.close()
dlg = widget.actionColocal()
dlg.close()
widget.contextMenuEvent(
QtGui.QContextMenuEvent(QtGui.QContextMenuEvent.Mouse,
QtCore.QPoint(0, 0)))
# Test contextmenu
widget.graphics.mask = np.ones((10, 10), dtype=bool)
widget.contextMenuEvent(
QtGui.QContextMenuEvent(
QtGui.QContextMenuEvent.Mouse,
widget.graphics.mapFromScene(QtCore.QPointF(0, 0)),
))
widget.actionCopySelectionText()
widget.actionCropSelection()
dlg = widget.actionStatisticsSelection()
dlg.close()
dlg = widget.actionColocalSelection()
dlg.close()
def test_laser_view(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
laser = view.addLaser(Laser(rand_data(["A1", "B2", "C3"])))
qtbot.waitExposed(laser)
view.tabs.setTabText(0, "newname")
assert view.stack.widget(0).laserName() == "newname"
view.contextMenuEvent(
QtGui.QContextMenuEvent(QtGui.QContextMenuEvent.Mouse,
QtCore.QPoint(0, 0)))
# Drop event
drag_mime = QtCore.QMimeData()
path = Path(__file__).parent.joinpath("data", "io", "npz", "test.npz")
drag_mime.setUrls([QtCore.QUrl.fromLocalFile(str(path.resolve()))])
drag_event = QtGui.QDragEnterEvent(
QtCore.QPoint(0, 0),
QtCore.Qt.CopyAction,
drag_mime,
QtCore.Qt.LeftButton,
QtCore.Qt.NoModifier,
)
view.dragEnterEvent(drag_event)
assert drag_event.isAccepted()
drop_event = QtGui.QDropEvent(
QtCore.QPoint(0, 0),
QtCore.Qt.CopyAction,
drag_mime,
QtCore.Qt.LeftButton,
QtCore.Qt.NoModifier,
)
with qtbot.waitSignal(view.numTabsChanged):
view.dropEvent(drop_event)
assert drop_event.isAccepted()
assert len(view.widgets()) == 2
dlg = view.actionOpen()
dlg.show()
dlg.close()
def test_merge_tool(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
data = rand_data(["A", "B"])
view.addLaser(Laser(data, info={"Name": "Laser 1", "File Path": "/test/laser1"}))
data = rand_data(["B", "C"])
view.addLaser(Laser(data, info={"Name": "Laser 2", "File Path": "/test/laser1"}))
tool = MergeTool(view.activeWidget())
view.addTab("Tool", tool)
qtbot.waitExposed(tool)
assert tool.list.count() == 1
# Test load via dialog
dlg = tool.addLaserDialog()
assert dlg.comboBoxItems() == ["Laser 2"]
dlg.accept()
assert tool.list.count() == 2
assert [row.offset() == (0, 0) for row in tool.list.rows]
assert tool.list.rows[0].combo_element.currentText() == "A"
assert tool.list.rows[1].combo_element.currentText() == "B"
tool.action_align_horz.trigger()
assert tool.list.rows[0].offset() == (0, 0)
assert tool.list.rows[1].offset() == (0, 10)
tool.action_align_vert.trigger()
assert tool.list.rows[0].offset() == (0, 0)
assert tool.list.rows[1].offset() == (10, 0)
tool.action_align_auto.trigger()
assert tool.list.rows[0].offset() == (0, 0)
# Second image position unknown
tool.apply()
def test_laser_view_space_apply_dialogs(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
qtbot.addWidget(viewspace)
viewspace.views[0].addLaser(Laser(rand_data("A1")))
viewspace.refresh()
dlg = viewspace.configDialog()
dlg.applyPressed.emit(dlg)
dlg.close()
dlg = viewspace.colortableRangeDialog()
dlg.applyPressed.emit(dlg)
dlg.close()
dlg = viewspace.fontsizeDialog()
dlg.intValueSelected.emit(5)
dlg.close()
assert viewspace.options.font.pointSize() == 5
def test_tool_filter(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
view.addLaser(Laser(rand_data(["a"])))
tool = FilteringTool(view.activeWidget())
view.addTab("Tool", tool)
qtbot.waitExposed(tool)
tool.combo_filter.setCurrentText("Mean")
tool.combo_filter.activated.emit(0)
tool.lineedit_fparams[0].setText("3.0")
tool.lineedit_fparams[1].setText("3.0")
tool.lineedit_fparams[0].editingFinished.emit()
assert np.all(tool.fparams == [3.0, 3.0])
assert tool.isComplete()
tool.lineedit_fparams[0].setText("5.0")
tool.lineedit_fparams[0].editingFinished.emit()
assert np.all(tool.fparams == [5.0, 3.0])
assert tool.isComplete()
tool.combo_filter.setCurrentText("Median")
tool.combo_filter.activated.emit(0)
assert np.all(tool.fparams == [5.0, 3.0])
assert tool.isComplete()
tool.lineedit_fparams[0].setText("4.0")
tool.lineedit_fparams[0].editingFinished.emit()
assert not tool.isComplete()
tool.combo_filter.setCurrentText("Mean")
tool.combo_filter.activated.emit(0)
assert np.all(tool.fparams == [5.0, 3.0])
assert tool.isComplete()
tool.apply()
def test_tool_widget(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
widget = view.addLaser(Laser(rand_data("A1"), info={"Name": "Widget"}))
tool = ToolWidget(widget, apply_all=True)
index = widget.index
widget.view.removeTab(index)
widget.view.insertTab(index, "Tool", tool)
qtbot.waitExposed(tool)
tool.requestClose()
view.tabs.tabText(index) == "Widget"
with qtbot.wait_signal(tool.applyPressed):
button = tool.button_box.button(QtWidgets.QDialogButtonBox.Apply)
button.click()
with qtbot.wait_signal(tool.applyPressed):
button = tool.button_box.button(QtWidgets.QDialogButtonBox.Ok)
button.click()
def test_overlay_tool(qtbot: QtBot):
data = np.zeros((10, 10), dtype=[("r", float), ("g", float), ("b", float)])
data["r"][:, :] = 1.0
data["g"][:10, :] = 1.0
data["b"][:, :10] = 1.0
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
view = viewspace.activeView()
view.addLaser(
Laser(data,
info={
"Name": "real",
"File Path": "/home/pewpew/real.npz"
}))
tool = OverlayTool(view.activeWidget())
view.addTab("Tool", tool)
qtbot.waitExposed(tool)
# Test rgb mode
assert tool.rows.color_model == "rgb"
tool.comboAdd(1) # r
assert np.all(tool.graphics.data[0, 0] == (255 << 24) + (255 << 16))
tool.comboAdd(2) # g
assert np.all(tool.graphics.data[:10] == (255 << 24) + (255 << 16) +
(255 << 8))
assert np.all(tool.graphics.data[10:] == (255 << 24) + (255 << 16))
tool.comboAdd(3) # g
assert np.all(tool.graphics.data[:10, :10] == (255 << 24) + (255 << 16) +
(255 << 8) + 255)
assert np.all(tool.graphics.data[10:, :10] == (255 << 24) + (255 << 16))
assert np.all(tool.graphics.data[10:, 10:] == (255 << 24) + (255 << 16) +
(255 << 8))
assert np.all(tool.graphics.data[10:, 10:] == (255 << 24) + (255 << 16))
# Test cmyk mode
tool.radio_cmyk.toggle()
assert tool.rows.color_model == "cmyk"
assert np.all(tool.graphics.data[:10, :10] == (255 << 24))
assert np.all(tool.graphics.data[10:, :10] == (255 << 24) + (255 << 8))
assert np.all(tool.graphics.data[10:, 10:] == (255 << 25) + 255)
assert np.all(tool.graphics.data[10:,
10:] == (255 << 24) + (255 << 8) + 255)
# Check that the rows are limited to 3
assert tool.rows.max_rows == 3
assert not tool.combo_add.isEnabled()
assert tool.rows.rowCount() == 3
with qtbot.assert_not_emitted(tool.rows.rowsChanged):
tool.addRow("r")
assert tool.rows.rowCount() == 3
# Check color buttons are not enabled
for row in tool.rows.rows:
assert not row.button_color.isEnabled()
# Test any mode
tool.radio_custom.toggle()
assert tool.rows.color_model == "any"
assert tool.combo_add.isEnabled()
assert tool.check_normalise.isEnabled()
for row in tool.rows.rows:
assert row.button_color.isEnabled()
tool.addRow("g")
tool.rows[3].setColor(QtGui.QColor.fromRgbF(0.0, 1.0, 1.0))
assert tool.rows.rowCount() == 4
# Test normalise
assert np.amin(tool.graphics.data) > (255 << 24)
tool.check_normalise.setChecked(True)
tool.refresh()
assert tool.graphics.data.min() == (255 << 24) + (255 << 8) + 255 # No red
tool.check_normalise.setChecked(False)
# Test export
dlg = tool.openExportDialog()
dlg2 = dlg.selectDirectory()
dlg2.close()
with tempfile.NamedTemporaryFile() as tf:
dlg.export(Path(tf.name))
assert Path(tf.name).exists()
with tempfile.TemporaryDirectory() as td:
dlg.lineedit_directory.setText(td)
dlg.lineedit_filename.setText("test.png")
dlg.check_individual.setChecked(True)
dlg.accept()
qtbot.wait(300)
assert Path(td).joinpath("test_1.png").exists()
assert Path(td).joinpath("test_2.png").exists()
assert Path(td).joinpath("test_3.png").exists()
# Test close
with qtbot.wait_signal(tool.rows.rowsChanged):
tool.rows.rows[-1].close()
assert tool.rows.rowCount() == 3
# Test hide
tool.radio_rgb.toggle()
with qtbot.wait_signal(tool.rows.rows[0].itemChanged):
tool.rows.rows[0].button_hide.click()
assert np.all(tool.graphics.data <= ((255 << 24) + (255 << 8) + 255))
dlg = tool.rows[0].selectColor()
dlg.close()
def test_laser_view_space(qtbot: QtBot):
viewspace = LaserViewSpace()
qtbot.addWidget(viewspace)
viewspace.show()
viewspace.toggleCalibrate(False)
viewspace.toggleColorbar(False)
viewspace.toggleLabel(False)
viewspace.toggleScalebar(False)
viewspace.toggleSmooth(False)
assert not viewspace.options.calibrate
assert not viewspace.options.items["colorbar"]
assert not viewspace.options.items["label"]
assert not viewspace.options.items["scalebar"]
assert not viewspace.options.smoothing
viewspace.splitActiveHorizontal()
assert viewspace.currentElement() is None
viewspace.views[0].addLaser(Laser(rand_data(["A1", "B2"])))
viewspace.views[0].addLaser(Laser(rand_data(["A1", "C3"])))
viewspace.views[1].addLaser(Laser(rand_data(["A1", "C3"])))
viewspace.views[1].addLaser(Laser(rand_data(["B2", "D4"])))
assert viewspace.uniqueElements() == ["A1", "B2", "C3", "D4"]
assert viewspace.currentElement() == "A1"
# Apply config
viewspace.applyConfig(Config(10, 10, 10))
for view in viewspace.views:
for widget in view.widgets():
assert widget.laser.config.spotsize == 10
assert widget.laser.config.speed == 10
assert widget.laser.config.scantime == 10
# Try to apply calibraiton
viewspace.applyCalibration({
"A1": Calibration(1.0, 1.0),
"B2": Calibration(2.0, 2.0)
})
qtbot.waitExposed(viewspace)
for view in viewspace.views:
for widget in view.widgets():
if "A1" in widget.laser.elements:
assert widget.laser.calibration["A1"].intercept == 1.0
assert widget.laser.calibration["A1"].gradient == 1.0
if "B2" in widget.laser.elements:
assert widget.laser.calibration["B2"].intercept == 2.0
assert widget.laser.calibration["B2"].gradient == 2.0
if "C3" in widget.laser.elements:
assert widget.laser.calibration["C3"].intercept == 0.0
assert widget.laser.calibration["C3"].gradient == 1.0
# Check element changed if avilable
assert viewspace.views[0].activeWidget().combo_element.currentText(
) == "A1"
assert viewspace.views[1].activeWidget().combo_element.currentText(
) == "A1"
viewspace.setCurrentElement("B2")
assert viewspace.views[0].activeWidget().combo_element.currentText(
) == "B2"
assert viewspace.views[1].activeWidget().combo_element.currentText(
) == "A1"
# Close all
for view in viewspace.views:
for widget in view.widgets():
widget.close()