From 411a6354ff64db4c3754e2bd2d7231097dbb88a3 Mon Sep 17 00:00:00 2001 From: Stefan Behnel Date: Fri, 18 Sep 2009 08:08:52 +0200 Subject: [PATCH] simpler test --- Cython/Compiler/Tests/TestTreePath.py | 48 +++++++++------------------ 1 file changed, 16 insertions(+), 32 deletions(-) diff --git a/Cython/Compiler/Tests/TestTreePath.py b/Cython/Compiler/Tests/TestTreePath.py index 7014a9ad..64815409 100644 --- a/Cython/Compiler/Tests/TestTreePath.py +++ b/Cython/Compiler/Tests/TestTreePath.py @@ -4,53 +4,37 @@ from Cython.TestUtils import TransformTest from Cython.Compiler.TreePath import find_first, find_all class TestTreePath(TransformTest): + _tree = None + + def _build_tree(self): + if self._tree is None: + self._tree = self.run_pipeline([], u""" + def decorator(fun): # DefNode + return fun # ReturnStatNode, NameNode + @decorator # NameNode + def decorated(): # DefNode + pass + """) + return self._tree def test_node_path(self): - t = self.run_pipeline([], u""" - def decorator(fun): # DefNode - return fun # ReturnStatNode, NameNode - @decorator # NameNode - def decorated(): # DefNode - pass - """) - + t = self._build_tree() self.assertEquals(2, len(find_all(t, "//DefNode"))) self.assertEquals(2, len(find_all(t, "//NameNode"))) self.assertEquals(1, len(find_all(t, "//ReturnStatNode"))) self.assertEquals(1, len(find_all(t, "//DefNode//ReturnStatNode"))) def test_node_path_child(self): - t = self.run_pipeline([], u""" - def decorator(fun): # DefNode - return fun # ReturnStatNode, NameNode - @decorator # NameNode - def decorated(): # DefNode - pass - """) - + t = self._build_tree() self.assertEquals(1, len(find_all(t, "//DefNode/ReturnStatNode/NameNode"))) self.assertEquals(1, len(find_all(t, "//ReturnStatNode/NameNode"))) def test_node_path_attribute_exists(self): - t = self.run_pipeline([], u""" - def decorator(fun): - return fun - @decorator - def decorated(): - pass - """) - + t = self._build_tree() self.assertEquals(2, len(find_all(t, "//NameNode[@name]"))) def test_node_path_attribute_string_predicate(self): - t = self.run_pipeline([], u""" - def decorator(fun): - return fun - @decorator - def decorated(): - pass - """) - + t = self._build_tree() self.assertEquals(1, len(find_all(t, "//NameNode[@name = 'decorator']"))) if __name__ == '__main__': -- 2.26.2