def test_getitem(self): # test the __getitem__ method nodes = [n.Node('a', dict(z='Z')), n.Node('b')] root = n.Node('root', nodes=nodes) self.assertEqual(root.a['z'], 'Z') self.assertEqual(root[0], nodes[0]) self.assertEqual(root[1], nodes[1]) self.assertEqual(list(root), nodes)
def test_setitem(self): root = n.Node('root') root['a'] = 'A' self.assertEqual(root['a'], 'A') self.assertEqual(root.attrib['a'], 'A')
def test_can_pickle(self): node = n.Node('tag') self.assertEqual(pickle.loads(pickle.dumps(node)), node)