tests/test_lib_sledrun_wikitext_to_json.py

   1 import unittest
   2
   3 from mwparserfromhell import parse
   4
   5 from wrpylib.lib_sledrun_wikitext_to_json import get_sledrun_description, optional_set
   6
   7
   8 class TestOptionalSet(unittest.TestCase):
   9     def test_optional_set_string(self):
  10         h = {'a': 3}
  11         optional_set(h, 'b', 5)
  12         expected = {'a': 3, 'b': 5}
  13         self.assertDictEqual(expected, h)
  14
  15     def test_optional_set_none(self):
  16         h = {'a': 3}
  17         expected = h.copy()
  18         optional_set(h, 'b', None)
  19         self.assertDictEqual(expected, h)
  20
  21
  22 class TestGetSledrunDescription(unittest.TestCase):
  23     def test_get_sledrun_description(self):
  24         wikitext = "== Allgemeines ==\n{{Rodelbahnbox}}\nDas ist ''die'' Beschreibung.\n" + \
  25                     "Weitere Details bei https://example.com.\n\n* Position\n"
  26         wikicode = parse(wikitext)
  27         expected = "Das ist ''die'' Beschreibung.\nWeitere Details bei https://example.com."
  28         actual = get_sledrun_description(wikicode)
  29         self.assertEqual(expected, actual)