Merge pull request #4634 from aturanjanin/owmosaic

[ENH] MosaicDisplay: data info displayed in the status bar

Author: VesnaT

Orange/widgets/visualize/owmosaic.py

@@ -27,6 +27,7 @@
                                                  ANNOTATED_DATA_SIGNAL_NAME)
 from Orange.widgets.utils.itemmodels import DomainModel
 from Orange.widgets.utils.widgetpreview import WidgetPreview
+from Orange.widgets.utils.state_summary import format_summary_details
 from Orange.widgets.visualize.utils import (
     CanvasText, CanvasRectangle, ViewWithPress, VizRankDialog)
 from Orange.widgets.visualize.utils.plotutils import wrap_legend_items
@@ -332,6 +333,9 @@ def __init__(self):
         self.canvas_view.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
         self.canvas_view.setRenderHint(QPainter.Antialiasing)
 
+        self.info.set_input_summary(self.info.NoInput)
+        self.info.set_output_summary(self.info.NoOutput)
+
         box = gui.vBox(self.controlArea, box=True)
         self.model_1 = DomainModel(
             order=DomainModel.MIXED, valid_types=DomainModel.PRIMITIVE)
@@ -442,8 +446,10 @@ def set_data(self, data):
         if self.data is None:
             self.discrete_data = None
             self.init_combos(None)
+            self.info.set_input_summary(self.info.NoInput)
             return
 
+        self.info.set_input_summary(len(data),format_summary_details(data))
         self.init_combos(self.data)
         self.openContext(self.data)
 
@@ -518,6 +524,7 @@ def send_selection(self):
             self.Outputs.selected_data.send(None)
             self.Outputs.annotated_data.send(
                 create_annotated_table(self.data, []))
+            self.info.set_output_summary(self.info.NoOutput)
             return
         filters = []
         self.Warning.no_cont_selection_sql.clear()
@@ -539,6 +546,10 @@ def send_selection(self):
         sel_idx = [i for i, id in enumerate(self.data.ids) if id in idset]
         if self.discrete_data is not self.data:
             selection = self.data[sel_idx]
+
+        summary = len(selection) if selection else self.info.NoOutput
+        details = format_summary_details(selection) if selection else ""
+        self.info.set_output_summary(summary, details)
         self.Outputs.selected_data.send(selection)
         self.Outputs.annotated_data.send(
             create_annotated_table(self.data, sel_idx))
@@ -994,5 +1005,5 @@ def get_conditional_distribution(data, attrs):
 
 
 if __name__ == "__main__":  # pragma: no cover
-    data = Table("zoo")
-    WidgetPreview(OWMosaicDisplay).run(data, set_subset_data=data[::10])
+    dataset = Table("zoo")
+    WidgetPreview(OWMosaicDisplay).run(dataset, set_subset_data=dataset[::10])

Orange/widgets/visualize/tests/test_owmosaic.py

@@ -13,6 +13,7 @@
 from Orange.widgets.tests.base import WidgetTest, WidgetOutputsTestMixin
 from Orange.widgets.visualize.owmosaic import OWMosaicDisplay
 from Orange.widgets.tests.utils import simulate
+from Orange.widgets.utils.state_summary import format_summary_details
 
 
 class TestOWMosaicDisplay(WidgetTest, WidgetOutputsTestMixin):
@@ -184,6 +185,31 @@ def test_selection_setting(self):
         self.assertEqual(w.selection, {1})
         self.assertIsNotNone(self.get_output(w.Outputs.selected_data, widget=w))
 
+    def test_summary(self):
+        """Check if status bar is updated when data is received"""
+        info = self.widget.info
+        data = Table("zoo")
+        no_input, no_output = "No data on input", "No data on output"
+
+        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._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)
+        self.assertEqual(info._StateInfo__output_summary.brief, "")
+        self.assertEqual(info._StateInfo__output_summary.details, no_output)
+
 
 # Derive from WidgetTest to simplify creation of the Mosaic widget, although
 # we are actually testing the MosaicVizRank dialog and not the widget