PEP8 improvements.

[philipp/winterrodeln/wrpylib.git] / tests / test_mwmarkup.py

diff --git a/tests/test_mwmarkup.py b/tests/test_mwmarkup.py
index 9955656d592b31e2361e52c42a1ae39173e50e22..14e0e17a65d8db76fa25d032fa1e4eabecd7cde2 100644 (file)
--- a/tests/test_mwmarkup.py
+++ b/tests/test_mwmarkup.py
@@ -1,4 +1,3 @@
-#!/usr/bin/python3.4
 import unittest
 import mwparserfromhell
 import wrpylib.mwmarkup
@@ -39,8 +38,8 @@ class TestMwParserFromHell(unittest.TestCase):
         rb = list(wikicode.filter_templates())[0]
         self.assertEqual(rb.name.strip(), 'Rodelbahnbox')
         self.assertEqual(rb.get('Aufstiegshilfe').value.strip(), 'Nein')
-        self.assertEqual(rb[:2], '{{')
-        self.assertEqual(rb[-2:], '}}')
+        self.assertEqual(str(rb)[:2], '{{')
+        self.assertEqual(str(rb)[-2:], '}}')
 
     def test_template_to_table(self):
         wikitext = '{{Rodelbahnbox | Unbenannt | Position = 47.309820 N 9.986508 E | Aufstieg möglich = Ja }}'
@@ -147,3 +146,10 @@ class TestFormatTemplate(unittest.TestCase):
         value = mwparserfromhell.parse('{{ my_template ||   | var = 7 }}').filter_templates()[0]
         wrpylib.mwmarkup.format_template_oneline(value)
         self.assertEqual('{{my_template| | |var=7}}', value)
+
+
+class TestMwApi(unittest.TestCase):
+    def test_dbkey_to_title(self):
+        self.assertEqual(wrpylib.mwmarkup.dbkey_to_title('My_Title'), 'My Title')
+        self.assertEqual(wrpylib.mwmarkup.dbkey_to_title('My Title'), 'My Title') # should not happen
+        self.assertEqual(wrpylib.mwmarkup.dbkey_to_title('My_nice_Title'), 'My nice Title')