.

Author: markotoplak

Orange/widgets/data/tests/test_owtable.py

@@ -172,20 +172,30 @@ def test_whole_rows(self):
         w = self.widget
         self.send_signal(w.Inputs.data, self.data, 0)
         self.assertTrue(w.select_rows)  # default value
-        w.controls.select_rows.toggle()
+        with excepthook_catch():
+            w.controls.select_rows.toggle()
         self.assertFalse(w.select_rows)
         w.selected_cols = [0, 1]
         w.selected_rows = [0, 1, 2, 3]
         w.set_selection()
         out = self.get_output(w.Outputs.selected_data)
         self.assertEqual(out.domain,
                          Domain([self.data.domain.attributes[0]], self.data.domain.class_var))
-        w.controls.select_rows.toggle()
+        with excepthook_catch():
+            w.controls.select_rows.toggle()
         out = self.get_output(w.Outputs.selected_data)
         self.assertTrue(w.select_rows)
         self.assertEqual(out.domain,
                          self.data.domain)
 
+    def test_show_attribute_labels(self):
+        w = self.widget
+        self.send_signal(w.Inputs.data, self.data, 0)
+        self.assertTrue(w.show_attribute_labels)  # default value
+        with excepthook_catch():
+            w.controls.show_attribute_labels.toggle()
+        self.assertFalse(w.show_attribute_labels)
+
 
 if __name__ == "__main__":
     unittest.main()