From 132c6584d059497374c7f0c53285e6251beb6675 Mon Sep 17 00:00:00 2001 From: Ben Kurtovic Date: Fri, 5 Apr 2013 10:52:43 -0400 Subject: [PATCH] test_heading() and test_comment() --- tests/test_builder.py | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/tests/test_builder.py b/tests/test_builder.py index 7dcbc0e..410eb4a 100644 --- a/tests/test_builder.py +++ b/tests/test_builder.py @@ -173,11 +173,31 @@ class TestBuilder(TreeEqualityTestCase): def test_heading(self): """tests for building Heading nodes""" - pass + tests = [ + ([tokens.HeadingStart(level=2), tokens.Text(text="foobar"), + tokens.HeadingEnd()], + wrap([Heading(wrap([Text("foobar")]), 2)])), + + ([tokens.HeadingStart(level=4), tokens.Text(text="spam"), + tokens.Text(text="eggs"), tokens.HeadingEnd()], + wrap([Heading(wrap([Text("spam"), Text("eggs")]), 4)])), + ] + for test, valid in tests: + self.assertWikicodeEqual(valid, self.builder.build(test)) def test_comment(self): """tests for building Comment nodes""" - pass + tests = [ + ([tokens.CommentStart(), tokens.Text(text="foobar"), + tokens.CommentEnd()], + wrap([Comment(wrap([Text("foobar")]))])), + + ([tokens.CommentStart(), tokens.Text(text="spam"), + tokens.Text(text="eggs"), tokens.CommentEnd()], + wrap([Comment(wrap([Text("spam"), Text("eggs")]))])), + ] + for test, valid in tests: + self.assertWikicodeEqual(valid, self.builder.build(test)) @unittest.skip("holding this until feature/html_tags is ready") def test_tag(self):