Merge pull request #1951 from nikicc/repr_on_small_sparse

[FIX] Table.__repr__: Fix for sparse data with < 5 instances
(cherry picked from commit 3878b7b)

Author: lanzagar

Orange/data/table.py

@@ -802,8 +802,11 @@ def __str__(self):
         return "[" + ",\n ".join(str(ex) for ex in self)
 
     def __repr__(self):
-        s = "[" + ",\n ".join(repr(ex) for ex in self[:5])
-        if len(self) > 5:
+        head = 5
+        if self.is_sparse():
+            head = min(self.X.shape[0], head)
+        s = "[" + ",\n ".join(repr(ex) for ex in self[:head])
+        if len(self) > head:
             s += ",\n ..."
         s += "\n]"
         return s

Orange/tests/test_table.py

@@ -1213,6 +1213,11 @@ def test_is_sparse(self):
         table.X = sp.csr_matrix(table.X)
         self.assertTrue(table.is_sparse())
 
+    def test_repr_sparse_with_one_row(self):
+        table = data.Table("iris")[:1]
+        table.X = sp.csr_matrix(table.X)
+        repr(table)     # make sure repr does not crash
+
 
 def column_sizes(table):
     return (len(table.domain.attributes),