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.
867 Commits
1 Branch
7.0 MiB
 
 
 
 
Tree: 03181bcd8b
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '03181bcd8b'
${ noResults }
mwparserfromhell/tests/test_template.py

613 lines
25 KiB
Raw Blame History 

  1. #

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

  3. #

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

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

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

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

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

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

  10. #

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

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

  13. #

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

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

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

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

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

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

  20. # SOFTWARE.

  21. 

  22. from difflib import unified_diff

  23. import pytest

  24. 

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

  26. from mwparserfromhell.nodes.extras import Parameter

  27. from mwparserfromhell import parse

  28. 

  29. from ._test_tree_equality import TreeEqualityTestCase, wrap, wraptext

  30. 

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

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

  33. 

  34. class TestTemplate(TreeEqualityTestCase):

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

  36. 

  37.     def test_unicode(self):

  38.         """test Template.__unicode__()"""

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

  40.         assert "{{foobar}}" == str(node)

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

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

  43.         assert "{{foo|bar|abc=def}}" == str(node2)

  44. 

  45.     def test_children(self):

  46.         """test Template.__children__()"""

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

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

  49.                             showkey=True)

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

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

  52. 

  53.         gen1 = node1.__children__()

  54.         gen2 = node2.__children__()

  55.         assert node1.name == next(gen1)

  56.         assert node2.name == next(gen2)

  57.         assert node2.params[0].value == next(gen2)

  58.         assert node2.params[1].name == next(gen2)

  59.         assert node2.params[1].value == next(gen2)

  60.         with pytest.raises(StopIteration):

  61.             next(gen1)

  62.         with pytest.raises(StopIteration):

  63.             next(gen2)

  64. 

  65.     def test_strip(self):

  66.         """test Template.__strip__()"""

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

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

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

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

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

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

  73.                       showkey=False),

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

  75. 

  76.         assert node1.__strip__(keep_template_params=False) is None

  77.         assert node2.__strip__(keep_template_params=False) is None

  78.         assert "" == node1.__strip__(keep_template_params=True)

  79.         assert "bar def" == node2.__strip__(keep_template_params=True)

  80.         assert "foo bar" == node3.__strip__(keep_template_params=True)

  81. 

  82.     def test_showtree(self):

  83.         """test Template.__showtree__()"""

  84.         output = []

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

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

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

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

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

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

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

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

  93.         valid = [

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

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

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

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

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

  99.         assert valid == output

  100. 

  101.     def test_name(self):

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

  103.         name = wraptext("foobar")

  104.         node1 = Template(name)

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

  106.         assert name is node1.name

  107.         assert name is node2.name

  108.         node1.name = "asdf"

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

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

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

  112. 

  113.     def test_params(self):

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

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

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

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

  118.         assert [] == node1.params

  119.         assert plist is node2.params

  120. 

  121.     def test_has(self):

  122.         """test Template.has()"""

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

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

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

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

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

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

  129.         assert node1.has("foobar", False) is False

  130.         assert node2.has(1, False) is True

  131.         assert node2.has("abc", False) is True

  132.         assert node2.has("def", False) is False

  133.         assert node3.has("1", False) is True

  134.         assert node3.has(" b ", False) is True

  135.         assert node4.has("b", False) is True

  136.         assert node3.has("b", True) is True

  137.         assert node4.has("b", True) is False

  138.         assert node1.has_param("foobar", False) is False

  139.         assert node2.has_param(1, False) is True

  140. 

  141.     def test_get(self):

  142.         """test Template.get()"""

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

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

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

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

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

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

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

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

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

  152.         with pytest.raises(ValueError):

  153.             node1.get("foobar")

  154.         assert node2p1 is node2.get(1)

  155.         assert node2p2 is node2.get("abc")

  156.         with pytest.raises(ValueError):

  157.             node2.get("def")

  158.         assert node3p1 is node3.get("b")

  159.         assert node3p2 is node3.get("1")

  160.         assert node4p1 is node4.get("b ")

  161. 

  162.     def test_add(self):

  163.         """test Template.add()"""

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  227. 

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

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

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

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

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

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

  234.         with pytest.raises(ValueError):

  235.             node7.add("e", "f", showkey=True, before="q")

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

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

  238.         with pytest.raises(ValueError):

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  268.         with pytest.raises(ValueError):

  269.             node38.add("z", "bar", showkey=False)

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

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

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

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

  274. 

  275.         assert "{{a|b=c|d|e=f}}" == node1

  276.         assert "{{a|b=c|d|g}}" == node2

  277.         assert "{{a|b=c|d|e=foo&#124;bar}}" == node3

  278.         assert isinstance(node3.params[2].value.get(1), HTMLEntity)

  279.         assert "{{a|e=f|b=c|d}}" == node4

  280.         assert "{{a|b=c|f=g|    d =e}}" == node5

  281.         assert "{{a|b=c|b=d|f=g|b=e}}" == node6

  282.         assert "{{a|b=c|d}}" == node7

  283.         assert "{{a|b=c|e=f|d}}" == node8

  284.         assert "{{a|b=c|e=f|d}}" == node9

  285.         assert "{{a|b=c|e}}" == node10

  286.         assert "{{a|b=c|d=foo=bar}}" == node11

  287.         assert "{{a|b=c|foo&#61;bar}}" == node12

  288.         assert isinstance(node12.params[1].value.get(1), HTMLEntity)

  289.         assert "{{a|\nb = c|\nd = e|\nf = g|\nh = i}}" == node13

  290.         assert "{{a\n|b =c\n|d = e|f =g\n|h = i\n|j =k\n}}" == node14

  291.         assert "{{a|b  = c\n|\nd  = e|\nf  =g |\nh  = i}}" == node15

  292.         assert "{{a|\nb = c|\nd = e|\nf = g|h=i}}" == node16

  293.         assert "{{a|b|c}}" == node17

  294.         assert "{{a|b|3=c}}" == node18

  295.         assert "{{a|b|c=d}}" == node19

  296.         assert "{{a|b|c|d|e|f}}" == node20

  297.         assert "{{a|b|c|4=d|5=e|f}}" == node21

  298.         assert "{{a|b|c|4=d|5=e|6=f}}" == node22

  299.         assert "{{a|b|c=foo=bar}}" == node23

  300.         assert "{{a|b|foo&#61;bar}}" == node24

  301.         assert isinstance(node24.params[1].value.get(1), HTMLEntity)

  302.         assert "{{a|b=d}}" == node25

  303.         assert "{{a|foo&#61;bar}}" == node26

  304.         assert isinstance(node26.params[0].value.get(1), HTMLEntity)

  305.         assert "{{a|1=foo=bar}}" == node27

  306.         assert "{{a|foo&#61;bar}}" == node28

  307.         assert isinstance(node28.params[0].value.get(1), HTMLEntity)

  308.         assert "{{a|\nb = c|\nd = foo|\nf = g}}" == node29

  309.         assert "{{a\n|b =c\n|d = e|f =foo\n|h = i\n}}" == node30

  310.         assert "{{a|b  = c\n|\nd  = e|\nf  =foo }}" == node31

  311.         assert "{{a|\nb = c |\nd =foo|\nf = g }}" == node32

  312.         assert "{{a|b=k|d=e|i=j}}" == node33

  313.         assert "{{a|1=e|x=y|2=d}}" == node34

  314.         assert "{{a|x=y|e|d}}" == node35

  315.         assert "{{a|b=c|d=h|f=g}}" == node36

  316.         assert "{{a|b}}" == node37

  317.         assert "{{abc|foo}}" == node38

  318.         assert "{{a|c}}" == node39

  319.         assert "{{a| b| c|d}}" == node40

  320.         assert "{{a|1= b|2= c|3= d}}" == node41

  321.         assert "{{a|b=hello  \n}}" == node42

  322. 

  323.     def test_remove(self):

  324.         """test Template.remove()"""

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  384. 

  385.         node2.remove("1")

  386.         node2.remove("abc")

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  412. 

  413.         with pytest.raises(ValueError):

  414.             node1.remove(1)

  415.         with pytest.raises(ValueError):

  416.             node1.remove("a")

  417.         with pytest.raises(ValueError):

  418.             node2.remove("1")

  419.         assert "{{foo}}" == node2

  420.         assert "{{foo||abc=}}" == node3

  421.         assert "{{foo|2=baz}}" == node4

  422.         assert "{{foo|b=c}}" == node5

  423.         assert "{{foo| a=|b=c}}" == node6

  424.         assert "{{foo|1  =|2=c}}" == node7

  425.         assert "{{foo|2=c}}" == node8

  426.         assert "{{foo||c}}" == node9

  427.         assert "{{foo|2=c}}" == node10

  428.         assert "{{foo|b=c|a =d}}" == node11

  429.         assert "{{foo| a=|b=c|a =d}}" == node12

  430.         assert "{{foo| a=b|a =d}}" == node13

  431.         assert "{{foo| a=b|b=|a =d}}" == node14

  432.         assert "{{foo| a=b|b=c}}" == node15

  433.         assert "{{foo| a=b|b=c|a =}}" == node16

  434.         assert "{{foo|b|c}}" == node17

  435.         assert "{{foo|1  =|b|c}}" == node18

  436.         assert "{{foo|1  =a|2=c}}" == node19

  437.         assert "{{foo|1  =a||c}}" == node20

  438.         assert "{{foo|c=d|e=f}}" == node21

  439.         assert "{{foo|a=|c=d|e=f}}" == node22

  440.         assert "{{foo|c=d|e=f|a=b|a=b}}" == node23

  441.         assert "{{foo|a=|c=d|e=f|a=b|a=b}}" == node24

  442.         assert "{{foo|a=b|c=d|e=f|a=b}}" == node25

  443.         assert "{{foo|a=b|c=d|e=f|a=|a=b}}" == node26

  444.         with pytest.raises(ValueError):

  445.             node27.remove(node28.get(1))

  446. 

  447.     def test_formatting(self):

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

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

  450.         tests = [

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

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

  453. | county = Lamar County

  454. | state = Georgia

  455. | seal =

  456. | founded = 1920

  457. | seat wl = Barnesville

  458. | largest city wl = Barnesville

  459. | area_total_sq_mi = 186

  460. | area_land_sq_mi = 184

  461. | area_water_sq_mi = 2.3

  462. | area percentage = 1.3%

  463. | census yr = 2010

  464. | pop = 18317

  465. | density_sq_mi = 100

  466. | time zone = Eastern

  467. | footnotes =

  468. | web = www.lamarcountyga.com

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

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

  471. | district = 3rd

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

  473. }}""",

  474.     """@@ -11,4 +11,4 @@

  475.  | area percentage = 1.3%

  476. -| census yr = 2010

  477. -| pop = 18317

  478. +| census estimate yr = 2016

  479. +| pop = 12345<ref>example ref</ref>

  480.  | density_sq_mi = 100"""),

  481. 

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

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

  484.  county = Rockdale County |

  485.  state = Georgia |

  486.  seal =  |

  487.  founded = October 18, 1870 |

  488.  seat wl = Conyers |

  489.  largest city wl = Conyers |

  490.  area_total_sq_mi = 132 |

  491.  area_land_sq_mi = 130 |

  492.  area_water_sq_mi = 2.3 |

  493.  area percentage = 1.7% |

  494.  census yr = 2010|

  495.  pop = 85215 |

  496.  density_sq_mi = 657 |

  497.  web = www.rockdalecounty.org

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

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

  500. | district = 4th

  501. | time zone= Eastern

  502. }}""",

  503.     """@@ -11,4 +11,4 @@

  504.   area percentage = 1.7% |

  505. - census yr = 2010|

  506. - pop = 85215 |

  507. + census estimate yr = 2016 |

  508. + pop = 12345<ref>example ref</ref> |

  509.   density_sq_mi = 657 |"""),

  510. 

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

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

  513. | county = Spalding County |

  514. | state = Georgia |

  515. | seal =  |

  516. | founded = 1851 |

  517. | seat wl = Griffin |

  518. | largest city wl = Griffin |

  519. | area_total_sq_mi = 200 |

  520. | area_land_sq_mi = 196 |

  521. | area_water_sq_mi = 3.1 |

  522. | area percentage = 1.6% |

  523. | census yr = 2010|

  524. | pop = 64073 |

  525. | density_sq_mi = 326 |

  526. | web = www.spaldingcounty.com |

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

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

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

  530. | district = 3rd

  531. | time zone = Eastern

  532. }}""",

  533.     """@@ -11,4 +11,4 @@

  534.  | area percentage = 1.6% |

  535. -| census yr = 2010|

  536. -| pop = 64073 |

  537. +|

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

  539.  | density_sq_mi = 326 |"""),

  540. 

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

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

  543.  |county  = Clinton County

  544.  |state = Illinois

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

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

  547.  |seal =

  548.  |founded = 1824

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

  550.  |seat wl= Carlyle

  551. | largest city wl = Breese

  552.  |time zone=Central

  553.  |area_total_sq_mi = 503

  554.  |area_land_sq_mi = 474

  555.  |area_water_sq_mi = 29

  556.  |area percentage = 5.8%

  557.  |census yr = 2010

  558.  |pop = 37762

  559.  |density_sq_mi = 80

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

  561. | district = 15th

  562. }}""",

  563.     """@@ -15,4 +15,4 @@

  564.   |area percentage = 5.8%

  565. - |census yr = 2010

  566. - |pop = 37762

  567. + |census estimate yr = 2016

  568. + |pop = 12345<ref>example ref</ref>

  569.   |density_sq_mi = 80"""),

  570. 

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

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

  573.  county  = Winnebago County |

  574.  state = Illinois |

  575.  seal = Winnebago County il seal.png |

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

  577.  seat wl= Rockford |

  578.  largest city wl = Rockford|

  579.  area_total_sq_mi = 519 |

  580.  area_land_sq_mi = 513|

  581.  area_water_sq_mi = 5.9 |

  582.  area percentage = 1.1% |

  583.  census yr = 2010|

  584.  pop = 295266 |

  585.  density_sq_mi = 575

  586. | web = www.wincoil.us

  587. | founded year = 1836

  588. | founded date = January 16

  589. | time zone = Central

  590. | district = 16th

  591. | district2 = 17th

  592. }}""",

  593.     """@@ -11,4 +11,4 @@

  594.   area percentage = 1.1% |

  595. - census yr = 2010|

  596. - pop = 295266 |

  597. + census estimate yr = 2016|

  598. + pop = 12345<ref>example ref</ref> |

  599.   density_sq_mi = 575""")]

  600. 

  601.         for (original, expected) in tests:

  602.             code = parse(original)

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

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

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

  606.             template.remove("census yr")

  607. 

  608.             oldlines = original.splitlines(True)

  609.             newlines = str(code).splitlines(True)

  610.             difflines = unified_diff(oldlines, newlines, n=1)

  611.             diff = "".join(list(difflines)[2:]).strip()

  612.             assert expected == diff