summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2008-01-05 02:17:24 (GMT)
committerRaymond Hettinger <python@rcn.com>2008-01-05 02:17:24 (GMT)
commit1b50fd7cb3e9a4483f6da49c027ebde4501d1c17 (patch)
tree5fbe7ba58aafb667a6c6e14920e17a7f19f9456c
parent02740f73ff0f12d276ef16b73208d4f9f8d62baa (diff)
downloadcpython-1b50fd7cb3e9a4483f6da49c027ebde4501d1c17.zip
cpython-1b50fd7cb3e9a4483f6da49c027ebde4501d1c17.tar.gz
cpython-1b50fd7cb3e9a4483f6da49c027ebde4501d1c17.tar.bz2
Add error-checking to namedtuple's _replace() method.
-rw-r--r--Doc/library/collections.rst5
-rw-r--r--Lib/collections.py5
-rw-r--r--Lib/test/test_collections.py7
3 files changed, 15 insertions, 2 deletions
diff --git a/Doc/library/collections.rst b/Doc/library/collections.rst
index 34c3118..eaa823a 100644
--- a/Doc/library/collections.rst
+++ b/Doc/library/collections.rst
@@ -410,7 +410,10 @@ Example::
def _replace(self, **kwds):
'Return a new Point object replacing specified fields with new values'
- return self.__class__._make(map(kwds.get, ('x', 'y'), self))
+ result = self.__class__._make(map(kwds.pop, ('x', 'y'), self))
+ if kwds:
+ raise ValueError('Got unexpected field names: %r' % kwds.keys())
+ return result
x = property(itemgetter(0))
y = property(itemgetter(1))
diff --git a/Lib/collections.py b/Lib/collections.py
index 2a58b86..9985f93 100644
--- a/Lib/collections.py
+++ b/Lib/collections.py
@@ -78,7 +78,10 @@ def namedtuple(typename, field_names, verbose=False):
return {%(dicttxt)s} \n
def _replace(self, **kwds):
'Return a new %(typename)s object replacing specified fields with new values'
- return self.__class__._make(map(kwds.get, %(field_names)r, self)) \n\n''' % locals()
+ result = self.__class__._make(map(kwds.pop, %(field_names)r, self))
+ if kwds:
+ raise ValueError('Got unexpected field names: %%r' %% kwds.keys())
+ return result \n\n''' % locals()
for i, name in enumerate(field_names):
template += ' %s = property(itemgetter(%d))\n' % (name, i)
if verbose:
diff --git a/Lib/test/test_collections.py b/Lib/test/test_collections.py
index c17b76d..dd9982a 100644
--- a/Lib/test/test_collections.py
+++ b/Lib/test/test_collections.py
@@ -55,6 +55,13 @@ class TestNamedTuple(unittest.TestCase):
self.assertEqual(p._replace(x=1), (1, 22)) # test _replace method
self.assertEqual(p._asdict(), dict(x=11, y=22)) # test _asdict method
+ try:
+ p._replace(x=1, error=2)
+ except ValueError:
+ pass
+ else:
+ self._fail('Did not detect an incorrect fieldname')
+
# verify that field string can have commas
Point = namedtuple('Point', 'x, y')
p = Point(x=11, y=22)