mirror of https://github.com/python/cpython
75 lines
3.4 KiB
Python
75 lines
3.4 KiB
Python
import unittest
|
|
from test import test_support
|
|
from collections import NamedTuple
|
|
|
|
class TestNamedTuple(unittest.TestCase):
|
|
|
|
def test_factory(self):
|
|
Point = NamedTuple('Point', 'x y')
|
|
self.assertEqual(Point.__name__, 'Point')
|
|
self.assertEqual(Point.__doc__, 'Point(x, y)')
|
|
self.assertEqual(Point.__slots__, ())
|
|
self.assertEqual(Point.__module__, __name__)
|
|
self.assertEqual(Point.__getitem__, tuple.__getitem__)
|
|
self.assertRaises(ValueError, NamedTuple, 'abc%', 'def ghi')
|
|
self.assertRaises(ValueError, NamedTuple, 'abc', 'def g%hi')
|
|
NamedTuple('Point0', 'x1 y2') # Verify that numbers are allowed in names
|
|
|
|
def test_instance(self):
|
|
Point = NamedTuple('Point', 'x y')
|
|
p = Point(11, 22)
|
|
self.assertEqual(p, Point(x=11, y=22))
|
|
self.assertEqual(p, Point(11, y=22))
|
|
self.assertEqual(p, Point(y=22, x=11))
|
|
self.assertEqual(p, Point(*(11, 22)))
|
|
self.assertEqual(p, Point(**dict(x=11, y=22)))
|
|
self.assertRaises(TypeError, Point, 1) # too few args
|
|
self.assertRaises(TypeError, Point, 1, 2, 3) # too many args
|
|
self.assertRaises(TypeError, eval, 'Point(XXX=1, y=2)', locals()) # wrong keyword argument
|
|
self.assertRaises(TypeError, eval, 'Point(x=1)', locals()) # missing keyword argument
|
|
self.assertEqual(repr(p), 'Point(x=11, y=22)')
|
|
self.assert_('__dict__' not in dir(p)) # verify instance has no dict
|
|
self.assert_('__weakref__' not in dir(p))
|
|
self.assertEqual(p.__fields__, ('x', 'y')) # test __fields__ attribute
|
|
self.assertEqual(p.__replace__('x', 1), (1, 22)) # test __replace__ method
|
|
|
|
# verify that field string can have commas
|
|
Point = NamedTuple('Point', 'x, y')
|
|
p = Point(x=11, y=22)
|
|
self.assertEqual(repr(p), 'Point(x=11, y=22)')
|
|
|
|
def test_tupleness(self):
|
|
Point = NamedTuple('Point', 'x y')
|
|
p = Point(11, 22)
|
|
|
|
self.assert_(isinstance(p, tuple))
|
|
self.assertEqual(p, (11, 22)) # matches a real tuple
|
|
self.assertEqual(tuple(p), (11, 22)) # coercable to a real tuple
|
|
self.assertEqual(list(p), [11, 22]) # coercable to a list
|
|
self.assertEqual(max(p), 22) # iterable
|
|
self.assertEqual(max(*p), 22) # star-able
|
|
x, y = p
|
|
self.assertEqual(p, (x, y)) # unpacks like a tuple
|
|
self.assertEqual((p[0], p[1]), (11, 22)) # indexable like a tuple
|
|
self.assertRaises(IndexError, p.__getitem__, 3)
|
|
|
|
self.assertEqual(p.x, x)
|
|
self.assertEqual(p.y, y)
|
|
self.assertRaises(AttributeError, eval, 'p.z', locals())
|
|
|
|
|
|
def test_odd_sizes(self):
|
|
Zero = NamedTuple('Zero', '')
|
|
self.assertEqual(Zero(), ())
|
|
Dot = NamedTuple('Dot', 'd')
|
|
self.assertEqual(Dot(1), (1,))
|
|
|
|
def test_main(verbose=None):
|
|
import collections as CollectionsModule
|
|
test_classes = [TestNamedTuple]
|
|
test_support.run_unittest(*test_classes)
|
|
test_support.run_doctest(CollectionsModule, verbose)
|
|
|
|
if __name__ == "__main__":
|
|
test_main(verbose=True)
|