Function test_DialogLoadMask_1

pyxrf/gui_module/tests/test_wd_load_data.py:13–57

Set/Read values from ROI widget group of DialogLoadMask object

(qtbot)

def test_DialogLoadMask_1(qtbot):
    """Set/Read values from ROI widget group of DialogLoadMask object"""
    dlg = DialogLoadMask()
    qtbot.addWidget(dlg)
    dlg.show()

    # Set image size
    n_rows, n_columns = 15, 17
    dlg.set_image_size(n_rows=n_rows, n_columns=n_columns)
    assert f"{n_rows} rows, {n_columns} columns" in dlg.label_map_size.text(), "Map size was not set correctly"

    def _check_roi_widgets(expected_values):
        l_edit_widgets = (dlg.le_roi_start_row, dlg.le_roi_start_col, dlg.le_roi_end_row, dlg.le_roi_end_col)
        for n, v in enumerate(expected_values):
            if n < 2:  # First 2 entries (are incremented when displayed)
                v += 1
            s = f"{v}" if v > 0 else ""
            assert l_edit_widgets[n].text() == s, "Line Edit text does not match the expected text"

    # Check if the default values were set correctly when image size was set
    #   (try to activate/deactivate the ROI selection group box)
    _check_roi_widgets([0, 0, n_rows, n_columns])  # Default values
    dlg.set_roi_active(True)  # Activate ROI group
    _check_roi_widgets([0, 0, n_rows, n_columns])  # Default values
    dlg.set_roi_active(False)
    _check_roi_widgets([0, 0, n_rows, n_columns])  # Default values

    # Set ROI. Initially the ROI group is inactive, so no text is displayed
    roi = (2, 3, 14, 15)
    dlg.set_roi(row_start=roi[0], column_start=roi[1], row_end=roi[2], column_end=roi[3])
    assert not dlg.gb_roi.isChecked(), "ROI group is checked, while it must be unchecked"
    assert dlg.get_roi_active() is False, "ROI 'enable' status was not read correctly"
    _check_roi_widgets(roi)
    # Activate ROI group
    dlg.set_roi_active(True)
    assert dlg.gb_roi.isChecked(), "ROI group is unchecked, while it must be checked"
    assert dlg.get_roi_active() is True, "ROI 'enable' status was not read correctly"
    _check_roi_widgets(roi)

    # Set the values outside the range. The values must be clipped
    roi = (0, 0, 50, 50)
    dlg.set_roi(row_start=roi[0], column_start=roi[1], row_end=roi[2], column_end=roi[3])
    assert dlg.get_roi() == (roi[0], roi[1], n_rows, n_columns), "Indexes were not clipped correctly"

    dlg.close()


def test_DialogLoadMask_2(qtbot):