61 lines
2.2 KiB
Python
61 lines
2.2 KiB
Python
from Cython.Compiler.ModuleNode import ModuleNode
|
|
from Cython.Compiler.Symtab import ModuleScope
|
|
from Cython.TestUtils import TransformTest
|
|
from Cython.Compiler.Visitor import MethodDispatcherTransform
|
|
from Cython.Compiler.ParseTreeTransforms import (
|
|
NormalizeTree, AnalyseDeclarationsTransform,
|
|
AnalyseExpressionsTransform, InterpretCompilerDirectives)
|
|
|
|
|
|
class TestMethodDispatcherTransform(TransformTest):
|
|
_tree = None
|
|
|
|
def _build_tree(self):
|
|
if self._tree is None:
|
|
context = None
|
|
|
|
def fake_module(node):
|
|
scope = ModuleScope('test', None, None)
|
|
return ModuleNode(node.pos, doc=None, body=node,
|
|
scope=scope, full_module_name='test',
|
|
directive_comments={})
|
|
pipeline = [
|
|
fake_module,
|
|
NormalizeTree(context),
|
|
InterpretCompilerDirectives(context, {}),
|
|
AnalyseDeclarationsTransform(context),
|
|
AnalyseExpressionsTransform(context),
|
|
]
|
|
self._tree = self.run_pipeline(pipeline, u"""
|
|
cdef bytes s = b'asdfg'
|
|
cdef dict d = {1:2}
|
|
x = s * 3
|
|
d.get('test')
|
|
""")
|
|
return self._tree
|
|
|
|
def test_builtin_method(self):
|
|
calls = [0]
|
|
class Test(MethodDispatcherTransform):
|
|
def _handle_simple_method_dict_get(self, node, func, args, unbound):
|
|
calls[0] += 1
|
|
return node
|
|
|
|
tree = self._build_tree()
|
|
Test(None)(tree)
|
|
self.assertEqual(1, calls[0])
|
|
|
|
def test_binop_method(self):
|
|
calls = {'bytes': 0, 'object': 0}
|
|
class Test(MethodDispatcherTransform):
|
|
def _handle_simple_method_bytes___mul__(self, node, func, args, unbound):
|
|
calls['bytes'] += 1
|
|
return node
|
|
def _handle_simple_method_object___mul__(self, node, func, args, unbound):
|
|
calls['object'] += 1
|
|
return node
|
|
|
|
tree = self._build_tree()
|
|
Test(None)(tree)
|
|
self.assertEqual(1, calls['bytes'])
|
|
self.assertEqual(0, calls['object'])
|