diff options
author | Raymond Hettinger <python@rcn.com> | 2009-08-04 19:08:05 (GMT) |
---|---|---|
committer | Raymond Hettinger <python@rcn.com> | 2009-08-04 19:08:05 (GMT) |
commit | 8fdab9526070362a02eb12f84ecc601e3182bbc2 (patch) | |
tree | 3b7eba20a49378b2aaf36831c25ba9b13f4dfb60 | |
parent | 6c0f337904c1a6414fbaee84f7923cc0877e47eb (diff) | |
download | cpython-8fdab9526070362a02eb12f84ecc601e3182bbc2.zip cpython-8fdab9526070362a02eb12f84ecc601e3182bbc2.tar.gz cpython-8fdab9526070362a02eb12f84ecc601e3182bbc2.tar.bz2 |
Issue 6637: defaultdict.copy() failed with an empty factory.
-rw-r--r-- | Lib/test/test_defaultdict.py | 7 | ||||
-rw-r--r-- | Modules/_collectionsmodule.c | 5 |
2 files changed, 11 insertions, 1 deletions
diff --git a/Lib/test/test_defaultdict.py b/Lib/test/test_defaultdict.py index 71d69c3..f9a6a17 100644 --- a/Lib/test/test_defaultdict.py +++ b/Lib/test/test_defaultdict.py @@ -59,6 +59,7 @@ class TestDefaultDict(unittest.TestCase): d1 = defaultdict() self.assertEqual(d1.default_factory, None) self.assertEqual(repr(d1), "defaultdict(None, {})") + self.assertEqual(eval(repr(d1)), d1) d1[11] = 41 self.assertEqual(repr(d1), "defaultdict(None, {11: 41})") d2 = defaultdict(int) @@ -111,6 +112,12 @@ class TestDefaultDict(unittest.TestCase): d4[12] self.assertEqual(d4, {42: [], 12: []}) + # Issue 6637: Copy fails for empty default dict + d = defaultdict() + d['a'] = 42 + e = d.copy() + self.assertEqual(e['a'], 42) + def test_shallow_copy(self): d1 = defaultdict(foobar, {1: 1}) d2 = copy.copy(d1) diff --git a/Modules/_collectionsmodule.c b/Modules/_collectionsmodule.c index a4fdf76..1873cab 100644 --- a/Modules/_collectionsmodule.c +++ b/Modules/_collectionsmodule.c @@ -1228,6 +1228,9 @@ defdict_copy(defdictobject *dd) whose class constructor has the same signature. Subclasses that define a different constructor signature must override copy(). */ + + if (dd->default_factory == NULL) + return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd), Py_None, dd, NULL); return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd), dd->default_factory, dd, NULL); } @@ -1391,7 +1394,7 @@ defdict_init(PyObject *self, PyObject *args, PyObject *kwds) Py_ssize_t n = PyTuple_GET_SIZE(args); if (n > 0) { newdefault = PyTuple_GET_ITEM(args, 0); - if (!PyCallable_Check(newdefault)) { + if (!PyCallable_Check(newdefault) && newdefault != Py_None) { PyErr_SetString(PyExc_TypeError, "first argument must be callable"); return -1; |