Merge pull request #4659 from aturanjanin/owheatmap

HeatMap: data info displayed in the status bar

Author: VesnaT

Orange/widgets/visualize/owheatmap.py

@@ -36,6 +36,7 @@
 from Orange.widgets.visualize.utils.heatmap import HeatmapGridWidget, \
     ColorMap, CategoricalColorMap, GradientColorMap
 from Orange.widgets.utils.widgetpreview import WidgetPreview
+from Orange.widgets.utils.state_summary import format_summary_details
 
 
 __all__ = []
@@ -234,6 +235,9 @@ def _():
         self.__rows_cache = {}
         self.__columns_cache = {}
 
+        self.info.set_input_summary(self.info.NoInput)
+        self.info.set_output_summary(self.info.NoOutput)
+
         # GUI definition
         colorbox = gui.vBox(self.controlArea, "Color")
         self.color_cb = gui.palette_combo_box(self.palette_name)
@@ -492,6 +496,7 @@ def set_dataset(self, data=None):
         self.closeContext()
         self.clear()
         self.clear_messages()
+        self._set_input_summary(data)
 
         if isinstance(data, SqlTable):
             if data.approx_len() < 4000:
@@ -563,6 +568,11 @@ def set_dataset(self, data=None):
 
         self.unconditional_commit()
 
+    def _set_input_summary(self, data):
+        summary = len(data) if data else self.info.NoInput
+        details = format_summary_details(data) if data else ""
+        self.info.set_input_summary(summary, details)
+
     def __on_split_rows_activated(self):
         self.set_split_variable(self.row_split_cb.currentData(Qt.EditRole))
 
@@ -1086,6 +1096,9 @@ def commit(self):
 
             data = self.input_data[indices]
 
+        summary = len(data) if data else self.info.NoOutput
+        details = format_summary_details(data) if data else ""
+        self.info.set_output_summary(summary, details)
         self.Outputs.selected_data.send(data)
         self.Outputs.annotated_data.send(create_annotated_table(self.input_data, indices))
 

Orange/widgets/visualize/tests/test_owheatmap.py

@@ -1,5 +1,5 @@
 # Test methods with long descriptive names can omit docstrings
-# pylint: disable=missing-docstring
+# pylint: disable=missing-docstring, protected-access
 import warnings
 import unittest
 from unittest.mock import patch
@@ -12,6 +12,7 @@
 from Orange.data import Table, Domain, ContinuousVariable, DiscreteVariable
 from Orange.preprocess import Continuize
 from Orange.widgets.utils import colorpalettes
+from Orange.widgets.utils.state_summary import format_summary_details
 from Orange.widgets.visualize.owheatmap import OWHeatMap, Clustering
 from Orange.widgets.tests.base import WidgetTest, WidgetOutputsTestMixin, datasets
 
@@ -273,6 +274,30 @@ def test_row_color_annotations(self):
         widget.set_annotation_color_var(None)
         self.assertFalse(widget.scene.widget.right_side_colors[0].isVisible())
 
+    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.housing
+        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)
+
 
 if __name__ == "__main__":
     unittest.main()