OWDBSCAN: data info displayed in the status bar

Author: aturanjanin

Orange/widgets/unsupervised/owdbscan.py

@@ -15,6 +15,7 @@
 from Orange.clustering import DBSCAN
 from Orange.widgets.utils.annotated_data import ANNOTATED_DATA_SIGNAL_NAME
 from Orange.widgets.utils.signals import Input, Output
+from Orange.widgets.utils.state_summary import format_summary_details
 from Orange.widgets.widget import Msg
 
 
@@ -89,6 +90,8 @@ def __init__(self):
         self.data_normalized = None
         self.db = None
         self.model = None
+        self._set_input_summary(None)
+        self._set_output_summary(None)
 
         box = gui.widgetBox(self.controlArea, "Parameters")
         gui.spin(box, self, "min_samples", 1, 100, 1,
@@ -169,10 +172,12 @@ def _compute_cut_point(self):
     @Inputs.data
     def set_data(self, data):
         self.Error.clear()
+        self._set_input_summary(data)
         if not self.check_data_size(data):
             data = None
         self.data = self.data_normalized = data
         if self.data is None:
+            self._set_output_summary(None)
             self.Outputs.annotated_data.send(None)
             self.plot.clear_plot()
             return
@@ -220,8 +225,19 @@ def send_data(self):
         domain = Domain(attributes, classes, meta_attrs)
         new_table = Table(domain, x, y, metas, self.data.W)
 
+        self._set_output_summary(new_table)
         self.Outputs.annotated_data.send(new_table)
 
+    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 _set_output_summary(self, output):
+        summary = len(output) if output else self.info.NoOutput
+        details = format_summary_details(output) if output else ""
+        self.info.set_output_summary(summary, details)
+
     def _invalidate(self):
         self.commit()
 

Orange/widgets/unsupervised/tests/test_owdbscan.py

@@ -1,3 +1,4 @@
+# pylint: disable=protected-access
 import numpy as np
 from scipy.sparse import csr_matrix, csc_matrix
 
@@ -6,6 +7,7 @@
 from Orange.widgets.tests.base import WidgetTest
 from Orange.widgets.tests.utils import simulate, possible_duplicate_table
 from Orange.widgets.unsupervised.owdbscan import OWDBSCAN, get_kth_distances
+from Orange.widgets.utils.state_summary import format_summary_details
 
 
 class TestOWDBSCAN(WidgetTest):
@@ -220,3 +222,23 @@ def test_missing_data(self):
         self.send_signal(w.Inputs.data, self.iris)
         output = self.get_output(w.Outputs.annotated_data)
         self.assertTupleEqual((150, 1), output[:, "Cluster"].metas.shape)
+
+    def test_summary(self):
+        """Check if the status bar updates when data on input"""
+        info = self.widget.info
+        no_input, no_output = "No data on input", "No data on output"
+
+        self.send_signal(self.widget.Inputs.data, self.iris)
+        summary, details = f"{len(self.iris)}", format_summary_details(self.iris)
+        self.assertEqual(info._StateInfo__input_summary.brief, summary)
+        self.assertEqual(info._StateInfo__input_summary.details, details)
+        output = self.get_output(self.widget.Outputs.annotated_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)