OWRuleViewer: data info displayed in the status bar

Author: aturanjanin

Orange/widgets/visualize/owruleviewer.py

@@ -14,6 +14,7 @@
 from Orange.widgets.utils.annotated_data import (create_annotated_table,
                                                  ANNOTATED_DATA_SIGNAL_NAME)
 from Orange.widgets.utils.widgetpreview import WidgetPreview
+from Orange.widgets.utils.state_summary import format_summary_details
 from Orange.widgets.widget import Input, Output
 
 
@@ -45,6 +46,9 @@ def __init__(self):
         self.classifier = None
         self.selected = None
 
+        self.info.set_input_summary(self.info.NoInput)
+        self.info.set_output_summary(self.info.NoOutput)
+
         self.model = CustomRuleViewerTableModel(parent=self)
         self.model.set_horizontal_header_labels(
             ["IF conditions", "", "THEN class", "Distribution",
@@ -81,6 +85,7 @@ def __init__(self):
     @Inputs.data
     def set_data(self, data):
         self.data = data
+        self._set_input_summary()
         self.commit()
 
     @Inputs.classifier
@@ -101,6 +106,11 @@ def set_classifier(self, classifier):
         self.on_update()
         self.commit()
 
+    def _set_input_summary(self):
+        summary = len(self.data) if self.data else self.info.NoInput
+        details = format_summary_details(self.data) if self.data else ""
+        self.info.set_input_summary(summary, details)
+
     def on_update(self):
         self._save_selected()
 
@@ -168,6 +178,9 @@ def commit(self):
             data_output = data.from_table_rows(data, selected_indices) \
                 if len(selected_indices) else None
 
+        summary = len(data_output) if data_output else self.info.NoOutput
+        details = format_summary_details(data_output) if data_output else ""
+        self.info.set_output_summary(summary, details)
         self.Outputs.selected_data.send(data_output)
         self.Outputs.annotated_data.send(create_annotated_table(data, selected_indices))
 

Orange/widgets/visualize/tests/test_owruleviewer.py

@@ -6,6 +6,7 @@
 from Orange.widgets.tests.base import WidgetTest, WidgetOutputsTestMixin
 from Orange.classification import CN2Learner
 from Orange.widgets.visualize.owruleviewer import OWRuleViewer
+from Orange.widgets.utils.state_summary import format_summary_details
 
 
 class TestOWRuleViewer(WidgetTest, WidgetOutputsTestMixin):
@@ -156,3 +157,41 @@ def _select_data(self):
         selection_model.select(self.widget.proxy_model.index(2, 0),
                                selection_model.Select | selection_model.Rows)
         return list(range(586, 597))
+
+    def test_summary(self):
+        """Check if status bar is updated when data is received"""
+        info = self.widget.info
+        no_input, no_output = "No data on input", "No data on output"
+
+        data = self.titanic
+        self.send_signal(self.widget.Inputs.data, data)
+        summary, details = f"{len(data)}", format_summary_details(data)
+        self.assertEqual(info._StateInfo__input_summary.brief, summary)
+        self.assertEqual(info._StateInfo__input_summary.details, details)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+
+        self.send_signal(self.widget.Inputs.classifier, self.classifier)
+        summary, details = f"{len(data)}", format_summary_details(data)
+        self.assertEqual(info._StateInfo__input_summary.brief, summary)
+        self.assertEqual(info._StateInfo__input_summary.details, details)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+        self._select_data()
+        output = self.get_output(self.widget.Outputs.selected_data)
+        summary, details = f"{len(output)}", format_summary_details(output)
+        self.assertEqual(info._StateInfo__output_summary.brief, summary)
+        self.assertEqual(info._StateInfo__output_summary.details, details)
+
+        self.send_signal(self.widget.Inputs.data, None)
+        self.assertEqual(info._StateInfo__input_summary.brief, "")
+        self.assertEqual(info._StateInfo__input_summary.details, no_input)
+        summary, details = f"{len(output)}", format_summary_details(output)
+        self.assertEqual(info._StateInfo__output_summary.brief, summary)
+        self.assertEqual(info._StateInfo__output_summary.details, details)
+
+        self.send_signal(self.widget.Inputs.classifier, None)
+        self.assertEqual(info._StateInfo__input_summary.brief, "")
+        self.assertEqual(info._StateInfo__input_summary.details, no_input)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)