Пример #1
0
 def test_iternodes(self):
     """test Wikilink.__iternodes__()"""
     node1n1 = Text("foobar")
     node2n1, node2n2, node2n3 = Text("foo"), Text("bar"), Text("baz")
     node1 = Wikilink(wrap([node1n1]))
     node2 = Wikilink(wrap([node2n1]), wrap([node2n2, node2n3]))
     gen1 = node1.__iternodes__(getnodes)
     gen2 = node2.__iternodes__(getnodes)
     self.assertEqual((None, node1), next(gen1))
     self.assertEqual((None, node2), next(gen2))
     self.assertEqual((node1.title, node1n1), next(gen1))
     self.assertEqual((node2.title, node2n1), next(gen2))
     self.assertEqual((node2.text, node2n2), next(gen2))
     self.assertEqual((node2.text, node2n3), next(gen2))
     self.assertRaises(StopIteration, next, gen1)
     self.assertRaises(StopIteration, next, gen2)