1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64
|
# mode: run
# tag: fused, pure3.6
#cython: annotation_typing=True
import cython
InPy = cython.fused_type(cython.int, cython.float)
class TestCls:
# although annotations as strings isn't recommended and generates a warning
# it does allow the test to run on more (pure) Python versions
def func1(self, arg: 'NotInPy'):
"""
>>> TestCls().func1(1.0)
'float'
>>> TestCls().func1(2)
'int'
"""
loc: 'NotInPy' = arg
return cython.typeof(arg)
if cython.compiled:
@cython.locals(arg=NotInPy, loc=NotInPy) # NameError for 'NotInPy' in pure Python
def func2(self, arg):
"""
>>> TestCls().func2(1.0)
'float'
>>> TestCls().func2(2)
'int'
"""
loc = arg
return cython.typeof(arg)
def cpfunc(self, arg):
"""
>>> TestCls().cpfunc(1.0)
'float'
>>> TestCls().cpfunc(2)
'int'
"""
loc = arg
return cython.typeof(arg)
def func1_inpy(self, arg: InPy):
"""
>>> TestCls().func1_inpy(1.0)
'float'
>>> TestCls().func1_inpy(2)
'int'
"""
loc: InPy = arg
return cython.typeof(arg)
@cython.locals(arg = InPy, loc = InPy)
def func2_inpy(self, arg):
"""
>>> TestCls().func2_inpy(1.0)
'float'
>>> TestCls().func2_inpy(2)
'int'
"""
loc = arg
return cython.typeof(arg)
|