From 5379d05dc3e81023847dc0afa0243479ae244f5d Mon Sep 17 00:00:00 2001 From: Fred Drake Date: Tue, 22 May 2001 20:38:44 +0000 Subject: Convert copy_reg test to PyUnit. --- Lib/test/test_copy_reg.py | 48 +++++++++++++++++++---------------------------- 1 file 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) -- cgit v0.12