ben
/
mwparserfromhell
의 미러 https://github.com/earwig/mwparserfromhell
1
1
포크 0
코드 릴리즈 24 활동
A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
238 커밋
1 브렌치
21 MiB
 
 
 
 
트리: 6a741db7ce
브랜치 태그
${ item.name }
${ searchTerm } 브랜치 생성
from '6a741db7ce'
${ noResults }
mwparserfromhell/tests/test_smart_list.py

389 lines
16 KiB
Raw Blame 히스토리 

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

  2. #

  3. # Copyright (C) 2012-2013 Ben Kurtovic <ben.kurtovic@verizon.net>

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

  25. 

  26. from mwparserfromhell.compat import py3k, range

  27. from mwparserfromhell.smart_list import SmartList, _ListProxy

  28. 

  29. class TestSmartList(unittest.TestCase):

  30.     """Test cases for the SmartList class and its child, _ListProxy."""

  31. 

  32.     def _test_get_set_del_item(self, builder):

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

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

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

  36.         def delete(L, s1):

  37.             del L[s1]

  38. 

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

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

  41. 

  42.         self.assertEquals(1, list1[1])

  43.         self.assertEquals("one", list1[-2])

  44.         self.assertEquals([2, 3], list1[2:4])

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

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

  47. 

  48.         self.assertEquals([0, 1, 2], list1[:3])

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

  50.         self.assertEquals([3, "one", "two"], list1[3:])

  51.         self.assertEquals(["one", "two"], list1[-2:])

  52.         self.assertEquals([0, 1], list1[:-4])

  53.         self.assertEquals([], list1[6:])

  54.         self.assertEquals([], list1[4:2])

  55. 

  56.         self.assertEquals([0, 2, "one"], list1[0:5:2])

  57.         self.assertEquals([0, 2], list1[0:-3:2])

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

  59.         self.assertEquals([2, 3, "one", "two"], list1[2::])

  60.         self.assertEquals([0, 1, 2, 3], list1[:4:])

  61.         self.assertEquals([2, 3], list1[2:4:])

  62.         self.assertEquals([0, 2, 4, 6, 8], list2[::2])

  63.         self.assertEquals([2, 5, 8], list2[2::3])

  64.         self.assertEquals([0, 3], list2[:6:3])

  65.         self.assertEquals([2, 5, 8], list2[-8:9:3])

  66.         self.assertEquals([], list2[100000:1000:-100])

  67. 

  68.         list1[3] = 100

  69.         self.assertEquals(100, list1[3])

  70.         list1[-3] = 101

  71.         self.assertEquals([0, 1, 2, 101, "one", "two"], list1)

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

  73.         self.assertEquals([6, 7, 8], list1[5:])

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

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

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

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

  78.         self.assertEquals([99, 6, 7, 8], list1)

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

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

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

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

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

  84.         self.assertEquals([0, 1, 2, 3, 4, 5, 6], list2)

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

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

  87. 

  88.         del list2[2]

  89.         self.assertEquals([0, 1, 3, 4, 5, 6], list2)

  90.         del list2[-3]

  91.         self.assertEquals([0, 1, 3, 5, 6], list2)

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

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

  94.         list2[:] = range(10)

  95.         del list2[3:6]

  96.         self.assertEquals([0, 1, 2, 6, 7, 8, 9], list2)

  97.         del list2[-2:]

  98.         self.assertEquals([0, 1, 2, 6, 7], list2)

  99.         del list2[:2]

  100.         self.assertEquals([2, 6, 7], list2)

  101.         list2[:] = range(10)

  102.         del list2[2:8:2]

  103.         self.assertEquals([0, 1, 3, 5, 7, 8, 9], list2)

  104. 

  105.     def _test_add_radd_iadd(self, builder):

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

  107.         list1 = builder(range(5))

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

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

  110.         self.assertEquals([0, 1, 2, 3, 4], list1)

  111.         self.assertEquals(list(range(10)), list1 + list2)

  112.         self.assertEquals([-2, -1, 0, 1, 2, 3, 4], [-2, -1] + list1)

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

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

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

  116. 

  117.     def _test_other_magic_methods(self, builder):

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

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

  120.         list2 = builder([])

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

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

  123. 

  124.         if py3k:

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

  126.             self.assertEquals(b"[0, 1, 2, 3, 'one', 'two']", bytes(list1))

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

  128.         else:

  129.             self.assertEquals("[0, 1, 2, 3, u'one', u'two']", unicode(list1))

  130.             self.assertEquals(b"[0, 1, 2, 3, u'one', u'two']", str(list1))

  131.             self.assertEquals(b"[0, 1, 2, 3, u'one', u'two']", repr(list1))

  132. 

  133.         self.assertTrue(list1 < list3)

  134.         self.assertTrue(list1 <= list3)

  135.         self.assertFalse(list1 == list3)

  136.         self.assertTrue(list1 != list3)

  137.         self.assertFalse(list1 > list3)

  138.         self.assertFalse(list1 >= list3)

  139. 

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

  141.         self.assertTrue(list1 < other1)

  142.         self.assertTrue(list1 <= other1)

  143.         self.assertFalse(list1 == other1)

  144.         self.assertTrue(list1 != other1)

  145.         self.assertFalse(list1 > other1)

  146.         self.assertFalse(list1 >= other1)

  147. 

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

  149.         self.assertFalse(list1 < other2)

  150.         self.assertFalse(list1 <= other2)

  151.         self.assertFalse(list1 == other2)

  152.         self.assertTrue(list1 != other2)

  153.         self.assertTrue(list1 > other2)

  154.         self.assertTrue(list1 >= other2)

  155. 

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

  157.         self.assertFalse(list1 < other3)

  158.         self.assertTrue(list1 <= other3)

  159.         self.assertTrue(list1 == other3)

  160.         self.assertFalse(list1 != other3)

  161.         self.assertFalse(list1 > other3)

  162.         self.assertTrue(list1 >= other3)

  163. 

  164.         self.assertTrue(bool(list1))

  165.         self.assertFalse(bool(list2))

  166. 

  167.         self.assertEquals(6, len(list1))

  168.         self.assertEquals(0, len(list2))

  169. 

  170.         out = []

  171.         for obj in list1:

  172.             out.append(obj)

  173.         self.assertEquals([0, 1, 2, 3, "one", "two"], out)

  174. 

  175.         out = []

  176.         for ch in list2:

  177.             out.append(ch)

  178.         self.assertEquals([], out)

  179. 

  180.         gen1 = iter(list1)

  181.         out = []

  182.         for i in range(len(list1)):

  183.             out.append(gen1.next())

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

  185.         self.assertEquals([0, 1, 2, 3, "one", "two"], out)

  186.         gen2 = iter(list2)

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

  188. 

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

  190.         self.assertEquals([], list(reversed(list2)))

  191. 

  192.         self.assertTrue("one" in list1)

  193.         self.assertTrue(3 in list1)

  194.         self.assertFalse(10 in list1)

  195.         self.assertFalse(0 in list2)

  196. 

  197.         self.assertEquals([], list2 * 5)

  198.         self.assertEquals([], 5 * list2)

  199.         self.assertEquals([0, 1, 2, 0, 1, 2, 0, 1, 2], list4 * 3)

  200.         self.assertEquals([0, 1, 2, 0, 1, 2, 0, 1, 2], 3 * list4)

  201.         list4 *= 2

  202.         self.assertEquals([0, 1, 2, 0, 1, 2], list4)

  203. 

  204.     def _test_list_methods(self, builder):

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

  206.         list1 = builder(range(5))

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

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

  209. 

  210.         list1.append(5)

  211.         list1.append(1)

  212.         list1.append(2)

  213.         self.assertEquals([0, 1, 2, 3, 4, 5, 1, 2], list1)

  214. 

  215.         self.assertEquals(0, list1.count(6))

  216.         self.assertEquals(2, list1.count(1))

  217. 

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

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

  220. 

  221.         self.assertEquals(1, list1.index(1))

  222.         self.assertEquals(6, list1.index(1, 3))

  223.         self.assertEquals(6, list1.index(1, 3, 7))

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

  225. 

  226.         list1.insert(0, -1)

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

  228.         list1.insert(-1, 6.5)

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

  230.         list1.insert(13, 8)

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

  232. 

  233.         self.assertEquals(8, list1.pop())

  234.         self.assertEquals(7, list1.pop())

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

  236.         self.assertEquals(-1, list1.pop(0))

  237.         self.assertEquals(5, list1.pop(5))

  238.         self.assertEquals(6.5, list1.pop(-1))

  239.         self.assertEquals([0, 1, 2, 3, 4, 1, 2, 5, 6], list1)

  240.         self.assertEquals("foo", list2.pop())

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

  242.         self.assertEquals([], list2)

  243. 

  244.         list1.remove(6)

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

  246.         list1.remove(1)

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

  248.         list1.remove(1)

  249.         self.assertEquals([0, 2, 3, 4, 2, 5], list1)

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

  251. 

  252.         list1.reverse()

  253.         self.assertEquals([5, 2, 4, 3, 2, 0], list1)

  254. 

  255.         list1.sort()

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

  257.         list1.sort(reverse=True)

  258.         self.assertEquals([5, 4, 3, 2, 2, 0], list1)

  259.         list1.sort(cmp=lambda x, y: abs(3 - x) - abs(3 - y))  # Distance from 3

  260.         self.assertEquals([3, 4, 2, 2, 5, 0], list1)

  261.         list1.sort(cmp=lambda x, y: abs(3 - x) - abs(3 - y), reverse=True)

  262.         self.assertEquals([0, 5, 4, 2, 2, 3], list1)

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

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

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

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

  267. 

  268.     def test_docs(self):

  269.         """make sure the methods of SmartList/_ListProxy have docstrings"""

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

  271.                    "remove", "reverse", "sort"]

  272.         for meth in methods:

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

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

  275.             listproxy_doc = getattr(_ListProxy, meth).__doc__

  276.             self.assertEquals(expected, smartlist_doc)

  277.             self.assertEquals(expected, listproxy_doc)

  278. 

  279.     def test_doctest(self):

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

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

  282.         self.assertEquals([0, 1, 2, 3], parent)

  283.         child = parent[2:]

  284.         self.assertEquals([2, 3], child)

  285.         child.append(4)

  286.         self.assertEquals([2, 3, 4], child)

  287.         self.assertEquals([0, 1, 2, 3, 4], parent)

  288. 

  289.     def test_parent_get_set_del(self):

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

  291.         self._test_get_set_del_item(lambda L: SmartList(L))

  292. 

  293.     def test_parent_add(self):

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

  295.         self._test_add_radd_iadd(lambda L: SmartList(L))

  296. 

  297.     def test_parent_unaffected_magics(self):

  298.         """sanity checks against SmartList features that were not modified"""

  299.         self._test_other_magic_methods(lambda L: SmartList(L))

  300. 

  301.     def test_parent_methods(self):

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

  303.         self._test_list_methods(lambda L: SmartList(L))

  304. 

  305.     def test_child_get_set_del(self):

  306.         """make sure _ListProxy's getitem/setitem/delitem work"""

  307.         self._test_get_set_del_item(lambda L: SmartList(list(L))[:])

  308.         self._test_get_set_del_item(lambda L: SmartList([999] + list(L))[1:])

  309.         self._test_get_set_del_item(lambda L: SmartList(list(L) + [999])[:-1])

  310.         builder = lambda L: SmartList([101, 102] + list(L) + [201, 202])[2:-2]

  311.         self._test_get_set_del_item(builder)

  312. 

  313.     def test_child_add(self):

  314.         """make sure _ListProxy's add/radd/iadd work"""

  315.         self._test_add_radd_iadd(lambda L: SmartList(list(L))[:])

  316.         self._test_add_radd_iadd(lambda L: SmartList([999] + list(L))[1:])

  317.         self._test_add_radd_iadd(lambda L: SmartList(list(L) + [999])[:-1])

  318.         builder = lambda L: SmartList([101, 102] + list(L) + [201, 202])[2:-2]

  319.         self._test_add_radd_iadd(builder)

  320. 

  321.     def test_child_other_magics(self):

  322.         """make sure _ListProxy's other magically implemented features work"""

  323.         self._test_other_magic_methods(lambda L: SmartList(list(L))[:])

  324.         self._test_other_magic_methods(lambda L: SmartList([999] + list(L))[1:])

  325.         self._test_other_magic_methods(lambda L: SmartList(list(L) + [999])[:-1])

  326.         builder = lambda L: SmartList([101, 102] + list(L) + [201, 202])[2:-2]

  327.         self._test_other_magic_methods(builder)

  328. 

  329.     def test_child_methods(self):

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

  331.         self._test_list_methods(lambda L: SmartList(list(L))[:])

  332.         self._test_list_methods(lambda L: SmartList([999] + list(L))[1:])

  333.         self._test_list_methods(lambda L: SmartList(list(L) + [999])[:-1])

  334.         builder = lambda L: SmartList([101, 102] + list(L) + [201, 202])[2:-2]

  335.         self._test_list_methods(builder)

  336. 

  337.     def test_influence(self):

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

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

  340.         child1 = parent[2:]

  341.         child2 = parent[2:5]

  342. 

  343.         parent.append(6)

  344.         child1.append(7)

  345.         child2.append(4.5)

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

  347.         self.assertEquals([2, 3, 4, 4.5, 5, 6, 7], child1)

  348.         self.assertEquals([2, 3, 4, 4.5], child2)

  349. 

  350.         parent.insert(0, -1)

  351.         parent.insert(4, 2.5)

  352.         parent.insert(10, 6.5)

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

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

  355.         self.assertEquals([2, 2.5, 3, 4, 4.5], child2)

  356. 

  357.         self.assertEquals(7, parent.pop())

  358.         self.assertEquals(6.5, child1.pop())

  359.         self.assertEquals(4.5, child2.pop())

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

  361.         self.assertEquals([2, 2.5, 3, 4, 5, 6], child1)

  362.         self.assertEquals([2, 2.5, 3, 4], child2)

  363. 

  364.         parent.remove(-1)

  365.         child1.remove(2.5)

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

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

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

  369. 

  370.         self.assertEquals(0, parent.pop(0))

  371.         self.assertEquals([1, 2, 3, 4, 5, 6], parent)

  372.         self.assertEquals([2, 3, 4, 5, 6], child1)

  373.         self.assertEquals([2, 3, 4], child2)

  374. 

  375.         child2.reverse()

  376.         self.assertEquals([1, 4, 3, 2, 5, 6], parent)

  377.         self.assertEquals([4, 3, 2, 5, 6], child1)

  378.         self.assertEquals([4, 3, 2], child2)

  379. 

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

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

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

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

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

  385.         self.assertEquals([4, 3, 2, 1.9, 1.8], child2)

  386. 

  387. if __name__ == "__main__":

  388.     unittest.main(verbosity=2)