summaryrefslogtreecommitdiffstats
path: root/Lib
diff options
context:
space:
mode:
authorRaymond Hettinger <python@rcn.com>2008-01-04 03:22:53 (GMT)
committerRaymond Hettinger <python@rcn.com>2008-01-04 03:22:53 (GMT)
commite0734e7dc0dcccc91ed657191b804b3a846ad3f6 (patch)
tree70cb770b9fb3fbca03fcbb1a53df10f69612a78b /Lib
parent123d5c9396a6c12d418bbdf6d7ec861537f4c199 (diff)
downloadcpython-e0734e7dc0dcccc91ed657191b804b3a846ad3f6.zip
cpython-e0734e7dc0dcccc91ed657191b804b3a846ad3f6.tar.gz
cpython-e0734e7dc0dcccc91ed657191b804b3a846ad3f6.tar.bz2
Minor fix-ups to named tuples:
* Make the _replace() method respect subclassing. * Using property() to make _fields read-only wasn't a good idea. It caused len(Point._fields) to fail. * Add note to _cast() about length checking and alternative with the star-operator.
Diffstat (limited to 'Lib')
-rw-r--r--Lib/collections.py6
-rw-r--r--Lib/test/test_collections.py9
2 files changed, 3 insertions, 12 deletions
diff --git a/Lib/collections.py b/Lib/collections.py
index c6d0d0f..487b119 100644
--- a/Lib/collections.py
+++ b/Lib/collections.py
@@ -60,6 +60,7 @@ def namedtuple(typename, field_names, verbose=False):
template = '''class %(typename)s(tuple):
'%(typename)s(%(argtxt)s)' \n
__slots__ = () \n
+ _fields = %(field_names)r \n
def __new__(cls, %(argtxt)s):
return tuple.__new__(cls, (%(argtxt)s)) \n
_cast = classmethod(tuple.__new__) \n
@@ -70,10 +71,7 @@ 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 %(typename)s._cast(map(kwds.get, %(field_names)r, self)) \n
- @property
- def _fields(self):
- return %(field_names)r \n\n''' % locals()
+ return self.__class__._cast(map(kwds.get, %(field_names)r, self)) \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 edffbbe..5e71399 100644
--- a/Lib/test/test_collections.py
+++ b/Lib/test/test_collections.py
@@ -17,6 +17,7 @@ class TestNamedTuple(unittest.TestCase):
self.assertEqual(Point.__slots__, ())
self.assertEqual(Point.__module__, __name__)
self.assertEqual(Point.__getitem__, tuple.__getitem__)
+ self.assertEqual(Point._fields, ('x', 'y'))
self.assertRaises(ValueError, namedtuple, 'abc%', 'efg ghi') # type has non-alpha char
self.assertRaises(ValueError, namedtuple, 'class', 'efg ghi') # type has keyword
@@ -51,14 +52,6 @@ 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
- # Verify that _fields is read-only
- try:
- p._fields = ('F1' ,'F2')
- except AttributeError:
- pass
- else:
- self.fail('The _fields attribute needs to be read-only')
-
# verify that field string can have commas
Point = namedtuple('Point', 'x, y')
p = Point(x=11, y=22)