class WikiFormatterTest(unittest.TestCase):
def setUp(self):
self.stream = StringIO.StringIO()
self.formatter = WikiFormatter(self.stream)
def test_title(self):
self.formatter.title('foo')
self.assertEqual(self.stream.getvalue(), '= foo =\n\n')
def test_section(self):
self.formatter.section('foo')
self.assertEqual(self.stream.getvalue(), '\n== foo ==\n\n')
def test_subsection(self):
self.formatter.subsection('foo')
self.assertEqual(self.stream.getvalue(), '=== foo ===\n\n')
def test_paragraph(self):
self.formatter.paragraph('\nfoo\nbar\n')
self.assertEqual(self.stream.getvalue(), 'foo\nbar\n\n')
def setUp(self):
self.stream = StringIO.StringIO()
self.formatter = WikiFormatter(self.stream)
