@@ -237,11 +237,16 @@ def __init__(self, widget_name, parent=None):
237
237
super (TestCentralWidget , self ).__init__ (parent )
238
238
self .widget_name = widget_name
239
239
self .plots = None
240
- self .widget_of_interest = self .parent ()
241
240
self .setLayout (QW .QVBoxLayout ())
242
241
self .options = TestOptions (self )
243
242
self .add_widget (self .options )
244
243
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
+
245
250
def add_widget (self , widget ):
246
251
"""Add new sub-widget"""
247
252
self .layout ().addWidget (widget )
@@ -257,8 +262,6 @@ def add_widget(self, widget):
257
262
plot_name , "Enable new flat style option" , plot .setFlatStyle
258
263
)
259
264
widget .setChecked (plot .flatStyle ())
260
- if len (self .plots ) == 1 :
261
- self .widget_of_interest = self .plots [0 ]
262
265
263
266
264
267
def take_screenshot (widget ):
@@ -297,7 +300,7 @@ def test_widget(widget_class, size=None, title=None, options=True):
297
300
central_widget = TestCentralWidget (widget_name , parent = window )
298
301
central_widget .add_widget (widget )
299
302
window .setCentralWidget (central_widget )
300
- widget_of_interest = central_widget .widget_of_interest
303
+ widget_of_interest = central_widget .get_widget_of_interest ()
301
304
else :
302
305
widget_of_interest = window
303
306
widget_of_interest .setObjectName (widget_name )
0 commit comments