@@ -488,9 +488,10 @@ class WidgetOutputsTestMixin:
488488 _compare_selected_annotated_domains.
489489 """
490490
491- def init (self ):
491+ def init (self , same_table_attributes = True ):
492492 self .data = Table ("iris" )
493493 self .same_input_output_domain = True
494+ self .same_table_attributes = same_table_attributes
494495
495496 def test_outputs (self , timeout = DEFAULT_TIMEOUT ):
496497 self .send_signal (self .signal_name , self .signal_data )
@@ -516,12 +517,14 @@ def test_outputs(self, timeout=DEFAULT_TIMEOUT):
516517 self .same_input_output_domain )
517518 np .testing .assert_array_equal (selected .X [:, :n_attr ],
518519 self .data .X [selected_indices ])
519- self .assertEqual (selected .attributes , self .data .attributes )
520+ if self .same_table_attributes :
521+ self .assertEqual (selected .attributes , self .data .attributes )
520522
521523 # check annotated data output
522524 annotated = self .get_output (ANNOTATED_DATA_SIGNAL_NAME )
523525 self .assertEqual (n_sel , np .sum ([i [feature_name ] for i in annotated ]))
524- self .assertEqual (annotated .attributes , self .data .attributes )
526+ if self .same_table_attributes :
527+ self .assertEqual (annotated .attributes , self .data .attributes )
525528
526529 # compare selected and annotated data domains
527530 self ._compare_selected_annotated_domains (selected , annotated )
0 commit comments