mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 13:41:24 +00:00 
			
		
		
		
	 ce111caaeb
			
		
	
	
		ce111caaeb
		
	
	
	
	
		
			
			svn+ssh://pythondev@svn.python.org/python/branches/py3k ........ r84495 | antoine.pitrou | 2010-09-04 19:40:21 +0200 (sam., 04 sept. 2010) | 4 lines Issue #1100562: Fix deep-copying of objects derived from the list and dict types. Patch by Michele Orrù and Björn Lindqvist. ........ r84496 | antoine.pitrou | 2010-09-04 19:40:51 +0200 (sam., 04 sept. 2010) | 3 lines Fix Björn's name in ACKS. ........ r84497 | antoine.pitrou | 2010-09-04 19:46:44 +0200 (sam., 04 sept. 2010) | 3 lines Fix running the copy module from the command-line (however use{ful,less} it may be). ........
		
			
				
	
	
		
			718 lines
		
	
	
	
		
			21 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			718 lines
		
	
	
	
		
			21 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| """Unit tests for the copy module."""
 | |
| 
 | |
| import copy
 | |
| import copyreg
 | |
| import weakref
 | |
| from operator import le, lt, ge, gt, eq, ne
 | |
| 
 | |
| import unittest
 | |
| from test import support
 | |
| 
 | |
| order_comparisons = le, lt, ge, gt
 | |
| equality_comparisons = eq, ne
 | |
| comparisons = order_comparisons + equality_comparisons
 | |
| 
 | |
| class TestCopy(unittest.TestCase):
 | |
| 
 | |
|     # Attempt full line coverage of copy.py from top to bottom
 | |
| 
 | |
|     def test_exceptions(self):
 | |
|         self.assertTrue(copy.Error is copy.error)
 | |
|         self.assertTrue(issubclass(copy.Error, Exception))
 | |
| 
 | |
|     # The copy() method
 | |
| 
 | |
|     def test_copy_basic(self):
 | |
|         x = 42
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(x, y)
 | |
| 
 | |
|     def test_copy_copy(self):
 | |
|         class C(object):
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __copy__(self):
 | |
|                 return C(self.foo)
 | |
|         x = C(42)
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(y.__class__, x.__class__)
 | |
|         self.assertEqual(y.foo, x.foo)
 | |
| 
 | |
|     def test_copy_registry(self):
 | |
|         class C(object):
 | |
|             def __new__(cls, foo):
 | |
|                 obj = object.__new__(cls)
 | |
|                 obj.foo = foo
 | |
|                 return obj
 | |
|         def pickle_C(obj):
 | |
|             return (C, (obj.foo,))
 | |
|         x = C(42)
 | |
|         self.assertRaises(TypeError, copy.copy, x)
 | |
|         copyreg.pickle(C, pickle_C, C)
 | |
|         y = copy.copy(x)
 | |
| 
 | |
|     def test_copy_reduce_ex(self):
 | |
|         class C(object):
 | |
|             def __reduce_ex__(self, proto):
 | |
|                 return ""
 | |
|             def __reduce__(self):
 | |
|                 raise support.TestFailed("shouldn't call this")
 | |
|         x = C()
 | |
|         y = copy.copy(x)
 | |
|         self.assertTrue(y is x)
 | |
| 
 | |
|     def test_copy_reduce(self):
 | |
|         class C(object):
 | |
|             def __reduce__(self):
 | |
|                 return ""
 | |
|         x = C()
 | |
|         y = copy.copy(x)
 | |
|         self.assertTrue(y is x)
 | |
| 
 | |
|     def test_copy_cant(self):
 | |
|         class C(object):
 | |
|             def __getattribute__(self, name):
 | |
|                 if name.startswith("__reduce"):
 | |
|                     raise AttributeError(name)
 | |
|                 return object.__getattribute__(self, name)
 | |
|         x = C()
 | |
|         self.assertRaises(copy.Error, copy.copy, x)
 | |
| 
 | |
|     # Type-specific _copy_xxx() methods
 | |
| 
 | |
|     def test_copy_atomic(self):
 | |
|         class Classic:
 | |
|             pass
 | |
|         class NewStyle(object):
 | |
|             pass
 | |
|         def f():
 | |
|             pass
 | |
|         tests = [None, 42, 2**100, 3.14, True, False, 1j,
 | |
|                  "hello", "hello\u1234", f.__code__,
 | |
|                  NewStyle, range(10), Classic, max]
 | |
|         for x in tests:
 | |
|             self.assertTrue(copy.copy(x) is x, repr(x))
 | |
| 
 | |
|     def test_copy_list(self):
 | |
|         x = [1, 2, 3]
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_tuple(self):
 | |
|         x = (1, 2, 3)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_dict(self):
 | |
|         x = {"foo": 1, "bar": 2}
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_inst_vanilla(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C(42)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_inst_copy(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __copy__(self):
 | |
|                 return C(self.foo)
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C(42)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_inst_getinitargs(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __getinitargs__(self):
 | |
|                 return (self.foo,)
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C(42)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_inst_getstate(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __getstate__(self):
 | |
|                 return {"foo": self.foo}
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C(42)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_inst_setstate(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __setstate__(self, state):
 | |
|                 self.foo = state["foo"]
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C(42)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     def test_copy_inst_getstate_setstate(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __getstate__(self):
 | |
|                 return self.foo
 | |
|             def __setstate__(self, state):
 | |
|                 self.foo = state
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C(42)
 | |
|         self.assertEqual(copy.copy(x), x)
 | |
| 
 | |
|     # The deepcopy() method
 | |
| 
 | |
|     def test_deepcopy_basic(self):
 | |
|         x = 42
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
| 
 | |
|     def test_deepcopy_memo(self):
 | |
|         # Tests of reflexive objects are under type-specific sections below.
 | |
|         # This tests only repetitions of objects.
 | |
|         x = []
 | |
|         x = [x, x]
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y[0] is not x[0])
 | |
|         self.assertTrue(y[0] is y[1])
 | |
| 
 | |
|     def test_deepcopy_issubclass(self):
 | |
|         # XXX Note: there's no way to test the TypeError coming out of
 | |
|         # issubclass() -- this can only happen when an extension
 | |
|         # module defines a "type" that doesn't formally inherit from
 | |
|         # type.
 | |
|         class Meta(type):
 | |
|             pass
 | |
|         class C(metaclass=Meta):
 | |
|             pass
 | |
|         self.assertEqual(copy.deepcopy(C), C)
 | |
| 
 | |
|     def test_deepcopy_deepcopy(self):
 | |
|         class C(object):
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __deepcopy__(self, memo=None):
 | |
|                 return C(self.foo)
 | |
|         x = C(42)
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y.__class__, x.__class__)
 | |
|         self.assertEqual(y.foo, x.foo)
 | |
| 
 | |
|     def test_deepcopy_registry(self):
 | |
|         class C(object):
 | |
|             def __new__(cls, foo):
 | |
|                 obj = object.__new__(cls)
 | |
|                 obj.foo = foo
 | |
|                 return obj
 | |
|         def pickle_C(obj):
 | |
|             return (C, (obj.foo,))
 | |
|         x = C(42)
 | |
|         self.assertRaises(TypeError, copy.deepcopy, x)
 | |
|         copyreg.pickle(C, pickle_C, C)
 | |
|         y = copy.deepcopy(x)
 | |
| 
 | |
|     def test_deepcopy_reduce_ex(self):
 | |
|         class C(object):
 | |
|             def __reduce_ex__(self, proto):
 | |
|                 return ""
 | |
|             def __reduce__(self):
 | |
|                 raise support.TestFailed("shouldn't call this")
 | |
|         x = C()
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertTrue(y is x)
 | |
| 
 | |
|     def test_deepcopy_reduce(self):
 | |
|         class C(object):
 | |
|             def __reduce__(self):
 | |
|                 return ""
 | |
|         x = C()
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertTrue(y is x)
 | |
| 
 | |
|     def test_deepcopy_cant(self):
 | |
|         class C(object):
 | |
|             def __getattribute__(self, name):
 | |
|                 if name.startswith("__reduce"):
 | |
|                     raise AttributeError(name)
 | |
|                 return object.__getattribute__(self, name)
 | |
|         x = C()
 | |
|         self.assertRaises(copy.Error, copy.deepcopy, x)
 | |
| 
 | |
|     # Type-specific _deepcopy_xxx() methods
 | |
| 
 | |
|     def test_deepcopy_atomic(self):
 | |
|         class Classic:
 | |
|             pass
 | |
|         class NewStyle(object):
 | |
|             pass
 | |
|         def f():
 | |
|             pass
 | |
|         tests = [None, 42, 2**100, 3.14, True, False, 1j,
 | |
|                  "hello", "hello\u1234", f.__code__,
 | |
|                  NewStyle, range(10), Classic, max]
 | |
|         for x in tests:
 | |
|             self.assertTrue(copy.deepcopy(x) is x, repr(x))
 | |
| 
 | |
|     def test_deepcopy_list(self):
 | |
|         x = [[1, 2], 3]
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x[0] is not y[0])
 | |
| 
 | |
|     def test_deepcopy_reflexive_list(self):
 | |
|         x = []
 | |
|         x.append(x)
 | |
|         y = copy.deepcopy(x)
 | |
|         for op in comparisons:
 | |
|             self.assertRaises(RuntimeError, op, y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y[0] is y)
 | |
|         self.assertEqual(len(y), 1)
 | |
| 
 | |
|     def test_deepcopy_tuple(self):
 | |
|         x = ([1, 2], 3)
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x[0] is not y[0])
 | |
| 
 | |
|     def test_deepcopy_reflexive_tuple(self):
 | |
|         x = ([],)
 | |
|         x[0].append(x)
 | |
|         y = copy.deepcopy(x)
 | |
|         for op in comparisons:
 | |
|             self.assertRaises(RuntimeError, op, y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y[0] is not x[0])
 | |
|         self.assertTrue(y[0][0] is y)
 | |
| 
 | |
|     def test_deepcopy_dict(self):
 | |
|         x = {"foo": [1, 2], "bar": 3}
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x["foo"] is not y["foo"])
 | |
| 
 | |
|     def test_deepcopy_reflexive_dict(self):
 | |
|         x = {}
 | |
|         x['foo'] = x
 | |
|         y = copy.deepcopy(x)
 | |
|         for op in order_comparisons:
 | |
|             self.assertRaises(TypeError, op, y, x)
 | |
|         for op in equality_comparisons:
 | |
|             self.assertRaises(RuntimeError, op, y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y['foo'] is y)
 | |
|         self.assertEqual(len(y), 1)
 | |
| 
 | |
|     def test_deepcopy_keepalive(self):
 | |
|         memo = {}
 | |
|         x = 42
 | |
|         y = copy.deepcopy(x, memo)
 | |
|         self.assertTrue(memo[id(x)] is x)
 | |
| 
 | |
|     def test_deepcopy_inst_vanilla(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C([42])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_deepcopy_inst_deepcopy(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __deepcopy__(self, memo):
 | |
|                 return C(copy.deepcopy(self.foo, memo))
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C([42])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_deepcopy_inst_getinitargs(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __getinitargs__(self):
 | |
|                 return (self.foo,)
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C([42])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_deepcopy_inst_getstate(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __getstate__(self):
 | |
|                 return {"foo": self.foo}
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C([42])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_deepcopy_inst_setstate(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __setstate__(self, state):
 | |
|                 self.foo = state["foo"]
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C([42])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_deepcopy_inst_getstate_setstate(self):
 | |
|         class C:
 | |
|             def __init__(self, foo):
 | |
|                 self.foo = foo
 | |
|             def __getstate__(self):
 | |
|                 return self.foo
 | |
|             def __setstate__(self, state):
 | |
|                 self.foo = state
 | |
|             def __eq__(self, other):
 | |
|                 return self.foo == other.foo
 | |
|         x = C([42])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_deepcopy_reflexive_inst(self):
 | |
|         class C:
 | |
|             pass
 | |
|         x = C()
 | |
|         x.foo = x
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is y)
 | |
| 
 | |
|     # _reconstruct()
 | |
| 
 | |
|     def test_reconstruct_string(self):
 | |
|         class C(object):
 | |
|             def __reduce__(self):
 | |
|                 return ""
 | |
|         x = C()
 | |
|         y = copy.copy(x)
 | |
|         self.assertTrue(y is x)
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertTrue(y is x)
 | |
| 
 | |
|     def test_reconstruct_nostate(self):
 | |
|         class C(object):
 | |
|             def __reduce__(self):
 | |
|                 return (C, ())
 | |
|         x = C()
 | |
|         x.foo = 42
 | |
|         y = copy.copy(x)
 | |
|         self.assertTrue(y.__class__ is x.__class__)
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertTrue(y.__class__ is x.__class__)
 | |
| 
 | |
|     def test_reconstruct_state(self):
 | |
|         class C(object):
 | |
|             def __reduce__(self):
 | |
|                 return (C, (), self.__dict__)
 | |
|             def __eq__(self, other):
 | |
|                 return self.__dict__ == other.__dict__
 | |
|         x = C()
 | |
|         x.foo = [42]
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_reconstruct_state_setstate(self):
 | |
|         class C(object):
 | |
|             def __reduce__(self):
 | |
|                 return (C, (), self.__dict__)
 | |
|             def __setstate__(self, state):
 | |
|                 self.__dict__.update(state)
 | |
|             def __eq__(self, other):
 | |
|                 return self.__dict__ == other.__dict__
 | |
|         x = C()
 | |
|         x.foo = [42]
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(y, x)
 | |
|         self.assertTrue(y.foo is not x.foo)
 | |
| 
 | |
|     def test_reconstruct_reflexive(self):
 | |
|         class C(object):
 | |
|             pass
 | |
|         x = C()
 | |
|         x.foo = x
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertTrue(y is not x)
 | |
|         self.assertTrue(y.foo is y)
 | |
| 
 | |
|     # Additions for Python 2.3 and pickle protocol 2
 | |
| 
 | |
|     def test_reduce_4tuple(self):
 | |
|         class C(list):
 | |
|             def __reduce__(self):
 | |
|                 return (C, (), self.__dict__, iter(self))
 | |
|             def __eq__(self, other):
 | |
|                 return (list(self) == list(other) and
 | |
|                         self.__dict__ == other.__dict__)
 | |
|         x = C([[1, 2], 3])
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(x, y)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x[0] is y[0])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(x, y)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x[0] is not y[0])
 | |
| 
 | |
|     def test_reduce_5tuple(self):
 | |
|         class C(dict):
 | |
|             def __reduce__(self):
 | |
|                 return (C, (), self.__dict__, None, self.items())
 | |
|             def __eq__(self, other):
 | |
|                 return (dict(self) == dict(other) and
 | |
|                         self.__dict__ == other.__dict__)
 | |
|         x = C([("foo", [1, 2]), ("bar", 3)])
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(x, y)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x["foo"] is y["foo"])
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(x, y)
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x["foo"] is not y["foo"])
 | |
| 
 | |
|     def test_copy_slots(self):
 | |
|         class C(object):
 | |
|             __slots__ = ["foo"]
 | |
|         x = C()
 | |
|         x.foo = [42]
 | |
|         y = copy.copy(x)
 | |
|         self.assertTrue(x.foo is y.foo)
 | |
| 
 | |
|     def test_deepcopy_slots(self):
 | |
|         class C(object):
 | |
|             __slots__ = ["foo"]
 | |
|         x = C()
 | |
|         x.foo = [42]
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(x.foo, y.foo)
 | |
|         self.assertTrue(x.foo is not y.foo)
 | |
| 
 | |
|     def test_deepcopy_dict_subclass(self):
 | |
|         class C(dict):
 | |
|             def __init__(self, d=None):
 | |
|                 if not d:
 | |
|                     d = {}
 | |
|                 self._keys = list(d.keys())
 | |
|                 super().__init__(d)
 | |
|             def __setitem__(self, key, item):
 | |
|                 super().__setitem__(key, item)
 | |
|                 if key not in self._keys:
 | |
|                     self._keys.append(key)
 | |
|         x = C(d={'foo':0})
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(x, y)
 | |
|         self.assertEqual(x._keys, y._keys)
 | |
|         self.assertTrue(x is not y)
 | |
|         x['bar'] = 1
 | |
|         self.assertNotEqual(x, y)
 | |
|         self.assertNotEqual(x._keys, y._keys)
 | |
| 
 | |
|     def test_copy_list_subclass(self):
 | |
|         class C(list):
 | |
|             pass
 | |
|         x = C([[1, 2], 3])
 | |
|         x.foo = [4, 5]
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(list(x), list(y))
 | |
|         self.assertEqual(x.foo, y.foo)
 | |
|         self.assertTrue(x[0] is y[0])
 | |
|         self.assertTrue(x.foo is y.foo)
 | |
| 
 | |
|     def test_deepcopy_list_subclass(self):
 | |
|         class C(list):
 | |
|             pass
 | |
|         x = C([[1, 2], 3])
 | |
|         x.foo = [4, 5]
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(list(x), list(y))
 | |
|         self.assertEqual(x.foo, y.foo)
 | |
|         self.assertTrue(x[0] is not y[0])
 | |
|         self.assertTrue(x.foo is not y.foo)
 | |
| 
 | |
|     def test_copy_tuple_subclass(self):
 | |
|         class C(tuple):
 | |
|             pass
 | |
|         x = C([1, 2, 3])
 | |
|         self.assertEqual(tuple(x), (1, 2, 3))
 | |
|         y = copy.copy(x)
 | |
|         self.assertEqual(tuple(y), (1, 2, 3))
 | |
| 
 | |
|     def test_deepcopy_tuple_subclass(self):
 | |
|         class C(tuple):
 | |
|             pass
 | |
|         x = C([[1, 2], 3])
 | |
|         self.assertEqual(tuple(x), ([1, 2], 3))
 | |
|         y = copy.deepcopy(x)
 | |
|         self.assertEqual(tuple(y), ([1, 2], 3))
 | |
|         self.assertTrue(x is not y)
 | |
|         self.assertTrue(x[0] is not y[0])
 | |
| 
 | |
|     def test_getstate_exc(self):
 | |
|         class EvilState(object):
 | |
|             def __getstate__(self):
 | |
|                 raise ValueError("ain't got no stickin' state")
 | |
|         self.assertRaises(ValueError, copy.copy, EvilState())
 | |
| 
 | |
|     def test_copy_function(self):
 | |
|         self.assertEqual(copy.copy(global_foo), global_foo)
 | |
|         def foo(x, y): return x+y
 | |
|         self.assertEqual(copy.copy(foo), foo)
 | |
|         bar = lambda: None
 | |
|         self.assertEqual(copy.copy(bar), bar)
 | |
| 
 | |
|     def test_deepcopy_function(self):
 | |
|         self.assertEqual(copy.deepcopy(global_foo), global_foo)
 | |
|         def foo(x, y): return x+y
 | |
|         self.assertEqual(copy.deepcopy(foo), foo)
 | |
|         bar = lambda: None
 | |
|         self.assertEqual(copy.deepcopy(bar), bar)
 | |
| 
 | |
|     def _check_weakref(self, _copy):
 | |
|         class C(object):
 | |
|             pass
 | |
|         obj = C()
 | |
|         x = weakref.ref(obj)
 | |
|         y = _copy(x)
 | |
|         self.assertTrue(y is x)
 | |
|         del obj
 | |
|         y = _copy(x)
 | |
|         self.assertTrue(y is x)
 | |
| 
 | |
|     def test_copy_weakref(self):
 | |
|         self._check_weakref(copy.copy)
 | |
| 
 | |
|     def test_deepcopy_weakref(self):
 | |
|         self._check_weakref(copy.deepcopy)
 | |
| 
 | |
|     def _check_copy_weakdict(self, _dicttype):
 | |
|         class C(object):
 | |
|             pass
 | |
|         a, b, c, d = [C() for i in range(4)]
 | |
|         u = _dicttype()
 | |
|         u[a] = b
 | |
|         u[c] = d
 | |
|         v = copy.copy(u)
 | |
|         self.assertFalse(v is u)
 | |
|         self.assertEqual(v, u)
 | |
|         self.assertEqual(v[a], b)
 | |
|         self.assertEqual(v[c], d)
 | |
|         self.assertEqual(len(v), 2)
 | |
|         del c, d
 | |
|         self.assertEqual(len(v), 1)
 | |
|         x, y = C(), C()
 | |
|         # The underlying containers are decoupled
 | |
|         v[x] = y
 | |
|         self.assertNotIn(x, u)
 | |
| 
 | |
|     def test_copy_weakkeydict(self):
 | |
|         self._check_copy_weakdict(weakref.WeakKeyDictionary)
 | |
| 
 | |
|     def test_copy_weakvaluedict(self):
 | |
|         self._check_copy_weakdict(weakref.WeakValueDictionary)
 | |
| 
 | |
|     def test_deepcopy_weakkeydict(self):
 | |
|         class C(object):
 | |
|             def __init__(self, i):
 | |
|                 self.i = i
 | |
|         a, b, c, d = [C(i) for i in range(4)]
 | |
|         u = weakref.WeakKeyDictionary()
 | |
|         u[a] = b
 | |
|         u[c] = d
 | |
|         # Keys aren't copied, values are
 | |
|         v = copy.deepcopy(u)
 | |
|         self.assertNotEqual(v, u)
 | |
|         self.assertEqual(len(v), 2)
 | |
|         self.assertFalse(v[a] is b)
 | |
|         self.assertFalse(v[c] is d)
 | |
|         self.assertEqual(v[a].i, b.i)
 | |
|         self.assertEqual(v[c].i, d.i)
 | |
|         del c
 | |
|         self.assertEqual(len(v), 1)
 | |
| 
 | |
|     def test_deepcopy_weakvaluedict(self):
 | |
|         class C(object):
 | |
|             def __init__(self, i):
 | |
|                 self.i = i
 | |
|         a, b, c, d = [C(i) for i in range(4)]
 | |
|         u = weakref.WeakValueDictionary()
 | |
|         u[a] = b
 | |
|         u[c] = d
 | |
|         # Keys are copied, values aren't
 | |
|         v = copy.deepcopy(u)
 | |
|         self.assertNotEqual(v, u)
 | |
|         self.assertEqual(len(v), 2)
 | |
|         (x, y), (z, t) = sorted(v.items(), key=lambda pair: pair[0].i)
 | |
|         self.assertFalse(x is a)
 | |
|         self.assertEqual(x.i, a.i)
 | |
|         self.assertTrue(y is b)
 | |
|         self.assertFalse(z is c)
 | |
|         self.assertEqual(z.i, c.i)
 | |
|         self.assertTrue(t is d)
 | |
|         del x, y, z, t
 | |
|         del d
 | |
|         self.assertEqual(len(v), 1)
 | |
| 
 | |
|     def test_deepcopy_bound_method(self):
 | |
|         class Foo(object):
 | |
|             def m(self):
 | |
|                 pass
 | |
|         f = Foo()
 | |
|         f.b = f.m
 | |
|         g = copy.deepcopy(f)
 | |
|         self.assertEqual(g.m, g.b)
 | |
|         self.assertTrue(g.b.__self__ is g)
 | |
|         g.b()
 | |
| 
 | |
| 
 | |
| def global_foo(x, y): return x+y
 | |
| 
 | |
| def test_main():
 | |
|     support.run_unittest(TestCopy)
 | |
| 
 | |
| if __name__ == "__main__":
 | |
|     test_main()
 |