Merge pull request #552 from GavinHuttley/develop

ENH: added str, repr dunder methods for table.Columns

src/cogent3/util/table.py

@@ -430,6 +430,18 @@ def __setstate__(self, data):
 
         self.__dict__.update(new.__dict__)
 
+    def __repr__(self):
+        d = [f"'{c}': {v.dtype}" for c, v in self.items()]
+        num = len(d)
+        v = d[:5]
+        if num > 5:
+            v.append(f"... + {num - 5} more")
+        txt = f"{self.__class__.__name__}({', '.join(v)})"
+        return txt
+
+    def __str__(self):
+        return repr(self)
+
     def iter_rows(self):
         columns = [self[c] for c in self]
         for row in zip(*columns):

tests/test_util/test_table.py

@@ -133,6 +133,14 @@ def test_index_name(self):
         with self.assertRaises(ValueError):
             t.index_name
 
+    def test_column_repr_str(self):
+        """repr and str of Columns"""
+        t = Table(header=list("abcdefg"), data=[[0, 1.1, 2, 3, 4, 5.5, 6.0]])
+        r = repr(t.columns)
+        s = str(t.columns)
+        self.assertIsInstance(r, str)
+        self.assertEqual(r, s)
+
     def test_slicing_columns(self):
         """works using names, ints, bool array"""
         t = Table(header=self.t5_header, data=self.t5_rows)