diff options
| -rw-r--r-- | Lib/test/test_copy_reg.py | 48 |
1 files changed, 19 insertions, 29 deletions
diff --git a/Lib/test/test_copy_reg.py b/Lib/test/test_copy_reg.py index 0324d92..51ec60b 100644 --- a/Lib/test/test_copy_reg.py +++ b/Lib/test/test_copy_reg.py @@ -1,35 +1,25 @@ import copy_reg +import test_support +import unittest + class C: pass -try: - copy_reg.pickle(C, None, None) -except TypeError, e: - print "Caught expected TypeError:" - print e -else: - print "Failed to catch expected TypeError when registering a class type." - - -print -try: - copy_reg.pickle(type(1), "not a callable") -except TypeError, e: - print "Caught expected TypeError:" - print e -else: - print "Failed to catch TypeError " \ - "when registering a non-callable reduction function." - - -print -try: - copy_reg.pickle(type(1), int, "not a callable") -except TypeError, e: - print "Caught expected TypeError:" - print e -else: - print "Failed to catch TypeError " \ - "when registering a non-callable constructor." +class CopyRegTestCase(unittest.TestCase): + + def test_class(self): + self.assertRaises(TypeError, copy_reg.pickle, + C, None, None) + + def test_noncallable_reduce(self): + self.assertRaises(TypeError, copy_reg.pickle, + type(1), "not a callable") + + def test_noncallable_constructor(self): + self.assertRaises(TypeError, copy_reg.pickle, + type(1), int, "not a callable") + + +test_support.run_unittest(CopyRegTestCase) |