From 4f10bd8a797653485de28574655e9e6bb7ac0a59 2012-07-14 17:32:33
From: Bradley M. Froehle <brad.froehle@gmail.com>
Date: 2012-07-14 17:32:33
Subject: [PATCH] Merge pull request #2123 from rkern/fix-pretty

BUG: Look up the `_repr_pretty_` method on the class using the MRO
---

diff --git a/IPython/lib/pretty.py b/IPython/lib/pretty.py
index a06743c..c4cd9b0 100644
--- a/IPython/lib/pretty.py
+++ b/IPython/lib/pretty.py
@@ -353,8 +353,8 @@ class RepresentationPrinter(PrettyPrinter):
                         # Some objects automatically create any requested
                         # attribute. Try to ignore most of them by checking for
                         # callability.
-                        if '_repr_pretty_' in obj_class.__dict__:
-                            meth = obj_class._repr_pretty_
+                        if '_repr_pretty_' in cls.__dict__:
+                            meth = cls._repr_pretty_
                             if callable(meth):
                                 return meth(obj, self, cycle)
             return _default_pprint(obj, self, cycle)
diff --git a/IPython/lib/tests/test_pretty.py b/IPython/lib/tests/test_pretty.py
index 9b65bbf..240ded2 100644
--- a/IPython/lib/tests/test_pretty.py
+++ b/IPython/lib/tests/test_pretty.py
@@ -45,6 +45,14 @@ class MyDict(dict):
         p.text("MyDict(...)")
 
 
+class Dummy1(object):
+    def _repr_pretty_(self, p, cycle):
+        p.text("Dummy1(...)")
+
+class Dummy2(Dummy1):
+    _repr_pretty_ = None
+
+
 def test_indentation():
     """Test correct indentation in groups"""
     count = 40
@@ -63,3 +71,14 @@ def test_dispatch():
     expectedoutput = "MyDict(...)"
 
     nt.assert_equals(gotoutput, expectedoutput)
+
+
+def test_callability_checking():
+    """
+    Test that the _repr_pretty_ method is tested for callability and skipped if
+    not.
+    """
+    gotoutput = pretty.pretty(Dummy2())
+    expectedoutput = "Dummy1(...)"
+
+    nt.assert_equals(gotoutput, expectedoutput)