"""Unit tests for the copy module."""

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, 2L**100, 3.14, True, False, 1j,

                 "hello", u"hello\u1234", f.func_code,

                 NewStyle, xrange(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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

        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, 2L**100, 3.14, True, False, 1j,

                 "hello", u"hello\u1234", f.func_code,

                 NewStyle, xrange(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(RuntimeError, 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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(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 __cmp__(self, other):

                return cmp(self.__dict__, other.__dict__)

            __hash__ = None # Silence Py3k warning

        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 __cmp__(self, other):

                return cmp(self.__dict__, other.__dict__)

            __hash__ = None # Silence Py3k warning

        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 __cmp__(self, other):

                return (cmp(list(self), list(other)) or

                        cmp(self.__dict__, other.__dict__))

            __hash__ = None # Silence Py3k warning

        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.iteritems())

            def __cmp__(self, other):

                return (cmp(dict(self), list(dict)) or

                        cmp(self.__dict__, other.__dict__))

            __hash__ = None # Silence Py3k warning

        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):