mirror of
https://github.com/python/cpython.git
synced 2024-12-12 03:04:15 +08:00
805365ee39
svn+ssh://pythondev@svn.python.org/python/branches/p3yk ........ r55077 | guido.van.rossum | 2007-05-02 11:54:37 -0700 (Wed, 02 May 2007) | 2 lines Use the new print syntax, at least. ........ r55142 | fred.drake | 2007-05-04 21:27:30 -0700 (Fri, 04 May 2007) | 1 line remove old cruftiness ........ r55143 | fred.drake | 2007-05-04 21:52:16 -0700 (Fri, 04 May 2007) | 1 line make this work with the new Python ........ r55162 | neal.norwitz | 2007-05-06 22:29:18 -0700 (Sun, 06 May 2007) | 1 line Get asdl code gen working with Python 2.3. Should continue to work with 3.0 ........ r55164 | neal.norwitz | 2007-05-07 00:00:38 -0700 (Mon, 07 May 2007) | 1 line Verify checkins to p3yk (sic) branch go to 3000 list. ........ r55166 | neal.norwitz | 2007-05-07 00:12:35 -0700 (Mon, 07 May 2007) | 1 line Fix this test so it runs again by importing warnings_test properly. ........ r55167 | neal.norwitz | 2007-05-07 01:03:22 -0700 (Mon, 07 May 2007) | 8 lines So long xrange. range() now supports values that are outside -sys.maxint to sys.maxint. floats raise a TypeError. This has been sitting for a long time. It probably has some problems and needs cleanup. Objects/rangeobject.c now uses 4-space indents since it is almost completely new. ........ r55171 | guido.van.rossum | 2007-05-07 10:21:26 -0700 (Mon, 07 May 2007) | 4 lines Fix two tests that were previously depending on significant spaces at the end of a line (and before that on Python 2.x print behavior that has no exact equivalent in 3.0). ........
592 lines
17 KiB
Python
592 lines
17 KiB
Python
"""Unit tests for the copy module."""
|
|
|
|
import sys
|
|
import copy
|
|
import copy_reg
|
|
|
|
import unittest
|
|
from test import test_support
|
|
|
|
class TestCopy(unittest.TestCase):
|
|
|
|
# Attempt full line coverage of copy.py from top to bottom
|
|
|
|
def test_exceptions(self):
|
|
self.assert_(copy.Error is copy.error)
|
|
self.assert_(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)
|
|
copy_reg.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 test_support.TestFailed, "shouldn't call this"
|
|
x = C()
|
|
y = copy.copy(x)
|
|
self.assert_(y is x)
|
|
|
|
def test_copy_reduce(self):
|
|
class C(object):
|
|
def __reduce__(self):
|
|
return ""
|
|
x = C()
|
|
y = copy.copy(x)
|
|
self.assert_(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.assert_(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.assert_(y is not x)
|
|
self.assert_(y[0] is not x[0])
|
|
self.assert_(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)
|
|
copy_reg.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 test_support.TestFailed, "shouldn't call this"
|
|
x = C()
|
|
y = copy.deepcopy(x)
|
|
self.assert_(y is x)
|
|
|
|
def test_deepcopy_reduce(self):
|
|
class C(object):
|
|
def __reduce__(self):
|
|
return ""
|
|
x = C()
|
|
y = copy.deepcopy(x)
|
|
self.assert_(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.assert_(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.assert_(x is not y)
|
|
self.assert_(x[0] is not y[0])
|
|
|
|
def test_deepcopy_reflexive_list(self):
|
|
x = []
|
|
x.append(x)
|
|
y = copy.deepcopy(x)
|
|
self.assertRaises(RuntimeError, cmp, y, x)
|
|
self.assert_(y is not x)
|
|
self.assert_(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.assert_(x is not y)
|
|
self.assert_(x[0] is not y[0])
|
|
|
|
def test_deepcopy_reflexive_tuple(self):
|
|
x = ([],)
|
|
x[0].append(x)
|
|
y = copy.deepcopy(x)
|
|
self.assertRaises(RuntimeError, cmp, y, x)
|
|
self.assert_(y is not x)
|
|
self.assert_(y[0] is not x[0])
|
|
self.assert_(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.assert_(x is not y)
|
|
self.assert_(x["foo"] is not y["foo"])
|
|
|
|
def test_deepcopy_reflexive_dict(self):
|
|
x = {}
|
|
x['foo'] = x
|
|
y = copy.deepcopy(x)
|
|
self.assertRaises(TypeError, cmp, y, x)
|
|
self.assert_(y is not x)
|
|
self.assert_(y['foo'] is y)
|
|
self.assertEqual(len(y), 1)
|
|
|
|
def test_deepcopy_keepalive(self):
|
|
memo = {}
|
|
x = 42
|
|
y = copy.deepcopy(x, memo)
|
|
self.assert_(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.assert_(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.assert_(y is not x)
|
|
self.assert_(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.assert_(y is not x)
|
|
self.assert_(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.assert_(y is not x)
|
|
self.assert_(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.assert_(y is not x)
|
|
self.assert_(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.assert_(y is not x)
|
|
self.assert_(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.assert_(y is not x)
|
|
self.assert_(y.foo is y)
|
|
|
|
# _reconstruct()
|
|
|
|
def test_reconstruct_string(self):
|
|
class C(object):
|
|
def __reduce__(self):
|
|
return ""
|
|
x = C()
|
|
y = copy.copy(x)
|
|
self.assert_(y is x)
|
|
y = copy.deepcopy(x)
|
|
self.assert_(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.assert_(y.__class__ is x.__class__)
|
|
y = copy.deepcopy(x)
|
|
self.assert_(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.assert_(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.assert_(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.assert_(y is not x)
|
|
self.assert_(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.assert_(x is not y)
|
|
self.assert_(x[0] is y[0])
|
|
y = copy.deepcopy(x)
|
|
self.assertEqual(x, y)
|
|
self.assert_(x is not y)
|
|
self.assert_(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.assert_(x is not y)
|
|
self.assert_(x["foo"] is y["foo"])
|
|
y = copy.deepcopy(x)
|
|
self.assertEqual(x, y)
|
|
self.assert_(x is not y)
|
|
self.assert_(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.assert_(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.assert_(x.foo is not y.foo)
|
|
|
|
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.assert_(x[0] is y[0])
|
|
self.assert_(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.assert_(x[0] is not y[0])
|
|
self.assert_(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.assert_(x is not y)
|
|
self.assert_(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 global_foo(x, y): return x+y
|
|
|
|
def test_main():
|
|
test_support.run_unittest(TestCopy)
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|