test: test_widget_io.py fixed

2025-07-14 11:41:49 +02:00 · 2023-11-08 11:20:40 +01:00
parent 3c28cf0e01
commit 679d3e1980
2 changed files with 6 additions and 8 deletions
--- a/tests/test_widget_io.py
+++ b/tests/test_widget_io.py
@ -0,0 +1,96 @@
+import pytest
+from PyQt5.QtWidgets import (
+    QWidget,
+    QVBoxLayout,
+    QLineEdit,
+    QComboBox,
+    QTableWidget,
+    QSpinBox,
+)
+
+from bec_widgets.qt_utils.widget_IO import WidgetHierarchy
+
+
+@pytest.fixture(scope="function")
+def example_widget(qtbot):
+    # Create a widget with a few child widgets
+    main_widget = QWidget()
+    layout = QVBoxLayout(main_widget)
+    line_edit = QLineEdit(main_widget)
+    combo_box = QComboBox(main_widget)
+    table_widget = QTableWidget(2, 2, main_widget)
+    spin_box = QSpinBox(main_widget)
+    layout.addWidget(line_edit)
+    layout.addWidget(combo_box)
+    layout.addWidget(table_widget)
+    layout.addWidget(spin_box)
+
+    # Add text items to the combo box
+    combo_box.addItems(["Option 1", "Option 2", "Option 3"])
+
+    qtbot.addWidget(main_widget)
+    qtbot.waitExposed(main_widget)
+    yield main_widget
+
+
+def test_export_import_config(example_widget):
+    initial_config = {
+        "QWidget ()": {
+            "QLineEdit ()": {"value": "New Text"},
+            "QComboBox ()": {"value": 1},
+            "QTableWidget ()": {"value": [["a", "b"], ["c", "d"]]},
+            "QSpinBox ()": {"value": 10},
+        }
+    }
+    WidgetHierarchy.import_config_from_dict(example_widget, initial_config, set_values=True)
+
+    exported_config_full = WidgetHierarchy.export_config_to_dict(example_widget, grab_values=True)
+    exported_config_reduced = WidgetHierarchy.export_config_to_dict(
+        example_widget, grab_values=True, save_all=False
+    )
+
+    expected_full = {
+        "QWidget ()": {
+            "QVBoxLayout ()": {},
+            "QLineEdit ()": {"value": "New Text", "QObject ()": {}},
+            "QComboBox ()": {"value": 1, "QStandardItemModel ()": {}},
+            "QTableWidget ()": {
+                "value": [["a", "b"], ["c", "d"]],
+                "QWidget (qt_scrollarea_viewport)": {},
+                "QStyledItemDelegate ()": {},
+                "QHeaderView ()": {
+                    "QWidget (qt_scrollarea_viewport)": {},
+                    "QWidget (qt_scrollarea_hcontainer)": {
+                        "QScrollBar ()": {},
+                        "QBoxLayout ()": {},
+                    },
+                    "QWidget (qt_scrollarea_vcontainer)": {
+                        "QScrollBar ()": {},
+                        "QBoxLayout ()": {},
+                    },
+                    "QItemSelectionModel ()": {},
+                },
+                "QAbstractButton ()": {},
+                "QAbstractTableModel ()": {},
+                "QItemSelectionModel ()": {},
+                "QWidget (qt_scrollarea_hcontainer)": {"QScrollBar ()": {}, "QBoxLayout ()": {}},
+                "QWidget (qt_scrollarea_vcontainer)": {"QScrollBar ()": {}, "QBoxLayout ()": {}},
+            },
+            "QSpinBox ()": {
+                "value": 10,
+                "QLineEdit (qt_spinbox_lineedit)": {"value": "10", "QObject ()": {}},
+                "QValidator (qt_spinboxvalidator)": {},
+            },
+        }
+    }
+    expected_reduced = {
+        "QWidget ()": {
+            "QLineEdit ()": {"value": "New Text"},
+            "QComboBox ()": {"value": 1},
+            "QTableWidget ()": {"value": [["a", "b"], ["c", "d"]]},
+            "QSpinBox ()": {"value": 10, "QLineEdit (qt_spinbox_lineedit)": {"value": "10"}},
+        }
+    }
+
+    assert exported_config_full == expected_full
+    assert exported_config_reduced == expected_reduced