wgroeneveld
/
dokuwiki-to-hugo
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update tests

This commit is contained in:
wgroeneveld 2017-01-12 18:02:09 +01:00
parent fad6925ecd
commit 04e4da24b4
3 changed files with 26 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.gitignore vendored
View File

@ -1,3 +1,6 @@
# idea stuff
.idea/
# Byte-compiled / optimized / DLL files # Byte-compiled / optimized / DLL files
__pycache__/ __pycache__/
*.py[cod] *.py[cod]

23
src/markdown_headers.py
View File

@ -1,12 +1,23 @@
from collections import OrderedDict
class MarkdownHeader: class MarkdownHeader:
head = "="
config = {
'======': 1,
'=====': 2,
'====': 3,
'===': 4,
'==': 5
}
def convert(self, text): def convert(self, text):
if not text.startswith("="): config = OrderedDict(sorted(MarkdownHeader.config.items(), key = lambda t : t[1]))
return text
elif text.startswith("== "): for key, val in config.items():
return "######" + self.strip(text) if text.startswith(key):
return "##" return ('#' * val) + self.strip(text)
return text
def strip(self, text): def strip(self, text):
return text.replace("=", "") return text.replace(MarkdownHeader.head, "")

16
test/test_markdown_header.py
View File

@ -13,24 +13,20 @@ class TestMarkdownHeader(TestCase):
def test_convert_h1(self): def test_convert_h1(self):
result = self.converter.convert("====== Classes ======") result = self.converter.convert("====== Classes ======")
self.assertEqual(result, "# Classes") self.assertEqual(result, "# Classes ")
def test_convert_h2(self): def test_convert_h2(self):
result = self.converter.convert("===== Classes =====") result = self.converter.convert("===== Classes =====")
self.assertEqual(result, "## Classes") self.assertEqual(result, "## Classes ")
def test_convert_h3(self): def test_convert_h3(self):
result = self.converter.convert("==== Classes ====") result = self.converter.convert("==== Classes ====")
self.assertEqual(result, "### Classes") self.assertEqual(result, "### Classes ")
def test_convert_h4(self): def test_convert_h4(self):
result = self.converter.convert("==== Classes ====") result = self.converter.convert("=== Classes ===")
self.assertEqual(result, "#### Classes") self.assertEqual(result, "#### Classes ")
def test_convert_h5(self): def test_convert_h5(self):
result = self.converter.convert("=== Classes ===")
self.assertEqual(result, "##### Classes")
def test_convert_h6(self):
result = self.converter.convert("== Classes ==") result = self.converter.convert("== Classes ==")
self.assertEqual(result, "###### Classes ") self.assertEqual(result, "##### Classes ")