ben
/
mwparserfromhell
kopie van https://github.com/earwig/mwparserfromhell
1
1
Vork 0
Code Publicaties 24 Activiteit
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.
881 Commits
1 Branch
5.9 MiB
 
 
 
 
Tree: 2721de2447
Branches Labels
${ item.name }
Maak branch ${ searchTerm }
van '2721de2447'
${ noResults }
mwparserfromhell/tests/test_smart_list.py

410 regels
16 KiB
Ruw Blame Geschiedenis 

  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. import unittest

  22. 

  23. from mwparserfromhell.smart_list import SmartList

  24. from mwparserfromhell.smart_list.list_proxy import ListProxy

  25. 

  26. 

  27. class TestSmartList(unittest.TestCase):

  28.     """Test cases for the SmartList class and its child, ListProxy."""

  29. 

  30.     def _test_get_set_del_item(self, builder):

  31.         """Run tests on __get/set/delitem__ of a list built with *builder*."""

  32.         def assign(L, s1, s2, s3, val):

  33.             L[s1:s2:s3] = val

  34.         def delete(L, s1):

  35.             del L[s1]

  36. 

  37.         list1 = builder([0, 1, 2, 3, "one", "two"])

  38.         list2 = builder(list(range(10)))

  39. 

  40.         self.assertEqual(1, list1[1])

  41.         self.assertEqual("one", list1[-2])

  42.         self.assertEqual([2, 3], list1[2:4])

  43.         self.assertRaises(IndexError, lambda: list1[6])

  44.         self.assertRaises(IndexError, lambda: list1[-7])

  45. 

  46.         self.assertEqual([0, 1, 2], list1[:3])

  47.         self.assertEqual([0, 1, 2, 3, "one", "two"], list1[:])

  48.         self.assertEqual([3, "one", "two"], list1[3:])

  49.         self.assertEqual([3, "one", "two"], list1[3:100])

  50.         self.assertEqual(["one", "two"], list1[-2:])

  51.         self.assertEqual([0, 1], list1[:-4])

  52.         self.assertEqual([], list1[6:])

  53.         self.assertEqual([], list1[4:2])

  54. 

  55.         self.assertEqual([0, 2, "one"], list1[0:5:2])

  56.         self.assertEqual([0, 2], list1[0:-3:2])

  57.         self.assertEqual([0, 1, 2, 3, "one", "two"], list1[::])

  58.         self.assertEqual([2, 3, "one", "two"], list1[2::])

  59.         self.assertEqual([0, 1, 2, 3], list1[:4:])

  60.         self.assertEqual([2, 3], list1[2:4:])

  61.         self.assertEqual([0, 2, 4, 6, 8], list2[::2])

  62.         self.assertEqual([2, 5, 8], list2[2::3])

  63.         self.assertEqual([0, 3], list2[:6:3])

  64.         self.assertEqual([2, 5, 8], list2[-8:9:3])

  65.         self.assertEqual([], list2[100000:1000:-100])

  66. 

  67.         list1[3] = 100

  68.         self.assertEqual(100, list1[3])

  69.         list1[-3] = 101

  70.         self.assertEqual([0, 1, 2, 101, "one", "two"], list1)

  71.         list1[5:] = [6, 7, 8]

  72.         self.assertEqual([6, 7, 8], list1[5:])

  73.         self.assertEqual([0, 1, 2, 101, "one", 6, 7, 8], list1)

  74.         list1[2:4] = [-1, -2, -3, -4, -5]

  75.         self.assertEqual([0, 1, -1, -2, -3, -4, -5, "one", 6, 7, 8], list1)

  76.         list1[0:-3] = [99]

  77.         self.assertEqual([99, 6, 7, 8], list1)

  78.         list2[0:6:2] = [100, 102, 104]

  79.         self.assertEqual([100, 1, 102, 3, 104, 5, 6, 7, 8, 9], list2)

  80.         list2[::3] = [200, 203, 206, 209]

  81.         self.assertEqual([200, 1, 102, 203, 104, 5, 206, 7, 8, 209], list2)

  82.         list2[::] = range(7)

  83.         self.assertEqual([0, 1, 2, 3, 4, 5, 6], list2)

  84.         self.assertRaises(ValueError, assign, list2, 0, 5, 2,

  85.                           [100, 102, 104, 106])

  86.         with self.assertRaises(IndexError):

  87.             list2[7] = "foo"

  88.         with self.assertRaises(IndexError):

  89.             list2[-8] = "foo"

  90. 

  91.         del list2[2]

  92.         self.assertEqual([0, 1, 3, 4, 5, 6], list2)

  93.         del list2[-3]

  94.         self.assertEqual([0, 1, 3, 5, 6], list2)

  95.         self.assertRaises(IndexError, delete, list2, 100)

  96.         self.assertRaises(IndexError, delete, list2, -6)

  97.         list2[:] = range(10)

  98.         del list2[3:6]

  99.         self.assertEqual([0, 1, 2, 6, 7, 8, 9], list2)

  100.         del list2[-2:]

  101.         self.assertEqual([0, 1, 2, 6, 7], list2)

  102.         del list2[:2]

  103.         self.assertEqual([2, 6, 7], list2)

  104.         list2[:] = range(10)

  105.         del list2[2:8:2]

  106.         self.assertEqual([0, 1, 3, 5, 7, 8, 9], list2)

  107. 

  108.     def _test_add_radd_iadd(self, builder):

  109.         """Run tests on __r/i/add__ of a list built with *builder*."""

  110.         list1 = builder(range(5))

  111.         list2 = builder(range(5, 10))

  112.         self.assertEqual([0, 1, 2, 3, 4, 5, 6], list1 + [5, 6])

  113.         self.assertEqual([0, 1, 2, 3, 4], list1)

  114.         self.assertEqual(list(range(10)), list1 + list2)

  115.         self.assertEqual([-2, -1, 0, 1, 2, 3, 4], [-2, -1] + list1)

  116.         self.assertEqual([0, 1, 2, 3, 4], list1)

  117.         list1 += ["foo", "bar", "baz"]

  118.         self.assertEqual([0, 1, 2, 3, 4, "foo", "bar", "baz"], list1)

  119. 

  120.     def _test_other_magic_methods(self, builder):

  121.         """Run tests on other magic methods of a list built with *builder*."""

  122.         list1 = builder([0, 1, 2, 3, "one", "two"])

  123.         list2 = builder([])

  124.         list3 = builder([0, 2, 3, 4])

  125.         list4 = builder([0, 1, 2])

  126. 

  127.         self.assertEqual("[0, 1, 2, 3, 'one', 'two']", str(list1))

  128.         self.assertEqual(b"\x00\x01\x02", bytes(list4))

  129.         self.assertEqual("[0, 1, 2, 3, 'one', 'two']", repr(list1))

  130. 

  131.         self.assertLess(list1, list3)

  132.         self.assertLessEqual(list1, list3)

  133.         self.assertNotEqual(list1, list3)

  134.         self.assertNotEqual(list1, list3)

  135.         self.assertLessEqual(list1, list3)

  136.         self.assertLess(list1, list3)

  137. 

  138.         other1 = [0, 2, 3, 4]

  139.         self.assertLess(list1, other1)

  140.         self.assertLessEqual(list1, other1)

  141.         self.assertNotEqual(list1, other1)

  142.         self.assertNotEqual(list1, other1)

  143.         self.assertLessEqual(list1, other1)

  144.         self.assertLess(list1, other1)

  145. 

  146.         other2 = [0, 0, 1, 2]

  147.         self.assertGreaterEqual(list1, other2)

  148.         self.assertGreater(list1, other2)

  149.         self.assertNotEqual(list1, other2)

  150.         self.assertNotEqual(list1, other2)

  151.         self.assertGreater(list1, other2)

  152.         self.assertGreaterEqual(list1, other2)

  153. 

  154.         other3 = [0, 1, 2, 3, "one", "two"]

  155.         self.assertGreaterEqual(list1, other3)

  156.         self.assertLessEqual(list1, other3)

  157.         self.assertEqual(list1, other3)

  158.         self.assertEqual(list1, other3)

  159.         self.assertLessEqual(list1, other3)

  160.         self.assertGreaterEqual(list1, other3)

  161. 

  162.         self.assertTrue(bool(list1))

  163.         self.assertFalse(bool(list2))

  164. 

  165.         self.assertEqual(6, len(list1))

  166.         self.assertEqual(0, len(list2))

  167. 

  168.         out = []

  169.         for obj in list1:

  170.             out.append(obj)

  171.         self.assertEqual([0, 1, 2, 3, "one", "two"], out)

  172. 

  173.         out = []

  174.         for ch in list2:

  175.             out.append(ch)

  176.         self.assertEqual([], out)

  177. 

  178.         gen1 = iter(list1)

  179.         out = []

  180.         for _ in range(len(list1)):

  181.             out.append(next(gen1))

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

  183.         self.assertEqual([0, 1, 2, 3, "one", "two"], out)

  184.         gen2 = iter(list2)

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

  186. 

  187.         self.assertEqual(["two", "one", 3, 2, 1, 0], list(reversed(list1)))

  188.         self.assertEqual([], list(reversed(list2)))

  189. 

  190.         self.assertIn("one", list1)

  191.         self.assertIn(3, list1)

  192.         self.assertNotIn(10, list1)

  193.         self.assertNotIn(0, list2)

  194. 

  195.         self.assertEqual([], list2 * 5)

  196.         self.assertEqual([], 5 * list2)

  197.         self.assertEqual([0, 1, 2, 0, 1, 2, 0, 1, 2], list4 * 3)

  198.         self.assertEqual([0, 1, 2, 0, 1, 2, 0, 1, 2], 3 * list4)

  199.         list4 *= 2

  200.         self.assertEqual([0, 1, 2, 0, 1, 2], list4)

  201. 

  202.     def _test_list_methods(self, builder):

  203.         """Run tests on the public methods of a list built with *builder*."""

  204.         list1 = builder(range(5))

  205.         list2 = builder(["foo"])

  206.         list3 = builder([("a", 5), ("d", 2), ("b", 8), ("c", 3)])

  207. 

  208.         list1.append(5)

  209.         list1.append(1)

  210.         list1.append(2)

  211.         self.assertEqual([0, 1, 2, 3, 4, 5, 1, 2], list1)

  212. 

  213.         self.assertEqual(0, list1.count(6))

  214.         self.assertEqual(2, list1.count(1))

  215. 

  216.         list1.extend(range(5, 8))

  217.         self.assertEqual([0, 1, 2, 3, 4, 5, 1, 2, 5, 6, 7], list1)

  218. 

  219.         self.assertEqual(1, list1.index(1))

  220.         self.assertEqual(6, list1.index(1, 3))

  221.         self.assertEqual(6, list1.index(1, 3, 7))

  222.         self.assertRaises(ValueError, list1.index, 1, 3, 5)

  223. 

  224.         list1.insert(0, -1)

  225.         self.assertEqual([-1, 0, 1, 2, 3, 4, 5, 1, 2, 5, 6, 7], list1)

  226.         list1.insert(-1, 6.5)

  227.         self.assertEqual([-1, 0, 1, 2, 3, 4, 5, 1, 2, 5, 6, 6.5, 7], list1)

  228.         list1.insert(13, 8)

  229.         self.assertEqual([-1, 0, 1, 2, 3, 4, 5, 1, 2, 5, 6, 6.5, 7, 8], list1)

  230. 

  231.         self.assertEqual(8, list1.pop())

  232.         self.assertEqual(7, list1.pop())

  233.         self.assertEqual([-1, 0, 1, 2, 3, 4, 5, 1, 2, 5, 6, 6.5], list1)

  234.         self.assertEqual(-1, list1.pop(0))

  235.         self.assertEqual(5, list1.pop(5))

  236.         self.assertEqual(6.5, list1.pop(-1))

  237.         self.assertEqual([0, 1, 2, 3, 4, 1, 2, 5, 6], list1)

  238.         self.assertEqual("foo", list2.pop())

  239.         self.assertRaises(IndexError, list2.pop)

  240.         self.assertEqual([], list2)

  241. 

  242.         list1.remove(6)

  243.         self.assertEqual([0, 1, 2, 3, 4, 1, 2, 5], list1)

  244.         list1.remove(1)

  245.         self.assertEqual([0, 2, 3, 4, 1, 2, 5], list1)

  246.         list1.remove(1)

  247.         self.assertEqual([0, 2, 3, 4, 2, 5], list1)

  248.         self.assertRaises(ValueError, list1.remove, 1)

  249. 

  250.         list1.reverse()

  251.         self.assertEqual([5, 2, 4, 3, 2, 0], list1)

  252. 

  253.         list1.sort()

  254.         self.assertEqual([0, 2, 2, 3, 4, 5], list1)

  255.         list1.sort(reverse=True)

  256.         self.assertEqual([5, 4, 3, 2, 2, 0], list1)

  257.         list3.sort(key=lambda i: i[1])

  258.         self.assertEqual([("d", 2), ("c", 3), ("a", 5), ("b", 8)], list3)

  259.         list3.sort(key=lambda i: i[1], reverse=True)

  260.         self.assertEqual([("b", 8), ("a", 5), ("c", 3), ("d", 2)], list3)

  261. 

  262.     @staticmethod

  263.     def _dispatch_test_for_children(meth):

  264.         """Run a test method on various different types of children."""

  265.         meth(lambda L: SmartList(list(L))[:])

  266.         meth(lambda L: SmartList([999] + list(L))[1:])

  267.         meth(lambda L: SmartList(list(L) + [999])[:-1])

  268.         meth(lambda L: SmartList([101, 102] + list(L) + [201, 202])[2:-2])

  269. 

  270.     def test_docs(self):

  271.         """make sure the methods of SmartList/ListProxy have docstrings"""

  272.         methods = ["append", "count", "extend", "index", "insert", "pop",

  273.                    "remove", "reverse", "sort"]

  274.         for meth in methods:

  275.             expected = getattr(list, meth).__doc__

  276.             smartlist_doc = getattr(SmartList, meth).__doc__

  277.             listproxy_doc = getattr(ListProxy, meth).__doc__

  278.             self.assertEqual(expected, smartlist_doc)

  279.             self.assertEqual(expected, listproxy_doc)

  280. 

  281.     def test_doctest(self):

  282.         """make sure the test embedded in SmartList's docstring passes"""

  283.         parent = SmartList([0, 1, 2, 3])

  284.         self.assertEqual([0, 1, 2, 3], parent)

  285.         child = parent[2:]

  286.         self.assertEqual([2, 3], child)

  287.         child.append(4)

  288.         self.assertEqual([2, 3, 4], child)

  289.         self.assertEqual([0, 1, 2, 3, 4], parent)

  290. 

  291.     def test_parent_get_set_del(self):

  292.         """make sure SmartList's getitem/setitem/delitem work"""

  293.         self._test_get_set_del_item(SmartList)

  294. 

  295.     def test_parent_add(self):

  296.         """make sure SmartList's add/radd/iadd work"""

  297.         self._test_add_radd_iadd(SmartList)

  298. 

  299.     def test_parent_other_magics(self):

  300.         """make sure SmartList's other magically implemented features work"""

  301.         self._test_other_magic_methods(SmartList)

  302. 

  303.     def test_parent_methods(self):

  304.         """make sure SmartList's non-magic methods work, like append()"""

  305.         self._test_list_methods(SmartList)

  306. 

  307.     def test_child_get_set_del(self):

  308.         """make sure ListProxy's getitem/setitem/delitem work"""

  309.         self._dispatch_test_for_children(self._test_get_set_del_item)

  310. 

  311.     def test_child_add(self):

  312.         """make sure ListProxy's add/radd/iadd work"""

  313.         self._dispatch_test_for_children(self._test_add_radd_iadd)

  314. 

  315.     def test_child_other_magics(self):

  316.         """make sure ListProxy's other magically implemented features work"""

  317.         self._dispatch_test_for_children(self._test_other_magic_methods)

  318. 

  319.     def test_child_methods(self):

  320.         """make sure ListProxy's non-magic methods work, like append()"""

  321.         self._dispatch_test_for_children(self._test_list_methods)

  322. 

  323.     def test_influence(self):

  324.         """make sure changes are propagated from parents to children"""

  325.         parent = SmartList([0, 1, 2, 3, 4, 5])

  326.         child1 = parent[2:]

  327.         child2 = parent[2:5]

  328.         self.assertEqual([0, 1, 2, 3, 4, 5], parent)

  329.         self.assertEqual([2, 3, 4, 5], child1)

  330.         self.assertEqual([2, 3, 4], child2)

  331.         self.assertEqual(2, len(parent._children))

  332. 

  333.         parent.append(6)

  334.         child1.append(7)

  335.         child2.append(4.5)

  336.         self.assertEqual([0, 1, 2, 3, 4, 4.5, 5, 6, 7], parent)

  337.         self.assertEqual([2, 3, 4, 4.5, 5, 6, 7], child1)

  338.         self.assertEqual([2, 3, 4, 4.5], child2)

  339. 

  340.         parent.insert(0, -1)

  341.         parent.insert(4, 2.5)

  342.         parent.insert(10, 6.5)

  343.         self.assertEqual([-1, 0, 1, 2, 2.5, 3, 4, 4.5, 5, 6, 6.5, 7], parent)

  344.         self.assertEqual([2, 2.5, 3, 4, 4.5, 5, 6, 6.5, 7], child1)

  345.         self.assertEqual([2, 2.5, 3, 4, 4.5], child2)

  346. 

  347.         self.assertEqual(7, parent.pop())

  348.         self.assertEqual(6.5, child1.pop())

  349.         self.assertEqual(4.5, child2.pop())

  350.         self.assertEqual([-1, 0, 1, 2, 2.5, 3, 4, 5, 6], parent)

  351.         self.assertEqual([2, 2.5, 3, 4, 5, 6], child1)

  352.         self.assertEqual([2, 2.5, 3, 4], child2)

  353. 

  354.         parent.remove(-1)

  355.         child1.remove(2.5)

  356.         self.assertEqual([0, 1, 2, 3, 4, 5, 6], parent)

  357.         self.assertEqual([2, 3, 4, 5, 6], child1)

  358.         self.assertEqual([2, 3, 4], child2)

  359. 

  360.         self.assertEqual(0, parent.pop(0))

  361.         self.assertEqual([1, 2, 3, 4, 5, 6], parent)

  362.         self.assertEqual([2, 3, 4, 5, 6], child1)

  363.         self.assertEqual([2, 3, 4], child2)

  364. 

  365.         child2.reverse()

  366.         self.assertEqual([1, 4, 3, 2, 5, 6], parent)

  367.         self.assertEqual([4, 3, 2, 5, 6], child1)

  368.         self.assertEqual([4, 3, 2], child2)

  369. 

  370.         parent.extend([7, 8])

  371.         child1.extend([8.1, 8.2])

  372.         child2.extend([1.9, 1.8])

  373.         self.assertEqual([1, 4, 3, 2, 1.9, 1.8, 5, 6, 7, 8, 8.1, 8.2], parent)

  374.         self.assertEqual([4, 3, 2, 1.9, 1.8, 5, 6, 7, 8, 8.1, 8.2], child1)

  375.         self.assertEqual([4, 3, 2, 1.9, 1.8], child2)

  376. 

  377.         child3 = parent[9:]

  378.         self.assertEqual([8, 8.1, 8.2], child3)

  379. 

  380.         del parent[8:]

  381.         self.assertEqual([1, 4, 3, 2, 1.9, 1.8, 5, 6], parent)

  382.         self.assertEqual([4, 3, 2, 1.9, 1.8, 5, 6], child1)

  383.         self.assertEqual([4, 3, 2, 1.9, 1.8], child2)

  384.         self.assertEqual([], child3)

  385.         self.assertEqual(0, len(child3))

  386. 

  387.         del child1

  388.         self.assertEqual([1, 4, 3, 2, 1.9, 1.8, 5, 6], parent)

  389.         self.assertEqual([4, 3, 2, 1.9, 1.8], child2)

  390.         self.assertEqual([], child3)

  391.         self.assertEqual(2, len(parent._children))

  392. 

  393.         del child3

  394.         self.assertEqual([1, 4, 3, 2, 1.9, 1.8, 5, 6], parent)

  395.         self.assertEqual([4, 3, 2, 1.9, 1.8], child2)

  396.         self.assertEqual(1, len(parent._children))

  397. 

  398.         parent.remove(1.9)

  399.         parent.remove(1.8)

  400.         self.assertEqual([1, 4, 3, 2, 5, 6], parent)

  401.         self.assertEqual([4, 3, 2], child2)

  402. 

  403.         parent.reverse()

  404.         self.assertEqual([6, 5, 2, 3, 4, 1], parent)

  405.         self.assertEqual([4, 3, 2], child2)

  406.         self.assertEqual(0, len(parent._children))

  407. 

  408. if __name__ == "__main__":

  409.     unittest.main(verbosity=2)