fixed str(la) where la.shape[0] == 0

gdementen · gdementen · commit a3a1f6e63502 · 2015-01-28T16:20:58.000+01:00
diff --git a/larray/core.py b/larray/core.py
@@ -1082,6 +1082,8 @@ def __setslice__(self, i, j, value):
     def __str__(self):
         if not self.ndim:
             return str(np.asscalar(self))
+        elif not len(self):
+            return 'LArray([])'
         else:
             s = table2str(list(self.as_table()), 'nan', True,
                           keepcols=self.ndim - 1)
diff --git a/larray/tests/test_la.py b/larray/tests/test_la.py
@@ -535,6 +535,9 @@ def test_str(self):
         # zero dimension / scalar
         self.assertEqual(str(self.small[lipro['P01'], sex['F']]), "15")
 
+        # empty / len 0 first dimension
+        self.assertEqual(str(self.small[sex[[]]]), "LArray([])")
+
         # one dimension
         self.assertEqual(str(self.small[lipro3, sex['H']]), """
 lipro | P01 | P02 | P03
