tests/test_mwmarkup.py

   1 #!/usr/bin/python2.6
   2 # -*- coding: iso-8859-15 -*-
   3 import wrpylib.mwmarkup
   4
   5
   6 def test_find_template():
   7     wikitext = u'''== Allgemeines ==
   8     {{Rodelbahnbox
   9     | Position             = 47.309820 N 9.986508 E
  10     | Position oben        =
  11     | Höhe oben            = 1244
  12     | Position unten       =
  13     | Höhe unten           = 806
  14     | Länge                = 5045
  15     | Schwierigkeit        =
  16     | Lawinen              = gelegentlich
  17     | Betreiber            =
  18     | Öffentliche Anreise  = Ja
  19     | Gehzeit              = 105
  20     | Aufstieg getrennt    = Nein
  21     | Aufstiegshilfe       = Nein
  22     | Beleuchtungsanlage   = Nein
  23     | Beleuchtungstage     =
  24     | Rodelverleih         = Ja
  25     | Gütesiegel           =
  26     | Webauskunft          =
  27     | Telefonauskunft      = +43-664-1808482 (Bergkristallhütte)
  28     | Bild                 = Rodelbahn Bergkristallhütte 2009-03-03.jpg
  29     | In Übersichtskarte   = Ja
  30     | Forumid              = 72
  31     }}
  32     Die Rodelbahn zur Bergkristallhütte ist durchaus abwechslungsreich.'''
  33     start, end = wrpylib.mwmarkup.find_template(wikitext, u'Rodelbahnbox')
  34     assert start == wikitext.find(u'{{')
  35     assert end == wikitext.find(u'}}')+2
  36
  37
  38 def test_split_template():
  39     wikitext = u'''== Allgemeines ==
  40     {{Rodelbahnbox
  41     | Position             = 47.309820 N 9.986508 E
  42     | Position oben        =
  43     | Höhe oben            = 1244
  44     | Position unten       =
  45     | Höhe unten           = 806
  46     | Länge                = 5045
  47     | Schwierigkeit        =
  48     | Lawinen              = gelegentlich
  49     | Betreiber            =
  50     | Öffentliche Anreise  = Ja
  51     | Gehzeit              = 105
  52     | Aufstieg getrennt    = Nein
  53     | Aufstiegshilfe       = Nein
  54     | Beleuchtungsanlage   = Nein
  55     | Beleuchtungstage     =
  56     | Rodelverleih         = Ja
  57     | Gütesiegel           =
  58     | Webauskunft          =
  59     | Telefonauskunft      = +43-664-1808482 (Bergkristallhütte)
  60     | Bild                 = Rodelbahn Bergkristallhütte 2009-03-03.jpg
  61     | In Übersichtskarte   = Ja
  62     | Forumid              = 72
  63     }}
  64     Die Rodelbahn zur Bergkristallhütte ist durchaus abwechslungsreich.'''
  65     start, end = wrpylib.mwmarkup.find_template(wikitext, u'Rodelbahnbox')
  66     template_title, parameters = wrpylib.mwmarkup.split_template(wikitext[start:end])
  67     assert template_title == u'Rodelbahnbox'
  68     assert len(parameters) == 22
  69     assert parameters['Position'] == u'47.309820 N 9.986508 E'
  70     assert parameters['Telefonauskunft'] == u'+43-664-1808482 (Bergkristallhütte)'
  71     assert parameters['Schwierigkeit'] == u''
  72
  73
  74 def test_create_template():
  75     wikitext = u'''{{Nicetemplate|Red|Bold|Position=Top|Alignment=Right}}'''
  76     template_title, parameters = wrpylib.mwmarkup.split_template(wikitext)
  77     print parameters
  78     keys = [u'1', u'2', u'Position', u'Alignment']
  79     values = [parameters[k] for k in keys]
  80     wikitext_generated = wrpylib.mwmarkup.create_template(template_title, values[:2], keys[2:], values[2:])
  81     wikitext_table = wrpylib.mwmarkup.create_template(template_title, values[:2], keys[2:], values[2:], True)
  82     assert wikitext_generated == wikitext
  83
  84