@@ -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
264267def 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