2017-01-12 16:48:09 +01:00
|
|
|
from unittest import TestCase
|
|
|
|
|
2017-01-14 11:09:42 +01:00
|
|
|
from src.markdown.headers import MarkdownHeader
|
2017-01-12 16:48:09 +01:00
|
|
|
|
|
|
|
|
|
|
|
class TestMarkdownHeader(TestCase):
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
self.converter = MarkdownHeader()
|
|
|
|
|
|
|
|
def test_convert_does_nothing_if_no_header(self):
|
|
|
|
self.assertEqual("blabla", self.converter.convert("blabla"))
|
|
|
|
|
|
|
|
def test_convert_h1(self):
|
|
|
|
result = self.converter.convert("====== Classes ======")
|
2017-01-12 18:02:09 +01:00
|
|
|
self.assertEqual(result, "# Classes ")
|
2017-01-12 16:48:09 +01:00
|
|
|
|
|
|
|
def test_convert_h2(self):
|
|
|
|
result = self.converter.convert("===== Classes =====")
|
2017-01-12 18:02:09 +01:00
|
|
|
self.assertEqual(result, "## Classes ")
|
2017-01-12 16:48:09 +01:00
|
|
|
|
|
|
|
def test_convert_h3(self):
|
|
|
|
result = self.converter.convert("==== Classes ====")
|
2017-01-12 18:02:09 +01:00
|
|
|
self.assertEqual(result, "### Classes ")
|
2017-01-12 16:48:09 +01:00
|
|
|
|
|
|
|
def test_convert_h4(self):
|
|
|
|
result = self.converter.convert("=== Classes ===")
|
2017-01-12 18:02:09 +01:00
|
|
|
self.assertEqual(result, "#### Classes ")
|
2017-01-12 16:48:09 +01:00
|
|
|
|
2017-01-12 18:02:09 +01:00
|
|
|
def test_convert_h5(self):
|
2017-01-12 16:48:09 +01:00
|
|
|
result = self.converter.convert("== Classes ==")
|
2017-01-12 18:02:09 +01:00
|
|
|
self.assertEqual(result, "##### Classes ")
|