Call create_ui in another test where the UI is not closed

enthought · Jun 1, 2020 · 5217ebf · 5217ebf
1 parent 2d9c4a7
commit 5217ebf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/traitsui/tests/test_labels.py b/traitsui/tests/test_labels.py
@@ -133,10 +133,9 @@ def test_qt_show_labels_right_without_colon(self):
         # that are shown to the *right* of the corresponding elements
 
         from pyface import qt
-
-        with store_exceptions_on_all_threads():
-            dialog = ShowRightLabelsDialog()
-            ui = dialog.edit_traits()
+        dialog = ShowRightLabelsDialog()
+        with store_exceptions_on_all_threads(), \
+                create_ui(dialog) as ui:
 
             # get reference to label objects
             labels = ui.control.findChildren(qt.QtGui.QLabel)