Add str and repr for afw Record

lsst · Aug 2, 2017 · 5c8238e · 5c8238e
1 parent 7296eb9
commit 5c8238e
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/python/lsst/afw/table/base/baseContinued.py b/python/lsst/afw/table/base/baseContinued.py
@@ -89,6 +89,13 @@ def extract(self, *patterns, **kwds):
                 d[name] = self.get(schemaItem.key)
         return d
 
+    def __str__(self):
+        return '\n'.join("%s: %s"%(x.field.getName(), self.extract("*")[x.field.getName()])
+                         for x in self.schema)
+
+    def __repr__(self):
+        return "%s\n%s" % (type(self), str(self))
+
 
 class Catalog(with_metaclass(TemplateMeta, object)):
 

diff --git a/tests/test_sourceTable.py b/tests/test_sourceTable.py
@@ -646,6 +646,19 @@ def testStrNonContiguous(self):
         for field in ('id', 'coord_ra', 'coord_dec'):
             self.assertIn(field, string)
 
+    def testRecordStr(self):
+        """Test that str(record) contains expected things."""
+        string = str(self.catalog[0])
+        for field in ('id: 50', 'coord_ra: nan', 'coord_dec: nan'):
+            self.assertIn(field, string)
+
+    def testRecordRepr(self):
+        """Test that repr(record) contains expected things."""
+        string = repr(self.catalog[0])
+        self.assertIn(str(type(self.catalog[0])), string)
+        for field in ('id: 50', 'coord_ra: nan', 'coord_dec: nan'):
+            self.assertIn(field, string)
+
 
 class MemoryTester(lsst.utils.tests.MemoryTestCase):
     pass