aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py')
-rw-r--r--contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py198
1 files changed, 99 insertions, 99 deletions
diff --git a/contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py b/contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py
index 6f6731345e..07bf31f3e0 100644
--- a/contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py
+++ b/contrib/tools/cython/Cython/Compiler/AnalysedTreeTransforms.py
@@ -1,99 +1,99 @@
-from __future__ import absolute_import
-
-from .Visitor import ScopeTrackingTransform
-from .Nodes import StatListNode, SingleAssignmentNode, CFuncDefNode, DefNode
-from .ExprNodes import DictNode, DictItemNode, NameNode, UnicodeNode
-from .PyrexTypes import py_object_type
-from .StringEncoding import EncodedString
-from . import Symtab
-
-class AutoTestDictTransform(ScopeTrackingTransform):
- # Handles autotestdict directive
-
- blacklist = ['__cinit__', '__dealloc__', '__richcmp__',
- '__nonzero__', '__bool__',
- '__len__', '__contains__']
-
- def visit_ModuleNode(self, node):
- if node.is_pxd:
- return node
- self.scope_type = 'module'
- self.scope_node = node
-
- if not self.current_directives['autotestdict']:
- return node
- self.all_docstrings = self.current_directives['autotestdict.all']
- self.cdef_docstrings = self.all_docstrings or self.current_directives['autotestdict.cdef']
-
- assert isinstance(node.body, StatListNode)
-
- # First see if __test__ is already created
- if u'__test__' in node.scope.entries:
- # Do nothing
- return node
-
- pos = node.pos
-
- self.tests = []
- self.testspos = node.pos
-
- test_dict_entry = node.scope.declare_var(EncodedString(u'__test__'),
- py_object_type,
- pos,
- visibility='public')
- create_test_dict_assignment = SingleAssignmentNode(pos,
- lhs=NameNode(pos, name=EncodedString(u'__test__'),
- entry=test_dict_entry),
- rhs=DictNode(pos, key_value_pairs=self.tests))
- self.visitchildren(node)
- node.body.stats.append(create_test_dict_assignment)
- return node
-
- def add_test(self, testpos, path, doctest):
- pos = self.testspos
- keystr = u'%s (line %d)' % (path, testpos[1])
- key = UnicodeNode(pos, value=EncodedString(keystr))
- value = UnicodeNode(pos, value=doctest)
- self.tests.append(DictItemNode(pos, key=key, value=value))
-
- def visit_ExprNode(self, node):
- # expressions cannot contain functions and lambda expressions
- # do not have a docstring
- return node
-
- def visit_FuncDefNode(self, node):
- if not node.doc or (isinstance(node, DefNode) and node.fused_py_func):
- return node
- if not self.cdef_docstrings:
- if isinstance(node, CFuncDefNode) and not node.py_func:
- return node
- if not self.all_docstrings and '>>>' not in node.doc:
- return node
-
- pos = self.testspos
- if self.scope_type == 'module':
- path = node.entry.name
- elif self.scope_type in ('pyclass', 'cclass'):
- if isinstance(node, CFuncDefNode):
- if node.py_func is not None:
- name = node.py_func.name
- else:
- name = node.entry.name
- else:
- name = node.name
- if self.scope_type == 'cclass' and name in self.blacklist:
- return node
- if self.scope_type == 'pyclass':
- class_name = self.scope_node.name
- else:
- class_name = self.scope_node.class_name
- if isinstance(node.entry.scope, Symtab.PropertyScope):
- property_method_name = node.entry.scope.name
- path = "%s.%s.%s" % (class_name, node.entry.scope.name,
- node.entry.name)
- else:
- path = "%s.%s" % (class_name, node.entry.name)
- else:
- assert False
- self.add_test(node.pos, path, node.doc)
- return node
+from __future__ import absolute_import
+
+from .Visitor import ScopeTrackingTransform
+from .Nodes import StatListNode, SingleAssignmentNode, CFuncDefNode, DefNode
+from .ExprNodes import DictNode, DictItemNode, NameNode, UnicodeNode
+from .PyrexTypes import py_object_type
+from .StringEncoding import EncodedString
+from . import Symtab
+
+class AutoTestDictTransform(ScopeTrackingTransform):
+ # Handles autotestdict directive
+
+ blacklist = ['__cinit__', '__dealloc__', '__richcmp__',
+ '__nonzero__', '__bool__',
+ '__len__', '__contains__']
+
+ def visit_ModuleNode(self, node):
+ if node.is_pxd:
+ return node
+ self.scope_type = 'module'
+ self.scope_node = node
+
+ if not self.current_directives['autotestdict']:
+ return node
+ self.all_docstrings = self.current_directives['autotestdict.all']
+ self.cdef_docstrings = self.all_docstrings or self.current_directives['autotestdict.cdef']
+
+ assert isinstance(node.body, StatListNode)
+
+ # First see if __test__ is already created
+ if u'__test__' in node.scope.entries:
+ # Do nothing
+ return node
+
+ pos = node.pos
+
+ self.tests = []
+ self.testspos = node.pos
+
+ test_dict_entry = node.scope.declare_var(EncodedString(u'__test__'),
+ py_object_type,
+ pos,
+ visibility='public')
+ create_test_dict_assignment = SingleAssignmentNode(pos,
+ lhs=NameNode(pos, name=EncodedString(u'__test__'),
+ entry=test_dict_entry),
+ rhs=DictNode(pos, key_value_pairs=self.tests))
+ self.visitchildren(node)
+ node.body.stats.append(create_test_dict_assignment)
+ return node
+
+ def add_test(self, testpos, path, doctest):
+ pos = self.testspos
+ keystr = u'%s (line %d)' % (path, testpos[1])
+ key = UnicodeNode(pos, value=EncodedString(keystr))
+ value = UnicodeNode(pos, value=doctest)
+ self.tests.append(DictItemNode(pos, key=key, value=value))
+
+ def visit_ExprNode(self, node):
+ # expressions cannot contain functions and lambda expressions
+ # do not have a docstring
+ return node
+
+ def visit_FuncDefNode(self, node):
+ if not node.doc or (isinstance(node, DefNode) and node.fused_py_func):
+ return node
+ if not self.cdef_docstrings:
+ if isinstance(node, CFuncDefNode) and not node.py_func:
+ return node
+ if not self.all_docstrings and '>>>' not in node.doc:
+ return node
+
+ pos = self.testspos
+ if self.scope_type == 'module':
+ path = node.entry.name
+ elif self.scope_type in ('pyclass', 'cclass'):
+ if isinstance(node, CFuncDefNode):
+ if node.py_func is not None:
+ name = node.py_func.name
+ else:
+ name = node.entry.name
+ else:
+ name = node.name
+ if self.scope_type == 'cclass' and name in self.blacklist:
+ return node
+ if self.scope_type == 'pyclass':
+ class_name = self.scope_node.name
+ else:
+ class_name = self.scope_node.class_name
+ if isinstance(node.entry.scope, Symtab.PropertyScope):
+ property_method_name = node.entry.scope.name
+ path = "%s.%s.%s" % (class_name, node.entry.scope.name,
+ node.entry.name)
+ else:
+ path = "%s.%s" % (class_name, node.entry.name)
+ else:
+ assert False
+ self.add_test(node.pos, path, node.doc)
+ return node