def test_title(self): """test getter/setter for the title attribute""" title = wraptext("foobar") node = Heading(title, 3) self.assertIs(title, node.title) node.title = "héhehé" self.assertWikicodeEqual(wraptext("héhehé"), node.title)
def test_iternodes(self): """test Heading.__iternodes__()""" text1, text2 = Text("foo"), Text("bar") node = Heading(wrap([text1, text2]), 3) gen = node.__iternodes__(getnodes) self.assertEqual((None, node), next(gen)) self.assertEqual((node.title, text1), next(gen)) self.assertEqual((node.title, text2), next(gen)) self.assertRaises(StopIteration, next, gen)
def test_level(self): """test getter/setter for the level attribute""" node = Heading(wraptext("foobar"), 3) self.assertEqual(3, node.level) node.level = 5 self.assertEqual(5, node.level) self.assertRaises(ValueError, setattr, node, "level", 0) self.assertRaises(ValueError, setattr, node, "level", 7) self.assertRaises(ValueError, setattr, node, "level", "abc") self.assertRaises(ValueError, setattr, node, "level", False)
def test_heading(self): """tests for building Heading nodes""" tests = [ ([tokens.HeadingStart(level=2), tokens.Text(text="foobar"), tokens.HeadingEnd()], wrap([Heading(wraptext("foobar"), 2)])), ([tokens.HeadingStart(level=4), tokens.Text(text="spam"), tokens.Text(text="eggs"), tokens.HeadingEnd()], wrap([Heading(wraptext("spam", "eggs"), 4)])), ] for test, valid in tests: self.assertWikicodeEqual(valid, self.builder.build(test))
def test_showtree(self): """test Heading.__showtree__()""" output = [] getter = object() get = lambda code: output.append((getter, code)) node1 = Heading(wraptext("foobar"), 3) node2 = Heading(wraptext(" baz "), 4) node1.__showtree__(output.append, get, None) node2.__showtree__(output.append, get, None) valid = ["===", (getter, node1.title), "===", "====", (getter, node2.title), "===="] self.assertEqual(valid, output)
def test_strip(self): """test Heading.__strip__()""" node = Heading(wraptext("foobar"), 3) for a in (True, False): for b in (True, False): self.assertEqual("foobar", node.__strip__(a, b))
def test_children(self): """test Heading.__children__()""" node = Heading(wrap([Text("foo"), Text("bar")]), 3) gen = node.__children__() self.assertEqual(node.title, next(gen)) self.assertRaises(StopIteration, next, gen)
def test_unicode(self): """test Heading.__unicode__()""" node = Heading(wraptext("foobar"), 2) self.assertEqual("==foobar==", str(node)) node2 = Heading(wraptext(" zzz "), 5) self.assertEqual("===== zzz =====", str(node2))
def test_strip(self): """test Heading.__strip__()""" node = Heading(wraptext("foobar"), 3) self.assertEqual("foobar", node.__strip__())