ben
/
mwparserfromhell
mirror of https://github.com/earwig/mwparserfromhell
1
1
Fork 0
Code Releases 24 Activity
A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
810 Commits
1 Branch
6.1 MiB
 
 
 
 
Tree: 7a30e47f76
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7a30e47f76'
${ noResults }
mwparserfromhell/tests/test_template.py

669 lines
28 KiB
Raw Blame History 

  1. # -*- coding: utf-8  -*-

  2. #

  3. # Copyright (C) 2012-2017 Ben Kurtovic <ben.kurtovic@gmail.com>

  4. #

  5. # Permission is hereby granted, free of charge, to any person obtaining a copy

  6. # of this software and associated documentation files (the "Software"), to deal

  7. # in the Software without restriction, including without limitation the rights

  8. # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

  9. # copies of the Software, and to permit persons to whom the Software is

  10. # furnished to do so, subject to the following conditions:

  11. #

  12. # The above copyright notice and this permission notice shall be included in

  13. # all copies or substantial portions of the Software.

  14. #

  15. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

  16. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

  17. # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

  18. # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

  19. # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

  20. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

  21. # SOFTWARE.

  22. 

  23. from __future__ import unicode_literals

  24. 

  25. try:

  26.     import unittest2 as unittest

  27. except ImportError:

  28.     import unittest

  29. 

  30. from mwparserfromhell.compat import str

  31. from mwparserfromhell.nodes import HTMLEntity, Template, Text

  32. from mwparserfromhell.nodes.extras import Parameter

  33. from mwparserfromhell import parse

  34. 

  35. from ._test_tree_equality import TreeEqualityTestCase, wrap, wraptext

  36. 

  37. pgens = lambda k, v: Parameter(wraptext(k), wraptext(v), showkey=True)

  38. pgenh = lambda k, v: Parameter(wraptext(k), wraptext(v), showkey=False)

  39. 

  40. class TestTemplate(TreeEqualityTestCase):

  41.     """Test cases for the Template node."""

  42. 

  43.     def test_unicode(self):

  44.         """test Template.__unicode__()"""

  45.         node = Template(wraptext("foobar"))

  46.         self.assertEqual("{{foobar}}", str(node))

  47.         node2 = Template(wraptext("foo"),

  48.                          [pgenh("1", "bar"), pgens("abc", "def")])

  49.         self.assertEqual("{{foo|bar|abc=def}}", str(node2))

  50. 

  51.     def test_children(self):

  52.         """test Template.__children__()"""

  53.         node2p1 = Parameter(wraptext("1"), wraptext("bar"), showkey=False)

  54.         node2p2 = Parameter(wraptext("abc"), wrap([Text("def"), Text("ghi")]),

  55.                             showkey=True)

  56.         node1 = Template(wraptext("foobar"))

  57.         node2 = Template(wraptext("foo"), [node2p1, node2p2])

  58. 

  59.         gen1 = node1.__children__()

  60.         gen2 = node2.__children__()

  61.         self.assertEqual(node1.name, next(gen1))

  62.         self.assertEqual(node2.name, next(gen2))

  63.         self.assertEqual(node2.params[0].value, next(gen2))

  64.         self.assertEqual(node2.params[1].name, next(gen2))

  65.         self.assertEqual(node2.params[1].value, next(gen2))

  66.         self.assertRaises(StopIteration, next, gen1)

  67.         self.assertRaises(StopIteration, next, gen2)

  68. 

  69.     def test_strip(self):

  70.         """test Template.__strip__()"""

  71.         node1 = Template(wraptext("foobar"))

  72.         node2 = Template(wraptext("foo"), [

  73.             pgenh("1", "bar"), pgens("foo", ""), pgens("abc", "def")])

  74.         node3 = Template(wraptext("foo"), [

  75.             pgenh("1", "foo"),

  76.             Parameter(wraptext("2"), wrap([Template(wraptext("hello"))]),

  77.                       showkey=False),

  78.             pgenh("3", "bar")])

  79. 

  80.         self.assertEqual(None, node1.__strip__(keep_template_params=False))

  81.         self.assertEqual(None, node2.__strip__(keep_template_params=False))

  82.         self.assertEqual("", node1.__strip__(keep_template_params=True))

  83.         self.assertEqual("bar def", node2.__strip__(keep_template_params=True))

  84.         self.assertEqual("foo bar", node3.__strip__(keep_template_params=True))

  85. 

  86.     def test_showtree(self):

  87.         """test Template.__showtree__()"""

  88.         output = []

  89.         getter, marker = object(), object()

  90.         get = lambda code: output.append((getter, code))

  91.         mark = lambda: output.append(marker)

  92.         node1 = Template(wraptext("foobar"))

  93.         node2 = Template(wraptext("foo"),

  94.                          [pgenh("1", "bar"), pgens("abc", "def")])

  95.         node1.__showtree__(output.append, get, mark)

  96.         node2.__showtree__(output.append, get, mark)

  97.         valid = [

  98.             "{{", (getter, node1.name), "}}", "{{", (getter, node2.name),

  99.             "    | ", marker, (getter, node2.params[0].name), "    = ", marker,

  100.             (getter, node2.params[0].value), "    | ", marker,

  101.             (getter, node2.params[1].name), "    = ", marker,

  102.             (getter, node2.params[1].value), "}}"]

  103.         self.assertEqual(valid, output)

  104. 

  105.     def test_name(self):

  106.         """test getter/setter for the name attribute"""

  107.         name = wraptext("foobar")

  108.         node1 = Template(name)

  109.         node2 = Template(name, [pgenh("1", "bar")])

  110.         self.assertIs(name, node1.name)

  111.         self.assertIs(name, node2.name)

  112.         node1.name = "asdf"

  113.         node2.name = "téstïng"

  114.         self.assertWikicodeEqual(wraptext("asdf"), node1.name)

  115.         self.assertWikicodeEqual(wraptext("téstïng"), node2.name)

  116. 

  117.     def test_params(self):

  118.         """test getter for the params attribute"""

  119.         node1 = Template(wraptext("foobar"))

  120.         plist = [pgenh("1", "bar"), pgens("abc", "def")]

  121.         node2 = Template(wraptext("foo"), plist)

  122.         self.assertEqual([], node1.params)

  123.         self.assertIs(plist, node2.params)

  124. 

  125.     def test_has(self):

  126.         """test Template.has()"""

  127.         node1 = Template(wraptext("foobar"))

  128.         node2 = Template(wraptext("foo"),

  129.                          [pgenh("1", "bar"), pgens("\nabc ", "def")])

  130.         node3 = Template(wraptext("foo"),

  131.                          [pgenh("1", "a"), pgens("b", "c"), pgens("1", "d")])

  132.         node4 = Template(wraptext("foo"), [pgenh("1", "a"), pgens("b", " ")])

  133.         self.assertFalse(node1.has("foobar", False))

  134.         self.assertTrue(node2.has(1, False))

  135.         self.assertTrue(node2.has("abc", False))

  136.         self.assertFalse(node2.has("def", False))

  137.         self.assertTrue(node3.has("1", False))

  138.         self.assertTrue(node3.has(" b ", False))

  139.         self.assertTrue(node4.has("b", False))

  140.         self.assertTrue(node3.has("b", True))

  141.         self.assertFalse(node4.has("b", True))

  142.         self.assertFalse(node1.has_param("foobar", False))

  143.         self.assertTrue(node2.has_param(1, False))

  144. 

  145.     def test_get(self):

  146.         """test Template.get()"""

  147.         node1 = Template(wraptext("foobar"))

  148.         node2p1 = pgenh("1", "bar")

  149.         node2p2 = pgens("abc", "def")

  150.         node2 = Template(wraptext("foo"), [node2p1, node2p2])

  151.         node3p1 = pgens("b", "c")

  152.         node3p2 = pgens("1", "d")

  153.         node3 = Template(wraptext("foo"), [pgenh("1", "a"), node3p1, node3p2])

  154.         node4p1 = pgens(" b", " ")

  155.         node4 = Template(wraptext("foo"), [pgenh("1", "a"), node4p1])

  156.         self.assertRaises(ValueError, node1.get, "foobar")

  157.         self.assertIs(node2p1, node2.get(1))

  158.         self.assertIs(node2p2, node2.get("abc"))

  159.         self.assertRaises(ValueError, node2.get, "def")

  160.         self.assertIs(node3p1, node3.get("b"))

  161.         self.assertIs(node3p2, node3.get("1"))

  162.         self.assertIs(node4p1, node4.get("b "))

  163. 

  164.     def test_add(self):

  165.         """test Template.add()"""

  166.         node1 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "d")])

  167.         node2 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "d")])

  168.         node3 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "d")])

  169.         node4 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "d")])

  170.         node5 = Template(wraptext("a"), [pgens("b", "c"),

  171.                                          pgens("    d ", "e")])

  172.         node6 = Template(wraptext("a"), [pgens("b", "c"), pgens("b", "d"),

  173.                                          pgens("b", "e")])

  174.         node7 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "d")])

  175.         node8p = pgenh("1", "d")

  176.         node8 = Template(wraptext("a"), [pgens("b", "c"), node8p])

  177.         node9 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "d")])

  178.         node10 = Template(wraptext("a"), [pgens("b", "c"), pgenh("1", "e")])

  179.         node11 = Template(wraptext("a"), [pgens("b", "c")])

  180.         node12 = Template(wraptext("a"), [pgens("b", "c")])

  181.         node13 = Template(wraptext("a"), [

  182.             pgens("\nb ", " c"), pgens("\nd ", " e"), pgens("\nf ", " g")])

  183.         node14 = Template(wraptext("a\n"), [

  184.             pgens("b ", "c\n"), pgens("d ", " e"), pgens("f ", "g\n"),

  185.             pgens("h ", " i\n")])

  186.         node15 = Template(wraptext("a"), [

  187.             pgens("b  ", " c\n"), pgens("\nd  ", " e"), pgens("\nf  ", "g ")])

  188.         node16 = Template(wraptext("a"), [

  189.             pgens("\nb ", " c"), pgens("\nd ", " e"), pgens("\nf ", " g")])

  190.         node17 = Template(wraptext("a"), [pgenh("1", "b")])

  191.         node18 = Template(wraptext("a"), [pgenh("1", "b")])

  192.         node19 = Template(wraptext("a"), [pgenh("1", "b")])

  193.         node20 = Template(wraptext("a"), [pgenh("1", "b"), pgenh("2", "c"),

  194.                                           pgenh("3", "d"), pgenh("4", "e")])

  195.         node21 = Template(wraptext("a"), [pgenh("1", "b"), pgenh("2", "c"),

  196.                                           pgens("4", "d"), pgens("5", "e")])

  197.         node22 = Template(wraptext("a"), [pgenh("1", "b"), pgenh("2", "c"),

  198.                                           pgens("4", "d"), pgens("5", "e")])

  199.         node23 = Template(wraptext("a"), [pgenh("1", "b")])

  200.         node24 = Template(wraptext("a"), [pgenh("1", "b")])

  201.         node25 = Template(wraptext("a"), [pgens("b", "c")])

  202.         node26 = Template(wraptext("a"), [pgenh("1", "b")])

  203.         node27 = Template(wraptext("a"), [pgenh("1", "b")])

  204.         node28 = Template(wraptext("a"), [pgens("1", "b")])

  205.         node29 = Template(wraptext("a"), [

  206.             pgens("\nb ", " c"), pgens("\nd ", " e"), pgens("\nf ", " g")])

  207.         node30 = Template(wraptext("a\n"), [

  208.             pgens("b ", "c\n"), pgens("d ", " e"), pgens("f ", "g\n"),

  209.             pgens("h ", " i\n")])

  210.         node31 = Template(wraptext("a"), [

  211.             pgens("b  ", " c\n"), pgens("\nd  ", " e"), pgens("\nf  ", "g ")])

  212.         node32 = Template(wraptext("a"), [

  213.             pgens("\nb ", " c "), pgens("\nd ", " e "), pgens("\nf ", " g ")])

  214.         node33 = Template(wraptext("a"), [pgens("b", "c"), pgens("d", "e"),

  215.                                           pgens("b", "f"), pgens("b", "h"),

  216.                                           pgens("i", "j")])

  217.         node34 = Template(wraptext("a"), [pgens("1", "b"), pgens("x", "y"),

  218.                                           pgens("1", "c"), pgens("2", "d")])

  219.         node35 = Template(wraptext("a"), [pgens("1", "b"), pgens("x", "y"),

  220.                                           pgenh("1", "c"), pgenh("2", "d")])

  221.         node36 = Template(wraptext("a"), [pgens("b", "c"), pgens("d", "e"),

  222.                                           pgens("f", "g")])

  223.         node37 = Template(wraptext("a"), [pgenh("1", "")])

  224.         node38 = Template(wraptext("abc"))

  225.         node39 = Template(wraptext("a"), [pgenh("1", " b ")])

  226.         node40 = Template(wraptext("a"), [pgenh("1", " b"), pgenh("2", " c")])

  227.         node41 = Template(wraptext("a"), [pgens("1", " b"), pgens("2", " c")])

  228.         node42 = Template(wraptext("a"), [pgens("b", "  \n")])

  229. 

  230.         node1.add("e", "f", showkey=True)

  231.         node2.add(2, "g", showkey=False)

  232.         node3.add("e", "foo|bar", showkey=True)

  233.         node4.add("e", "f", showkey=True, before="b")

  234.         node5.add("f", "g", showkey=True, before=" d     ")

  235.         node6.add("f", "g", showkey=True, before="b")

  236.         self.assertRaises(ValueError, node7.add, "e", "f", showkey=True,

  237.                           before="q")

  238.         node8.add("e", "f", showkey=True, before=node8p)

  239.         node9.add("e", "f", showkey=True, before=pgenh("1", "d"))

  240.         self.assertRaises(ValueError, node10.add, "e", "f", showkey=True,

  241.                           before=pgenh("1", "d"))

  242.         node11.add("d", "foo=bar", showkey=True)

  243.         node12.add("1", "foo=bar", showkey=False)

  244.         node13.add("h", "i", showkey=True)

  245.         node14.add("j", "k", showkey=True)

  246.         node15.add("h", "i", showkey=True)

  247.         node16.add("h", "i", showkey=True, preserve_spacing=False)

  248.         node17.add("2", "c")

  249.         node18.add("3", "c")

  250.         node19.add("c", "d")

  251.         node20.add("5", "f")

  252.         node21.add("3", "f")

  253.         node22.add("6", "f")

  254.         node23.add("c", "foo=bar")

  255.         node24.add("2", "foo=bar")

  256.         node25.add("b", "d")

  257.         node26.add("1", "foo=bar")

  258.         node27.add("1", "foo=bar", showkey=True)

  259.         node28.add("1", "foo=bar", showkey=False)

  260.         node29.add("d", "foo")

  261.         node30.add("f", "foo")

  262.         node31.add("f", "foo")

  263.         node32.add("d", "foo", preserve_spacing=False)

  264.         node33.add("b", "k")

  265.         node34.add("1", "e")

  266.         node35.add("1", "e")

  267.         node36.add("d", "h", before="b")

  268.         node37.add(1, "b")

  269.         node38.add("1", "foo")

  270.         self.assertRaises(ValueError, node38.add, "z", "bar", showkey=False)

  271.         node39.add("1", "c")

  272.         node40.add("3", "d")

  273.         node41.add("3", "d")

  274.         node42.add("b", "hello")

  275. 

  276.         self.assertEqual("{{a|b=c|d|e=f}}", node1)

  277.         self.assertEqual("{{a|b=c|d|g}}", node2)

  278.         self.assertEqual("{{a|b=c|d|e=foo&#124;bar}}", node3)

  279.         self.assertIsInstance(node3.params[2].value.get(1), HTMLEntity)

  280.         self.assertEqual("{{a|e=f|b=c|d}}", node4)

  281.         self.assertEqual("{{a|b=c|f=g|    d =e}}", node5)

  282.         self.assertEqual("{{a|b=c|b=d|f=g|b=e}}", node6)

  283.         self.assertEqual("{{a|b=c|d}}", node7)

  284.         self.assertEqual("{{a|b=c|e=f|d}}", node8)

  285.         self.assertEqual("{{a|b=c|e=f|d}}", node9)

  286.         self.assertEqual("{{a|b=c|e}}", node10)

  287.         self.assertEqual("{{a|b=c|d=foo=bar}}", node11)

  288.         self.assertEqual("{{a|b=c|foo&#61;bar}}", node12)

  289.         self.assertIsInstance(node12.params[1].value.get(1), HTMLEntity)

  290.         self.assertEqual("{{a|\nb = c|\nd = e|\nf = g|\nh = i}}", node13)

  291.         self.assertEqual("{{a\n|b =c\n|d = e|f =g\n|h = i\n|j =k\n}}", node14)

  292.         self.assertEqual("{{a|b  = c\n|\nd  = e|\nf  =g |\nh  = i}}", node15)

  293.         self.assertEqual("{{a|\nb = c|\nd = e|\nf = g|h=i}}", node16)

  294.         self.assertEqual("{{a|b|c}}", node17)

  295.         self.assertEqual("{{a|b|3=c}}", node18)

  296.         self.assertEqual("{{a|b|c=d}}", node19)

  297.         self.assertEqual("{{a|b|c|d|e|f}}", node20)

  298.         self.assertEqual("{{a|b|c|4=d|5=e|f}}", node21)

  299.         self.assertEqual("{{a|b|c|4=d|5=e|6=f}}", node22)

  300.         self.assertEqual("{{a|b|c=foo=bar}}", node23)

  301.         self.assertEqual("{{a|b|foo&#61;bar}}", node24)

  302.         self.assertIsInstance(node24.params[1].value.get(1), HTMLEntity)

  303.         self.assertEqual("{{a|b=d}}", node25)

  304.         self.assertEqual("{{a|foo&#61;bar}}", node26)

  305.         self.assertIsInstance(node26.params[0].value.get(1), HTMLEntity)

  306.         self.assertEqual("{{a|1=foo=bar}}", node27)

  307.         self.assertEqual("{{a|foo&#61;bar}}", node28)

  308.         self.assertIsInstance(node28.params[0].value.get(1), HTMLEntity)

  309.         self.assertEqual("{{a|\nb = c|\nd = foo|\nf = g}}", node29)

  310.         self.assertEqual("{{a\n|b =c\n|d = e|f =foo\n|h = i\n}}", node30)

  311.         self.assertEqual("{{a|b  = c\n|\nd  = e|\nf  =foo }}", node31)

  312.         self.assertEqual("{{a|\nb = c |\nd =foo|\nf = g }}", node32)

  313.         self.assertEqual("{{a|b=k|d=e|i=j}}", node33)

  314.         self.assertEqual("{{a|1=e|x=y|2=d}}", node34)

  315.         self.assertEqual("{{a|x=y|e|d}}", node35)

  316.         self.assertEqual("{{a|b=c|d=h|f=g}}", node36)

  317.         self.assertEqual("{{a|b}}", node37)

  318.         self.assertEqual("{{abc|foo}}", node38)

  319.         self.assertEqual("{{a|c}}", node39)

  320.         self.assertEqual("{{a| b| c|d}}", node40)

  321.         self.assertEqual("{{a|1= b|2= c|3= d}}", node41)

  322.         self.assertEqual("{{a|b=hello  \n}}", node42)

  323. 

  324.     def test_remove(self):

  325.         """test Template.remove()"""

  326.         node1 = Template(wraptext("foobar"))

  327.         node2 = Template(wraptext("foo"),

  328.             [pgenh("1", "bar"), pgens("abc", "def")])

  329.         node3 = Template(wraptext("foo"),

  330.             [pgenh("1", "bar"), pgens("abc", "def")])

  331.         node4 = Template(wraptext("foo"),

  332.             [pgenh("1", "bar"), pgenh("2", "baz")])

  333.         node5 = Template(wraptext("foo"), [

  334.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  335.         node6 = Template(wraptext("foo"), [

  336.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  337.         node7 = Template(wraptext("foo"), [

  338.             pgens("1  ", "a"), pgens("  1", "b"), pgens("2", "c")])

  339.         node8 = Template(wraptext("foo"), [

  340.             pgens("1  ", "a"), pgens("  1", "b"), pgens("2", "c")])

  341.         node9 = Template(wraptext("foo"), [

  342.             pgens("1  ", "a"), pgenh("1", "b"), pgenh("2", "c")])

  343.         node10 = Template(wraptext("foo"), [

  344.             pgens("1  ", "a"), pgenh("1", "b"), pgenh("2", "c")])

  345.         node11 = Template(wraptext("foo"), [

  346.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  347.         node12 = Template(wraptext("foo"), [

  348.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  349.         node13 = Template(wraptext("foo"), [

  350.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  351.         node14 = Template(wraptext("foo"), [

  352.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  353.         node15 = Template(wraptext("foo"), [

  354.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  355.         node16 = Template(wraptext("foo"), [

  356.             pgens(" a", "b"), pgens("b", "c"), pgens("a ", "d")])

  357.         node17 = Template(wraptext("foo"), [

  358.             pgens("1  ", "a"), pgenh("1", "b"), pgenh("2", "c")])

  359.         node18 = Template(wraptext("foo"), [

  360.             pgens("1  ", "a"), pgenh("1", "b"), pgenh("2", "c")])

  361.         node19 = Template(wraptext("foo"), [

  362.             pgens("1  ", "a"), pgenh("1", "b"), pgenh("2", "c")])

  363.         node20 = Template(wraptext("foo"), [

  364.             pgens("1  ", "a"), pgenh("1", "b"), pgenh("2", "c")])

  365.         node21 = Template(wraptext("foo"), [

  366.             pgens("a", "b"), pgens("c", "d"), pgens("e", "f"), pgens("a", "b"),

  367.             pgens("a", "b")])

  368.         node22 = Template(wraptext("foo"), [

  369.             pgens("a", "b"), pgens("c", "d"), pgens("e", "f"), pgens("a", "b"),

  370.             pgens("a", "b")])

  371.         node23 = Template(wraptext("foo"), [

  372.             pgens("a", "b"), pgens("c", "d"), pgens("e", "f"), pgens("a", "b"),

  373.             pgens("a", "b")])

  374.         node24 = Template(wraptext("foo"), [

  375.             pgens("a", "b"), pgens("c", "d"), pgens("e", "f"), pgens("a", "b"),

  376.             pgens("a", "b")])

  377.         node25 = Template(wraptext("foo"), [

  378.             pgens("a", "b"), pgens("c", "d"), pgens("e", "f"), pgens("a", "b"),

  379.             pgens("a", "b")])

  380.         node26 = Template(wraptext("foo"), [

  381.             pgens("a", "b"), pgens("c", "d"), pgens("e", "f"), pgens("a", "b"),

  382.             pgens("a", "b")])

  383.         node27 = Template(wraptext("foo"), [pgenh("1", "bar")])

  384.         node28 = Template(wraptext("foo"), [pgenh("1", "bar")])

  385. 

  386.         node2.remove("1")

  387.         node2.remove("abc")

  388.         node3.remove(1, keep_field=True)

  389.         node3.remove("abc", keep_field=True)

  390.         node4.remove("1", keep_field=False)

  391.         node5.remove("a", keep_field=False)

  392.         node6.remove("a", keep_field=True)

  393.         node7.remove(1, keep_field=True)

  394.         node8.remove(1, keep_field=False)

  395.         node9.remove(1, keep_field=True)

  396.         node10.remove(1, keep_field=False)

  397.         node11.remove(node11.params[0], keep_field=False)

  398.         node12.remove(node12.params[0], keep_field=True)

  399.         node13.remove(node13.params[1], keep_field=False)

  400.         node14.remove(node14.params[1], keep_field=True)

  401.         node15.remove(node15.params[2], keep_field=False)

  402.         node16.remove(node16.params[2], keep_field=True)

  403.         node17.remove(node17.params[0], keep_field=False)

  404.         node18.remove(node18.params[0], keep_field=True)

  405.         node19.remove(node19.params[1], keep_field=False)

  406.         node20.remove(node20.params[1], keep_field=True)

  407.         node21.remove("a", keep_field=False)

  408.         node22.remove("a", keep_field=True)

  409.         node23.remove(node23.params[0], keep_field=False)

  410.         node24.remove(node24.params[0], keep_field=True)

  411.         node25.remove(node25.params[3], keep_field=False)

  412.         node26.remove(node26.params[3], keep_field=True)

  413. 

  414.         self.assertRaises(ValueError, node1.remove, 1)

  415.         self.assertRaises(ValueError, node1.remove, "a")

  416.         self.assertRaises(ValueError, node2.remove, "1")

  417.         self.assertEqual("{{foo}}", node2)

  418.         self.assertEqual("{{foo||abc=}}", node3)

  419.         self.assertEqual("{{foo|2=baz}}", node4)

  420.         self.assertEqual("{{foo|b=c}}", node5)

  421.         self.assertEqual("{{foo| a=|b=c}}", node6)

  422.         self.assertEqual("{{foo|1  =|2=c}}", node7)

  423.         self.assertEqual("{{foo|2=c}}", node8)

  424.         self.assertEqual("{{foo||c}}", node9)

  425.         self.assertEqual("{{foo|2=c}}", node10)

  426.         self.assertEqual("{{foo|b=c|a =d}}", node11)

  427.         self.assertEqual("{{foo| a=|b=c|a =d}}", node12)

  428.         self.assertEqual("{{foo| a=b|a =d}}", node13)

  429.         self.assertEqual("{{foo| a=b|b=|a =d}}", node14)

  430.         self.assertEqual("{{foo| a=b|b=c}}", node15)

  431.         self.assertEqual("{{foo| a=b|b=c|a =}}", node16)

  432.         self.assertEqual("{{foo|b|c}}", node17)

  433.         self.assertEqual("{{foo|1  =|b|c}}", node18)

  434.         self.assertEqual("{{foo|1  =a|2=c}}", node19)

  435.         self.assertEqual("{{foo|1  =a||c}}", node20)

  436.         self.assertEqual("{{foo|c=d|e=f}}", node21)

  437.         self.assertEqual("{{foo|a=|c=d|e=f}}", node22)

  438.         self.assertEqual("{{foo|c=d|e=f|a=b|a=b}}", node23)

  439.         self.assertEqual("{{foo|a=|c=d|e=f|a=b|a=b}}", node24)

  440.         self.assertEqual("{{foo|a=b|c=d|e=f|a=b}}", node25)

  441.         self.assertEqual("{{foo|a=b|c=d|e=f|a=|a=b}}", node26)

  442.         self.assertRaises(ValueError, node27.remove, node28.get(1))

  443. 

  444.     def test_formatting(self):

  445.         """test realistic param manipulation with complex whitespace formatting

  446.         (assumes that parsing works correctly)"""

  447.         tests = [

  448.     # https://en.wikipedia.org/w/index.php?title=Lamar_County,_Georgia&oldid=792356004

  449.     ("""{{Infobox U.S. county

  450. | county = Lamar County

  451. | state = Georgia

  452. | seal =

  453. | founded = 1920

  454. | seat wl = Barnesville

  455. | largest city wl = Barnesville

  456. | area_total_sq_mi = 186

  457. | area_land_sq_mi = 184

  458. | area_water_sq_mi = 2.3

  459. | area percentage = 1.3%

  460. | census yr = 2010

  461. | pop = 18317

  462. | density_sq_mi = 100

  463. | time zone = Eastern

  464. | footnotes =

  465. | web = www.lamarcountyga.com

  466. | ex image = Lamar County Georgia Courthouse.jpg

  467. | ex image cap = Lamar County courthouse in Barnesville

  468. | district = 3rd

  469. | named for = [[Lucius Quintus Cincinnatus Lamar II]]

  470. }}""", """{{Infobox U.S. county

  471. | county = Lamar County

  472. | state = Georgia

  473. | seal =

  474. | founded = 1920

  475. | seat wl = Barnesville

  476. | largest city wl = Barnesville

  477. | area_total_sq_mi = 186

  478. | area_land_sq_mi = 184

  479. | area_water_sq_mi = 2.3

  480. | area percentage = 1.3%

  481. | census estimate yr = 2016

  482. | pop = 12345<ref>example ref</ref>

  483. | density_sq_mi = 100

  484. | time zone = Eastern

  485. | footnotes =

  486. | web = www.lamarcountyga.com

  487. | ex image = Lamar County Georgia Courthouse.jpg

  488. | ex image cap = Lamar County courthouse in Barnesville

  489. | district = 3rd

  490. | named for = [[Lucius Quintus Cincinnatus Lamar II]]

  491. }}"""),

  492.     # https://en.wikipedia.org/w/index.php?title=Rockdale_County,_Georgia&oldid=792359760

  493.     ("""{{Infobox U.S. County|

  494.  county = Rockdale County |

  495.  state = Georgia |

  496.  seal =  |

  497.  founded = October 18, 1870 |

  498.  seat wl = Conyers |

  499.  largest city wl = Conyers |

  500.  area_total_sq_mi = 132 |

  501.  area_land_sq_mi = 130 |

  502.  area_water_sq_mi = 2.3 |

  503.  area percentage = 1.7% |

  504.  census yr = 2010|

  505.  pop = 85215 |

  506.  density_sq_mi = 657 |

  507.  web = www.rockdalecounty.org

  508. | ex image = Rockdale-county-courthouse.jpg

  509. | ex image cap = Rockdale County Courthouse in Conyers

  510. | district = 4th

  511. | time zone= Eastern

  512. }}""", """{{Infobox U.S. County|

  513.  county = Rockdale County |

  514.  state = Georgia |

  515.  seal =  |

  516.  founded = October 18, 1870 |

  517.  seat wl = Conyers |

  518.  largest city wl = Conyers |

  519.  area_total_sq_mi = 132 |

  520.  area_land_sq_mi = 130 |

  521.  area_water_sq_mi = 2.3 |

  522.  area percentage = 1.7% |

  523.  census estimate yr = 2016 |

  524.  pop = 12345<ref>example ref</ref> |

  525.  density_sq_mi = 657 |

  526.  web = www.rockdalecounty.org

  527. | ex image = Rockdale-county-courthouse.jpg

  528. | ex image cap = Rockdale County Courthouse in Conyers

  529. | district = 4th

  530. | time zone= Eastern

  531. }}"""),

  532.     # https://en.wikipedia.org/w/index.php?title=Spalding_County,_Georgia&oldid=792360413

  533.     ("""{{Infobox U.S. County|

  534. | county = Spalding County |

  535. | state = Georgia |

  536. | seal =  |

  537. | founded = 1851 |

  538. | seat wl = Griffin |

  539. | largest city wl = Griffin |

  540. | area_total_sq_mi = 200 |

  541. | area_land_sq_mi = 196 |

  542. | area_water_sq_mi = 3.1 |

  543. | area percentage = 1.6% |

  544. | census yr = 2010|

  545. | pop = 64073 |

  546. | density_sq_mi = 326 |

  547. | web = www.spaldingcounty.com |

  548. | named for = [[Thomas Spalding]]

  549. | ex image = Spalding County Courthouse (NE corner).JPG

  550. | ex image cap = Spalding County Courthouse in Griffin

  551. | district = 3rd

  552. | time zone = Eastern

  553. }}""", """{{Infobox U.S. County|

  554. | county = Spalding County |

  555. | state = Georgia |

  556. | seal =  |

  557. | founded = 1851 |

  558. | seat wl = Griffin |

  559. | largest city wl = Griffin |

  560. | area_total_sq_mi = 200 |

  561. | area_land_sq_mi = 196 |

  562. | area_water_sq_mi = 3.1 |

  563. | area percentage = 1.6% |

  564. |

  565. | census estimate yr = 2016 | pop = 12345<ref>example ref</ref> |

  566. | density_sq_mi = 326 |

  567. | web = www.spaldingcounty.com |

  568. | named for = [[Thomas Spalding]]

  569. | ex image = Spalding County Courthouse (NE corner).JPG

  570. | ex image cap = Spalding County Courthouse in Griffin

  571. | district = 3rd

  572. | time zone = Eastern

  573. }}"""),

  574.     # https://en.wikipedia.org/w/index.php?title=Clinton_County,_Illinois&oldid=794694648

  575.     ("""{{Infobox U.S. county

  576.  |county  = Clinton County

  577.  |state = Illinois

  578. | ex image           = File:Clinton County Courthouse, Carlyle.jpg

  579. | ex image cap       = [[Clinton County Courthouse (Illinois)|Clinton County Courthouse]]

  580.  |seal =

  581.  |founded = 1824

  582.  |named for = [[DeWitt Clinton]]

  583.  |seat wl= Carlyle

  584. | largest city wl = Breese

  585.  |time zone=Central

  586.  |area_total_sq_mi = 503

  587.  |area_land_sq_mi = 474

  588.  |area_water_sq_mi = 29

  589.  |area percentage = 5.8%

  590.  |census yr = 2010

  591.  |pop = 37762

  592.  |density_sq_mi = 80

  593.  |web = www.clintonco.illinois.gov

  594. | district = 15th

  595. }}""", """{{Infobox U.S. county

  596.  |county  = Clinton County

  597.  |state = Illinois

  598. | ex image           = File:Clinton County Courthouse, Carlyle.jpg

  599. | ex image cap       = [[Clinton County Courthouse (Illinois)|Clinton County Courthouse]]

  600.  |seal =

  601.  |founded = 1824

  602.  |named for = [[DeWitt Clinton]]

  603.  |seat wl= Carlyle

  604. | largest city wl = Breese

  605.  |time zone=Central

  606.  |area_total_sq_mi = 503

  607.  |area_land_sq_mi = 474

  608.  |area_water_sq_mi = 29

  609.  |area percentage = 5.8%

  610.  |census estimate yr = 2016

  611.  |pop = 12345<ref>example ref</ref>

  612.  |density_sq_mi = 80

  613.  |web = www.clintonco.illinois.gov

  614. | district = 15th

  615. }}"""),

  616.     # https://en.wikipedia.org/w/index.php?title=Winnebago_County,_Illinois&oldid=789193800

  617.     ("""{{Infobox U.S. county |

  618.  county  = Winnebago County |

  619.  state = Illinois |

  620.  seal = Winnebago County il seal.png |

  621.  named for = [[Winnebago (tribe)|Winnebago Tribe]] |

  622.  seat wl= Rockford |

  623.  largest city wl = Rockford|

  624.  area_total_sq_mi = 519 |

  625.  area_land_sq_mi = 513|

  626.  area_water_sq_mi = 5.9 |

  627.  area percentage = 1.1% |

  628.  census yr = 2010|

  629.  pop = 295266 |

  630.  density_sq_mi = 575

  631. | web = www.wincoil.us

  632. | founded year = 1836

  633. | founded date = January 16

  634. | time zone = Central

  635. | district = 16th

  636. | district2 = 17th

  637. }}""", """{{Infobox U.S. county |

  638.  county  = Winnebago County |

  639.  state = Illinois |

  640.  seal = Winnebago County il seal.png |

  641.  named for = [[Winnebago (tribe)|Winnebago Tribe]] |

  642.  seat wl= Rockford |

  643.  largest city wl = Rockford|

  644.  area_total_sq_mi = 519 |

  645.  area_land_sq_mi = 513|

  646.  area_water_sq_mi = 5.9 |

  647.  area percentage = 1.1% |

  648.  census estimate yr = 2016|

  649.  pop = 12345<ref>example ref</ref> |

  650.  density_sq_mi = 575

  651. | web = www.wincoil.us

  652. | founded year = 1836

  653. | founded date = January 16

  654. | time zone = Central

  655. | district = 16th

  656. | district2 = 17th

  657. }}""")]

  658. 

  659.         for (original, expected) in tests:

  660.             code = parse(original)

  661.             template = code.filter_templates()[0]

  662.             template.add("pop", "12345<ref>example ref</ref>")

  663.             template.add('census estimate yr', "2016", before="pop")

  664.             template.remove("census yr")

  665.             self.assertEqual(expected, str(code))

  666. 

  667. if __name__ == "__main__":

  668.     unittest.main(verbosity=2)