mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 05:31:20 +00:00 
			
		
		
		
	 8d26b0be6d
			
		
	
	
		8d26b0be6d
		
	
	
	
	
		
			
			svn+ssh://pythondev@svn.python.org/python/trunk
................
  r80934 | benjamin.peterson | 2010-05-07 13:58:23 -0500 (Fri, 07 May 2010) | 69 lines
  Merged revisions 79911,79916-79917,80018,80418,80572-80573,80635-80639,80668,80922 via svnmerge from
  svn+ssh://pythondev@svn.python.org/sandbox/trunk/2to3/lib2to3
  ........
    r79911 | benjamin.peterson | 2010-04-09 15:38:53 -0500 (Fri, 09 Apr 2010) | 1 line
    use absolute import
  ........
    r79916 | benjamin.peterson | 2010-04-09 16:05:21 -0500 (Fri, 09 Apr 2010) | 1 line
    generalize detection of __future__ imports and attach them to the tree
  ........
    r79917 | benjamin.peterson | 2010-04-09 16:11:44 -0500 (Fri, 09 Apr 2010) | 1 line
    don't try to 'fix' relative imports when absolute_import is enabled #8858
  ........
    r80018 | benjamin.peterson | 2010-04-12 16:12:12 -0500 (Mon, 12 Apr 2010) | 4 lines
    prevent diffs from being mangled is multiprocess mode #6409
    Patch by George Boutsioukis.
  ........
    r80418 | benjamin.peterson | 2010-04-23 16:00:03 -0500 (Fri, 23 Apr 2010) | 1 line
    remove unhelpful description
  ........
    r80572 | benjamin.peterson | 2010-04-27 20:33:54 -0500 (Tue, 27 Apr 2010) | 1 line
    use unicode literals
  ........
    r80573 | jeffrey.yasskin | 2010-04-27 23:08:27 -0500 (Tue, 27 Apr 2010) | 6 lines
    Don't transform imports that are already relative.  2to3 turned
      from . import refactor
    into
      from .. import refactor
    which broke the transformation of 2to3 itself.
  ........
    r80635 | benjamin.peterson | 2010-04-29 16:02:23 -0500 (Thu, 29 Apr 2010) | 1 line
    remove imports
  ........
    r80636 | benjamin.peterson | 2010-04-29 16:02:41 -0500 (Thu, 29 Apr 2010) | 1 line
    unicode literal
  ........
    r80637 | benjamin.peterson | 2010-04-29 16:03:42 -0500 (Thu, 29 Apr 2010) | 1 line
    must pass a string to Number
  ........
    r80638 | benjamin.peterson | 2010-04-29 16:05:34 -0500 (Thu, 29 Apr 2010) | 1 line
    unicode literals
  ........
    r80639 | benjamin.peterson | 2010-04-29 16:06:09 -0500 (Thu, 29 Apr 2010) | 1 line
    pass string to Number
  ........
    r80668 | jeffrey.yasskin | 2010-04-30 18:02:47 -0500 (Fri, 30 Apr 2010) | 4 lines
    Make 2to3 run under Python 2.5 so that the benchmark suite at
    http://hg.python.org/benchmarks/ can use it and still run on implementations
    that haven't gotten to 2.6 yet.  Fixes issue 8566.
  ........
    r80922 | benjamin.peterson | 2010-05-07 11:06:25 -0500 (Fri, 07 May 2010) | 1 line
    prevent xrange transformation from wrapping range calls it produces in list
  ........
................
		
	
			
		
			
				
	
	
		
			469 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			469 lines
		
	
	
	
		
			16 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| # Copyright 2006 Google, Inc. All Rights Reserved.
 | |
| # Licensed to PSF under a Contributor Agreement.
 | |
| 
 | |
| """Unit tests for pytree.py.
 | |
| 
 | |
| NOTE: Please *don't* add doc strings to individual test methods!
 | |
| In verbose mode, printing of the module, class and method name is much
 | |
| more helpful than printing of (the first line of) the docstring,
 | |
| especially when debugging a test.
 | |
| """
 | |
| 
 | |
| from __future__ import with_statement
 | |
| 
 | |
| import sys
 | |
| import warnings
 | |
| 
 | |
| # Testing imports
 | |
| from . import support
 | |
| 
 | |
| from lib2to3 import pytree
 | |
| 
 | |
| try:
 | |
|     sorted
 | |
| except NameError:
 | |
|     def sorted(lst):
 | |
|         l = list(lst)
 | |
|         l.sort()
 | |
|         return l
 | |
| 
 | |
| class TestNodes(support.TestCase):
 | |
| 
 | |
|     """Unit tests for nodes (Base, Leaf, Node)."""
 | |
| 
 | |
|     if sys.version_info >= (2,6):
 | |
|         # warnings.catch_warnings is new in 2.6.
 | |
|         def test_deprecated_prefix_methods(self):
 | |
|             l = pytree.Leaf(100, "foo")
 | |
|             with warnings.catch_warnings(record=True) as w:
 | |
|                 warnings.simplefilter("always", DeprecationWarning)
 | |
|                 self.assertEqual(l.get_prefix(), "")
 | |
|                 l.set_prefix("hi")
 | |
|             self.assertEqual(l.prefix, "hi")
 | |
|             self.assertEqual(len(w), 2)
 | |
|             for warning in w:
 | |
|                 self.assertTrue(warning.category is DeprecationWarning)
 | |
|             self.assertEqual(str(w[0].message), "get_prefix() is deprecated; " \
 | |
|                                  "use the prefix property")
 | |
|             self.assertEqual(str(w[1].message), "set_prefix() is deprecated; " \
 | |
|                                  "use the prefix property")
 | |
| 
 | |
|     def test_instantiate_base(self):
 | |
|         if __debug__:
 | |
|             # Test that instantiating Base() raises an AssertionError
 | |
|             self.assertRaises(AssertionError, pytree.Base)
 | |
| 
 | |
|     def test_leaf(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         self.assertEqual(l1.type, 100)
 | |
|         self.assertEqual(l1.value, "foo")
 | |
| 
 | |
|     def test_leaf_repr(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         self.assertEqual(repr(l1), "Leaf(100, 'foo')")
 | |
| 
 | |
|     def test_leaf_str(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         self.assertEqual(str(l1), "foo")
 | |
|         l2 = pytree.Leaf(100, "foo", context=(" ", (10, 1)))
 | |
|         self.assertEqual(str(l2), " foo")
 | |
| 
 | |
|     def test_leaf_str_numeric_value(self):
 | |
|         # Make sure that the Leaf's value is stringified. Failing to
 | |
|         #  do this can cause a TypeError in certain situations.
 | |
|         l1 = pytree.Leaf(2, 5)
 | |
|         l1.prefix = "foo_"
 | |
|         self.assertEqual(str(l1), "foo_5")
 | |
| 
 | |
|     def test_leaf_equality(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "foo", context=(" ", (1, 0)))
 | |
|         self.assertEqual(l1, l2)
 | |
|         l3 = pytree.Leaf(101, "foo")
 | |
|         l4 = pytree.Leaf(100, "bar")
 | |
|         self.assertNotEqual(l1, l3)
 | |
|         self.assertNotEqual(l1, l4)
 | |
| 
 | |
|     def test_leaf_prefix(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         self.assertEqual(l1.prefix, "")
 | |
|         self.assertFalse(l1.was_changed)
 | |
|         l1.prefix = "  ##\n\n"
 | |
|         self.assertEqual(l1.prefix, "  ##\n\n")
 | |
|         self.assertTrue(l1.was_changed)
 | |
| 
 | |
|     def test_node(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(200, "bar")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         self.assertEqual(n1.type, 1000)
 | |
|         self.assertEqual(n1.children, [l1, l2])
 | |
| 
 | |
|     def test_node_repr(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "bar", context=(" ", (1, 0)))
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         self.assertEqual(repr(n1),
 | |
|                          "Node(1000, [%s, %s])" % (repr(l1), repr(l2)))
 | |
| 
 | |
|     def test_node_str(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "bar", context=(" ", (1, 0)))
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         self.assertEqual(str(n1), "foo bar")
 | |
| 
 | |
|     def test_node_prefix(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         self.assertEqual(l1.prefix, "")
 | |
|         n1 = pytree.Node(1000, [l1])
 | |
|         self.assertEqual(n1.prefix, "")
 | |
|         n1.prefix = " "
 | |
|         self.assertEqual(n1.prefix, " ")
 | |
|         self.assertEqual(l1.prefix, " ")
 | |
| 
 | |
|     def test_get_suffix(self):
 | |
|         l1 = pytree.Leaf(100, "foo", prefix="a")
 | |
|         l2 = pytree.Leaf(100, "bar", prefix="b")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
| 
 | |
|         self.assertEqual(l1.get_suffix(), l2.prefix)
 | |
|         self.assertEqual(l2.get_suffix(), "")
 | |
|         self.assertEqual(n1.get_suffix(), "")
 | |
| 
 | |
|         l3 = pytree.Leaf(100, "bar", prefix="c")
 | |
|         n2 = pytree.Node(1000, [n1, l3])
 | |
| 
 | |
|         self.assertEqual(n1.get_suffix(), l3.prefix)
 | |
|         self.assertEqual(l3.get_suffix(), "")
 | |
|         self.assertEqual(n2.get_suffix(), "")
 | |
| 
 | |
|     def test_node_equality(self):
 | |
|         n1 = pytree.Node(1000, ())
 | |
|         n2 = pytree.Node(1000, [], context=(" ", (1, 0)))
 | |
|         self.assertEqual(n1, n2)
 | |
|         n3 = pytree.Node(1001, ())
 | |
|         self.assertNotEqual(n1, n3)
 | |
| 
 | |
|     def test_node_recursive_equality(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "foo")
 | |
|         n1 = pytree.Node(1000, [l1])
 | |
|         n2 = pytree.Node(1000, [l2])
 | |
|         self.assertEqual(n1, n2)
 | |
|         l3 = pytree.Leaf(100, "bar")
 | |
|         n3 = pytree.Node(1000, [l3])
 | |
|         self.assertNotEqual(n1, n3)
 | |
| 
 | |
|     def test_replace(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "+")
 | |
|         l3 = pytree.Leaf(100, "bar")
 | |
|         n1 = pytree.Node(1000, [l1, l2, l3])
 | |
|         self.assertEqual(n1.children, [l1, l2, l3])
 | |
|         self.assertTrue(isinstance(n1.children, list))
 | |
|         self.assertFalse(n1.was_changed)
 | |
|         l2new = pytree.Leaf(100, "-")
 | |
|         l2.replace(l2new)
 | |
|         self.assertEqual(n1.children, [l1, l2new, l3])
 | |
|         self.assertTrue(isinstance(n1.children, list))
 | |
|         self.assertTrue(n1.was_changed)
 | |
| 
 | |
|     def test_replace_with_list(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "+")
 | |
|         l3 = pytree.Leaf(100, "bar")
 | |
|         n1 = pytree.Node(1000, [l1, l2, l3])
 | |
| 
 | |
|         l2.replace([pytree.Leaf(100, "*"), pytree.Leaf(100, "*")])
 | |
|         self.assertEqual(str(n1), "foo**bar")
 | |
|         self.assertTrue(isinstance(n1.children, list))
 | |
| 
 | |
|     def test_post_order(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         self.assertEqual(list(n1.post_order()), [l1, l2, n1])
 | |
| 
 | |
|     def test_pre_order(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         self.assertEqual(list(n1.pre_order()), [n1, l1, l2])
 | |
| 
 | |
|     def test_changed(self):
 | |
|         l1 = pytree.Leaf(100, "f")
 | |
|         self.assertFalse(l1.was_changed)
 | |
|         l1.changed()
 | |
|         self.assertTrue(l1.was_changed)
 | |
| 
 | |
|         l1 = pytree.Leaf(100, "f")
 | |
|         n1 = pytree.Node(1000, [l1])
 | |
|         self.assertFalse(n1.was_changed)
 | |
|         n1.changed()
 | |
|         self.assertTrue(n1.was_changed)
 | |
| 
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "+")
 | |
|         l3 = pytree.Leaf(100, "bar")
 | |
|         n1 = pytree.Node(1000, [l1, l2, l3])
 | |
|         n2 = pytree.Node(1000, [n1])
 | |
|         self.assertFalse(l1.was_changed)
 | |
|         self.assertFalse(n1.was_changed)
 | |
|         self.assertFalse(n2.was_changed)
 | |
| 
 | |
|         n1.changed()
 | |
|         self.assertTrue(n1.was_changed)
 | |
|         self.assertTrue(n2.was_changed)
 | |
|         self.assertFalse(l1.was_changed)
 | |
| 
 | |
|     def test_leaf_constructor_prefix(self):
 | |
|         for prefix in ("xyz_", ""):
 | |
|             l1 = pytree.Leaf(100, "self", prefix=prefix)
 | |
|             self.assertTrue(str(l1), prefix + "self")
 | |
|             self.assertEqual(l1.prefix, prefix)
 | |
| 
 | |
|     def test_node_constructor_prefix(self):
 | |
|         for prefix in ("xyz_", ""):
 | |
|             l1 = pytree.Leaf(100, "self")
 | |
|             l2 = pytree.Leaf(100, "foo", prefix="_")
 | |
|             n1 = pytree.Node(1000, [l1, l2], prefix=prefix)
 | |
|             self.assertTrue(str(n1), prefix + "self_foo")
 | |
|             self.assertEqual(n1.prefix, prefix)
 | |
|             self.assertEqual(l1.prefix, prefix)
 | |
|             self.assertEqual(l2.prefix, "_")
 | |
| 
 | |
|     def test_remove(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "foo")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         n2 = pytree.Node(1000, [n1])
 | |
| 
 | |
|         self.assertEqual(n1.remove(), 0)
 | |
|         self.assertEqual(n2.children, [])
 | |
|         self.assertEqual(l1.parent, n1)
 | |
|         self.assertEqual(n1.parent, None)
 | |
|         self.assertEqual(n2.parent, None)
 | |
|         self.assertFalse(n1.was_changed)
 | |
|         self.assertTrue(n2.was_changed)
 | |
| 
 | |
|         self.assertEqual(l2.remove(), 1)
 | |
|         self.assertEqual(l1.remove(), 0)
 | |
|         self.assertEqual(n1.children, [])
 | |
|         self.assertEqual(l1.parent, None)
 | |
|         self.assertEqual(n1.parent, None)
 | |
|         self.assertEqual(n2.parent, None)
 | |
|         self.assertTrue(n1.was_changed)
 | |
|         self.assertTrue(n2.was_changed)
 | |
| 
 | |
|     def test_remove_parentless(self):
 | |
|         n1 = pytree.Node(1000, [])
 | |
|         n1.remove()
 | |
|         self.assertEqual(n1.parent, None)
 | |
| 
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l1.remove()
 | |
|         self.assertEqual(l1.parent, None)
 | |
| 
 | |
|     def test_node_set_child(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         n1 = pytree.Node(1000, [l1])
 | |
| 
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         n1.set_child(0, l2)
 | |
|         self.assertEqual(l1.parent, None)
 | |
|         self.assertEqual(l2.parent, n1)
 | |
|         self.assertEqual(n1.children, [l2])
 | |
| 
 | |
|         n2 = pytree.Node(1000, [l1])
 | |
|         n2.set_child(0, n1)
 | |
|         self.assertEqual(l1.parent, None)
 | |
|         self.assertEqual(n1.parent, n2)
 | |
|         self.assertEqual(n2.parent, None)
 | |
|         self.assertEqual(n2.children, [n1])
 | |
| 
 | |
|         self.assertRaises(IndexError, n1.set_child, 4, l2)
 | |
|         # I don't care what it raises, so long as it's an exception
 | |
|         self.assertRaises(Exception, n1.set_child, 0, list)
 | |
| 
 | |
|     def test_node_insert_child(self):
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         n1 = pytree.Node(1000, [l1])
 | |
| 
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         n1.insert_child(0, l2)
 | |
|         self.assertEqual(l2.parent, n1)
 | |
|         self.assertEqual(n1.children, [l2, l1])
 | |
| 
 | |
|         l3 = pytree.Leaf(100, "abc")
 | |
|         n1.insert_child(2, l3)
 | |
|         self.assertEqual(n1.children, [l2, l1, l3])
 | |
| 
 | |
|         # I don't care what it raises, so long as it's an exception
 | |
|         self.assertRaises(Exception, n1.insert_child, 0, list)
 | |
| 
 | |
|     def test_node_append_child(self):
 | |
|         n1 = pytree.Node(1000, [])
 | |
| 
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         n1.append_child(l1)
 | |
|         self.assertEqual(l1.parent, n1)
 | |
|         self.assertEqual(n1.children, [l1])
 | |
| 
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         n1.append_child(l2)
 | |
|         self.assertEqual(l2.parent, n1)
 | |
|         self.assertEqual(n1.children, [l1, l2])
 | |
| 
 | |
|         # I don't care what it raises, so long as it's an exception
 | |
|         self.assertRaises(Exception, n1.append_child, list)
 | |
| 
 | |
|     def test_node_next_sibling(self):
 | |
|         n1 = pytree.Node(1000, [])
 | |
|         n2 = pytree.Node(1000, [])
 | |
|         p1 = pytree.Node(1000, [n1, n2])
 | |
| 
 | |
|         self.assertTrue(n1.next_sibling is n2)
 | |
|         self.assertEqual(n2.next_sibling, None)
 | |
|         self.assertEqual(p1.next_sibling, None)
 | |
| 
 | |
|     def test_leaf_next_sibling(self):
 | |
|         l1 = pytree.Leaf(100, "a")
 | |
|         l2 = pytree.Leaf(100, "b")
 | |
|         p1 = pytree.Node(1000, [l1, l2])
 | |
| 
 | |
|         self.assertTrue(l1.next_sibling is l2)
 | |
|         self.assertEqual(l2.next_sibling, None)
 | |
|         self.assertEqual(p1.next_sibling, None)
 | |
| 
 | |
|     def test_node_prev_sibling(self):
 | |
|         n1 = pytree.Node(1000, [])
 | |
|         n2 = pytree.Node(1000, [])
 | |
|         p1 = pytree.Node(1000, [n1, n2])
 | |
| 
 | |
|         self.assertTrue(n2.prev_sibling is n1)
 | |
|         self.assertEqual(n1.prev_sibling, None)
 | |
|         self.assertEqual(p1.prev_sibling, None)
 | |
| 
 | |
|     def test_leaf_prev_sibling(self):
 | |
|         l1 = pytree.Leaf(100, "a")
 | |
|         l2 = pytree.Leaf(100, "b")
 | |
|         p1 = pytree.Node(1000, [l1, l2])
 | |
| 
 | |
|         self.assertTrue(l2.prev_sibling is l1)
 | |
|         self.assertEqual(l1.prev_sibling, None)
 | |
|         self.assertEqual(p1.prev_sibling, None)
 | |
| 
 | |
| 
 | |
| class TestPatterns(support.TestCase):
 | |
| 
 | |
|     """Unit tests for tree matching patterns."""
 | |
| 
 | |
|     def test_basic_patterns(self):
 | |
|         # Build a tree
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         l3 = pytree.Leaf(100, "foo")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         n2 = pytree.Node(1000, [l3])
 | |
|         root = pytree.Node(1000, [n1, n2])
 | |
|         # Build a pattern matching a leaf
 | |
|         pl = pytree.LeafPattern(100, "foo", name="pl")
 | |
|         r = {}
 | |
|         self.assertFalse(pl.match(root, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertFalse(pl.match(n1, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertFalse(pl.match(n2, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertTrue(pl.match(l1, results=r))
 | |
|         self.assertEqual(r, {"pl": l1})
 | |
|         r = {}
 | |
|         self.assertFalse(pl.match(l2, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         # Build a pattern matching a node
 | |
|         pn = pytree.NodePattern(1000, [pl], name="pn")
 | |
|         self.assertFalse(pn.match(root, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertFalse(pn.match(n1, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertTrue(pn.match(n2, results=r))
 | |
|         self.assertEqual(r, {"pn": n2, "pl": l3})
 | |
|         r = {}
 | |
|         self.assertFalse(pn.match(l1, results=r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertFalse(pn.match(l2, results=r))
 | |
|         self.assertEqual(r, {})
 | |
| 
 | |
|     def test_wildcard(self):
 | |
|         # Build a tree for testing
 | |
|         l1 = pytree.Leaf(100, "foo")
 | |
|         l2 = pytree.Leaf(100, "bar")
 | |
|         l3 = pytree.Leaf(100, "foo")
 | |
|         n1 = pytree.Node(1000, [l1, l2])
 | |
|         n2 = pytree.Node(1000, [l3])
 | |
|         root = pytree.Node(1000, [n1, n2])
 | |
|         # Build a pattern
 | |
|         pl = pytree.LeafPattern(100, "foo", name="pl")
 | |
|         pn = pytree.NodePattern(1000, [pl], name="pn")
 | |
|         pw = pytree.WildcardPattern([[pn], [pl, pl]], name="pw")
 | |
|         r = {}
 | |
|         self.assertFalse(pw.match_seq([root], r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertFalse(pw.match_seq([n1], r))
 | |
|         self.assertEqual(r, {})
 | |
|         self.assertTrue(pw.match_seq([n2], r))
 | |
|         # These are easier to debug
 | |
|         self.assertEqual(sorted(r.keys()), ["pl", "pn", "pw"])
 | |
|         self.assertEqual(r["pl"], l1)
 | |
|         self.assertEqual(r["pn"], n2)
 | |
|         self.assertEqual(r["pw"], [n2])
 | |
|         # But this is equivalent
 | |
|         self.assertEqual(r, {"pl": l1, "pn": n2, "pw": [n2]})
 | |
|         r = {}
 | |
|         self.assertTrue(pw.match_seq([l1, l3], r))
 | |
|         self.assertEqual(r, {"pl": l3, "pw": [l1, l3]})
 | |
|         self.assertTrue(r["pl"] is l3)
 | |
|         r = {}
 | |
| 
 | |
|     def test_generate_matches(self):
 | |
|         la = pytree.Leaf(1, "a")
 | |
|         lb = pytree.Leaf(1, "b")
 | |
|         lc = pytree.Leaf(1, "c")
 | |
|         ld = pytree.Leaf(1, "d")
 | |
|         le = pytree.Leaf(1, "e")
 | |
|         lf = pytree.Leaf(1, "f")
 | |
|         leaves = [la, lb, lc, ld, le, lf]
 | |
|         root = pytree.Node(1000, leaves)
 | |
|         pa = pytree.LeafPattern(1, "a", "pa")
 | |
|         pb = pytree.LeafPattern(1, "b", "pb")
 | |
|         pc = pytree.LeafPattern(1, "c", "pc")
 | |
|         pd = pytree.LeafPattern(1, "d", "pd")
 | |
|         pe = pytree.LeafPattern(1, "e", "pe")
 | |
|         pf = pytree.LeafPattern(1, "f", "pf")
 | |
|         pw = pytree.WildcardPattern([[pa, pb, pc], [pd, pe],
 | |
|                                      [pa, pb], [pc, pd], [pe, pf]],
 | |
|                                     min=1, max=4, name="pw")
 | |
|         self.assertEqual([x[0] for x in pw.generate_matches(leaves)],
 | |
|                          [3, 5, 2, 4, 6])
 | |
|         pr = pytree.NodePattern(type=1000, content=[pw], name="pr")
 | |
|         matches = list(pytree.generate_matches([pr], [root]))
 | |
|         self.assertEqual(len(matches), 1)
 | |
|         c, r = matches[0]
 | |
|         self.assertEqual(c, 1)
 | |
|         self.assertEqual(str(r["pr"]), "abcdef")
 | |
|         self.assertEqual(r["pw"], [la, lb, lc, ld, le, lf])
 | |
|         for c in "abcdef":
 | |
|             self.assertEqual(r["p" + c], pytree.Leaf(1, c))
 | |
| 
 | |
|     def test_has_key_example(self):
 | |
|         pattern = pytree.NodePattern(331,
 | |
|                                      (pytree.LeafPattern(7),
 | |
|                                       pytree.WildcardPattern(name="args"),
 | |
|                                       pytree.LeafPattern(8)))
 | |
|         l1 = pytree.Leaf(7, "(")
 | |
|         l2 = pytree.Leaf(3, "x")
 | |
|         l3 = pytree.Leaf(8, ")")
 | |
|         node = pytree.Node(331, [l1, l2, l3])
 | |
|         r = {}
 | |
|         self.assertTrue(pattern.match(node, r))
 | |
|         self.assertEqual(r["args"], [l2])
 |