Add tests for __nonzero__() problems.

jhylton · jhylton · commit 88bf26f529d2 · 2003-06-27T17:40:16.000Z
git-svn-id: http://svn.python.org/projects/python/trunk@33275 6015fed2-1504-0410-9fe1-9d1591cc4771
diff --git a/Lib/test/test_bool.py b/Lib/test/test_bool.py
@@ -320,6 +320,27 @@ def test_picklevalues(self):
         self.assertEqual(cPickle.dumps(True, True), "I01\n.")
         self.assertEqual(cPickle.dumps(False, True), "I00\n.")
 
+    def test_convert_to_bool(self):
+        # Verify that TypeError occurs when bad things are returned
+        # from __nonzero__().  This isn't really a bool test, but
+        # it's related.
+        check = lambda o: self.assertRaises(TypeError, bool, o)
+        class Foo(object):
+            def __nonzero__(self):
+                return self
+        check(Foo())
+        
+        class Bar(object):
+            def __nonzero__(self):
+                return "Yes"
+        check(Bar())
+        
+        class Baz(int):
+            def __nonzero__(self):
+                return self
+        check(Baz())
+        
+
 def test_main():
     test_support.run_unittest(BoolTest)
 
