Skip to content

Commit 56a94b5

Browse files
committed
Fix segfault in test_multidemo
1 parent 74dcfed commit 56a94b5

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

qwt/tests/utils.py

+7-4
Original file line numberDiff line numberDiff line change
@@ -237,11 +237,16 @@ def __init__(self, widget_name, parent=None):
237237
super(TestCentralWidget, self).__init__(parent)
238238
self.widget_name = widget_name
239239
self.plots = None
240-
self.widget_of_interest = self.parent()
241240
self.setLayout(QW.QVBoxLayout())
242241
self.options = TestOptions(self)
243242
self.add_widget(self.options)
244243

244+
def get_widget_of_interest(self):
245+
"""Return widget of interest"""
246+
if self.plots is not None and len(self.plots) == 1:
247+
return self.plots[0]
248+
return self.parent()
249+
245250
def add_widget(self, widget):
246251
"""Add new sub-widget"""
247252
self.layout().addWidget(widget)
@@ -257,8 +262,6 @@ def add_widget(self, widget):
257262
plot_name, "Enable new flat style option", plot.setFlatStyle
258263
)
259264
widget.setChecked(plot.flatStyle())
260-
if len(self.plots) == 1:
261-
self.widget_of_interest = self.plots[0]
262265

263266

264267
def take_screenshot(widget):
@@ -297,7 +300,7 @@ def test_widget(widget_class, size=None, title=None, options=True):
297300
central_widget = TestCentralWidget(widget_name, parent=window)
298301
central_widget.add_widget(widget)
299302
window.setCentralWidget(central_widget)
300-
widget_of_interest = central_widget.widget_of_interest
303+
widget_of_interest = central_widget.get_widget_of_interest()
301304
else:
302305
widget_of_interest = window
303306
widget_of_interest.setObjectName(widget_name)

0 commit comments

Comments
 (0)