Test for the safety check in wrap_cmpfunc().

diff --git a/Lib/test/test_descr.py b/Lib/test/test_descr.py
index dbfce0c28da1a83a35229add831485d8335f65c3..fc00318673765576aef619e07d649c0ec3ffb12d 100644 (file)
--- a/Lib/test/test_descr.py
+++ b/Lib/test/test_descr.py
@@ -1304,6 +1304,21 @@ def specials():
     for i in range(10):
         verify(i in p10)
     verify(10 not in p10)
+    # Safety test for __cmp__
+    def unsafecmp(a, b):
+        try:
+            a.__class__.__cmp__(a, b)
+        except TypeError:
+            pass
+        else:
+            raise TestFailed, "shouldn't allow %s.__cmp__(%r, %r)" % (
+                a.__class__, a, b)
+    unsafecmp(u"123", "123")
+    unsafecmp("123", u"123")
+    unsafecmp(1, 1.0)
+    unsafecmp(1.0, 1)
+    unsafecmp(1, 1L)
+    unsafecmp(1L, 1)
 
 def weakrefs():
     if verbose: print "Testing weak references..."