Heat map: Cannot cluster a single instance

biolab · Aug 22, 2019 · ba05484 · ba05484
1 parent bf6edbe
commit ba05484
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/Orange/widgets/visualize/owheatmap.py b/Orange/widgets/visualize/owheatmap.py
@@ -320,9 +320,9 @@ def __new__(cls, title, indices, sortindices, cluster=None,
     @property
     def is_empty(self):
         if isinstance(self.indices, slice):
-            return (self.indices.stop - self.indices.start) == 0
+            return (self.indices.stop - self.indices.start) < 2
         else:
-            return len(self.indices) == 0
+            return len(self.indices) < 2
 
     @property
     def cluster_ord(self):

diff --git a/Orange/widgets/visualize/tests/test_owheatmap.py b/Orange/widgets/visualize/tests/test_owheatmap.py
@@ -5,7 +5,7 @@
 import numpy as np
 from sklearn.exceptions import ConvergenceWarning
 
-from Orange.data import Table, Domain, ContinuousVariable
+from Orange.data import Table, Domain, ContinuousVariable, DiscreteVariable
 from Orange.preprocess import Continuize
 from Orange.widgets.visualize.owheatmap import OWHeatMap
 from Orange.widgets.tests.base import WidgetTest, WidgetOutputsTestMixin, datasets
@@ -168,3 +168,10 @@ def test_use_enough_colors(self):
             colors[r] = c.red(), c.green(), c.blue()
         unique_colors = len(np.unique(colors, axis=0))
         self.assertLessEqual(len(data)*self.widget.threshold_low, unique_colors)
+
+    def test_cls_with_single_instance(self):
+        table = Table(Domain([ContinuousVariable("c1")],
+                             [DiscreteVariable("c2", values=["a", "b"])]),
+                      np.array([[1], [2], [3]]), np.array([[0], [0], [1]]))
+        self.send_signal(self.widget.Inputs.data, table)
+        self.widget.row_check.setChecked(True)