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

612 lines
25 KiB
Raw Blame History 

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

  2. #

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

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

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

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

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

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

  9. #

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

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

  12. #

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

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

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

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

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

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

  19. # SOFTWARE.

  20. 

  21. from difflib import unified_diff

  22. import pytest

  23. 

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

  25. from mwparserfromhell.nodes.extras import Parameter

  26. from mwparserfromhell import parse

  27. 

  28. from ._test_tree_equality import TreeEqualityTestCase, wrap, wraptext

  29. 

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

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

  32. 

  33. class TestTemplate(TreeEqualityTestCase):

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

  35. 

  36.     def test_str(self):

  37.         """test Template.__str__()"""

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

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

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

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

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

  43. 

  44.     def test_children(self):

  45.         """test Template.__children__()"""

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

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

  48.                             showkey=True)

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

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

  51. 

  52.         gen1 = node1.__children__()

  53.         gen2 = node2.__children__()

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

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

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

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

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

  59.         with pytest.raises(StopIteration):

  60.             next(gen1)

  61.         with pytest.raises(StopIteration):

  62.             next(gen2)

  63. 

  64.     def test_strip(self):

  65.         """test Template.__strip__()"""

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

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

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

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

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

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

  72.                       showkey=False),

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

  74. 

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

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

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

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

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

  80. 

  81.     def test_showtree(self):

  82.         """test Template.__showtree__()"""

  83.         output = []

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

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

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

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

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

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

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

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

  92.         valid = [

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

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

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

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

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

  98.         assert valid == output

  99. 

  100.     def test_name(self):

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

  102.         name = wraptext("foobar")

  103.         node1 = Template(name)

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

  105.         assert name is node1.name

  106.         assert name is node2.name

  107.         node1.name = "asdf"

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

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

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

  111. 

  112.     def test_params(self):

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

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

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

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

  117.         assert [] == node1.params

  118.         assert plist is node2.params

  119. 

  120.     def test_has(self):

  121.         """test Template.has()"""

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  139. 

  140.     def test_get(self):

  141.         """test Template.get()"""

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

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

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

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

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

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

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

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

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

  151.         with pytest.raises(ValueError):

  152.             node1.get("foobar")

  153.         assert node2p1 is node2.get(1)

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

  155.         with pytest.raises(ValueError):

  156.             node2.get("def")

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

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

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

  160. 

  161.     def test_add(self):

  162.         """test Template.add()"""

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  226. 

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

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

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

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

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

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

  233.         with pytest.raises(ValueError):

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

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

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

  237.         with pytest.raises(ValueError):

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  267.         with pytest.raises(ValueError):

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

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

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

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

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

  273. 

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  321. 

  322.     def test_remove(self):

  323.         """test Template.remove()"""

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  383. 

  384.         node2.remove("1")

  385.         node2.remove("abc")

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  411. 

  412.         with pytest.raises(ValueError):

  413.             node1.remove(1)

  414.         with pytest.raises(ValueError):

  415.             node1.remove("a")

  416.         with pytest.raises(ValueError):

  417.             node2.remove("1")

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  443.         with pytest.raises(ValueError):

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

  445. 

  446.     def test_formatting(self):

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

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

  449.         tests = [

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

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

  452. | county = Lamar County

  453. | state = Georgia

  454. | seal =

  455. | founded = 1920

  456. | seat wl = Barnesville

  457. | largest city wl = Barnesville

  458. | area_total_sq_mi = 186

  459. | area_land_sq_mi = 184

  460. | area_water_sq_mi = 2.3

  461. | area percentage = 1.3%

  462. | census yr = 2010

  463. | pop = 18317

  464. | density_sq_mi = 100

  465. | time zone = Eastern

  466. | footnotes =

  467. | web = www.lamarcountyga.com

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

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

  470. | district = 3rd

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

  472. }}""",

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

  474.  | area percentage = 1.3%

  475. -| census yr = 2010

  476. -| pop = 18317

  477. +| census estimate yr = 2016

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

  479.  | density_sq_mi = 100"""),

  480. 

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

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

  483.  county = Rockdale County |

  484.  state = Georgia |

  485.  seal =  |

  486.  founded = October 18, 1870 |

  487.  seat wl = Conyers |

  488.  largest city wl = Conyers |

  489.  area_total_sq_mi = 132 |

  490.  area_land_sq_mi = 130 |

  491.  area_water_sq_mi = 2.3 |

  492.  area percentage = 1.7% |

  493.  census yr = 2010|

  494.  pop = 85215 |

  495.  density_sq_mi = 657 |

  496.  web = www.rockdalecounty.org

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

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

  499. | district = 4th

  500. | time zone= Eastern

  501. }}""",

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

  503.   area percentage = 1.7% |

  504. - census yr = 2010|

  505. - pop = 85215 |

  506. + census estimate yr = 2016 |

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

  508.   density_sq_mi = 657 |"""),

  509. 

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

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

  512. | county = Spalding County |

  513. | state = Georgia |

  514. | seal =  |

  515. | founded = 1851 |

  516. | seat wl = Griffin |

  517. | largest city wl = Griffin |

  518. | area_total_sq_mi = 200 |

  519. | area_land_sq_mi = 196 |

  520. | area_water_sq_mi = 3.1 |

  521. | area percentage = 1.6% |

  522. | census yr = 2010|

  523. | pop = 64073 |

  524. | density_sq_mi = 326 |

  525. | web = www.spaldingcounty.com |

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

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

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

  529. | district = 3rd

  530. | time zone = Eastern

  531. }}""",

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

  533.  | area percentage = 1.6% |

  534. -| census yr = 2010|

  535. -| pop = 64073 |

  536. +|

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

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

  539. 

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

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

  542.  |county  = Clinton County

  543.  |state = Illinois

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

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

  546.  |seal =

  547.  |founded = 1824

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

  549.  |seat wl= Carlyle

  550. | largest city wl = Breese

  551.  |time zone=Central

  552.  |area_total_sq_mi = 503

  553.  |area_land_sq_mi = 474

  554.  |area_water_sq_mi = 29

  555.  |area percentage = 5.8%

  556.  |census yr = 2010

  557.  |pop = 37762

  558.  |density_sq_mi = 80

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

  560. | district = 15th

  561. }}""",

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

  563.   |area percentage = 5.8%

  564. - |census yr = 2010

  565. - |pop = 37762

  566. + |census estimate yr = 2016

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

  568.   |density_sq_mi = 80"""),

  569. 

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

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

  572.  county  = Winnebago County |

  573.  state = Illinois |

  574.  seal = Winnebago County il seal.png |

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

  576.  seat wl= Rockford |

  577.  largest city wl = Rockford|

  578.  area_total_sq_mi = 519 |

  579.  area_land_sq_mi = 513|

  580.  area_water_sq_mi = 5.9 |

  581.  area percentage = 1.1% |

  582.  census yr = 2010|

  583.  pop = 295266 |

  584.  density_sq_mi = 575

  585. | web = www.wincoil.us

  586. | founded year = 1836

  587. | founded date = January 16

  588. | time zone = Central

  589. | district = 16th

  590. | district2 = 17th

  591. }}""",

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

  593.   area percentage = 1.1% |

  594. - census yr = 2010|

  595. - pop = 295266 |

  596. + census estimate yr = 2016|

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

  598.   density_sq_mi = 575""")]

  599. 

  600.         for (original, expected) in tests:

  601.             code = parse(original)

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

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

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

  605.             template.remove("census yr")

  606. 

  607.             oldlines = original.splitlines(True)

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

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

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

  611.             assert expected == diff