| Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame^] | 1 | import unittest |
| 2 | from test import test_support |
| 3 | from collections import NamedTuple |
| 4 | |
| 5 | class TestNamedTuple(unittest.TestCase): |
| 6 | |
| 7 | def test_factory(self): |
| 8 | Point = NamedTuple('Point', 'x y') |
| 9 | self.assertEqual(Point.__name__, 'Point') |
| 10 | self.assertEqual(Point.__doc__, 'Point(x, y)') |
| 11 | self.assertEqual(Point.__slots__, ()) |
| 12 | self.assertEqual(Point.__module__, __name__) |
| 13 | self.assertEqual(Point.__getitem__, tuple.__getitem__) |
| 14 | self.assert_('__getitem__' in Point.__dict__) # superclass methods localized |
| 15 | |
| 16 | def test_instance(self): |
| 17 | Point = NamedTuple('Point', 'x y') |
| 18 | p = Point(11, 22) |
| 19 | self.assertEqual(p, Point(x=11, y=22)) |
| 20 | self.assertEqual(p, Point(11, y=22)) |
| 21 | self.assertEqual(p, Point(y=22, x=11)) |
| 22 | self.assertEqual(p, Point(*(11, 22))) |
| 23 | self.assertEqual(p, Point(**dict(x=11, y=22))) |
| 24 | self.assertRaises(TypeError, Point, 1) # too few args |
| 25 | self.assertRaises(TypeError, Point, 1, 2, 3) # too many args |
| 26 | self.assertRaises(TypeError, eval, 'Point(XXX=1, y=2)', locals()) # wrong keyword argument |
| 27 | self.assertRaises(TypeError, eval, 'Point(x=1)', locals()) # missing keyword argument |
| 28 | self.assertEqual(repr(p), 'Point(x=11, y=22)') |
| 29 | self.assert_('__dict__' not in dir(p)) # verify instance has no dict |
| 30 | self.assert_('__weakref__' not in dir(p)) |
| 31 | |
| 32 | def test_tupleness(self): |
| 33 | Point = NamedTuple('Point', 'x y') |
| 34 | p = Point(11, 22) |
| 35 | |
| 36 | self.assert_(isinstance(p, tuple)) |
| 37 | self.assertEqual(p, (11, 22)) # matches a real tuple |
| 38 | self.assertEqual(tuple(p), (11, 22)) # coercable to a real tuple |
| 39 | self.assertEqual(list(p), [11, 22]) # coercable to a list |
| 40 | self.assertEqual(max(p), 22) # iterable |
| 41 | self.assertEqual(max(*p), 22) # star-able |
| 42 | x, y = p |
| 43 | self.assertEqual(p, (x, y)) # unpacks like a tuple |
| 44 | self.assertEqual((p[0], p[1]), (11, 22)) # indexable like a tuple |
| 45 | self.assertRaises(IndexError, p.__getitem__, 3) |
| 46 | |
| 47 | self.assertEqual(p.x, x) |
| 48 | self.assertEqual(p.y, y) |
| 49 | self.assertRaises(AttributeError, eval, 'p.z', locals()) |
| 50 | |
| 51 | |
| 52 | def test_main(verbose=None): |
| 53 | test_classes = [TestNamedTuple] |
| 54 | test_support.run_unittest(*test_classes) |
| 55 | |
| 56 | if __name__ == "__main__": |
| 57 | test_main(verbose=True) |