|
|
import os, tempfile |
|
|
from Cython.Shadow import inline |
|
|
from Cython.Build.Inline import safe_type |
|
|
from Cython.TestUtils import CythonTest |
|
|
|
|
|
try: |
|
|
import numpy |
|
|
has_numpy = True |
|
|
except: |
|
|
has_numpy = False |
|
|
|
|
|
test_kwds = dict(force=True, quiet=True) |
|
|
|
|
|
global_value = 100 |
|
|
|
|
|
class TestInline(CythonTest): |
|
|
def setUp(self): |
|
|
CythonTest.setUp(self) |
|
|
self.test_kwds = dict(test_kwds) |
|
|
if os.path.isdir('TEST_TMP'): |
|
|
lib_dir = os.path.join('TEST_TMP','inline') |
|
|
else: |
|
|
lib_dir = tempfile.mkdtemp(prefix='cython_inline_') |
|
|
self.test_kwds['lib_dir'] = lib_dir |
|
|
|
|
|
def test_simple(self): |
|
|
self.assertEqual(inline("return 1+2", **self.test_kwds), 3) |
|
|
|
|
|
def test_types(self): |
|
|
self.assertEqual(inline(""" |
|
|
cimport cython |
|
|
return cython.typeof(a), cython.typeof(b) |
|
|
""", a=1.0, b=[], **self.test_kwds), ('double', 'list object')) |
|
|
|
|
|
def test_locals(self): |
|
|
a = 1 |
|
|
b = 2 |
|
|
self.assertEqual(inline("return a+b", **self.test_kwds), 3) |
|
|
|
|
|
def test_globals(self): |
|
|
self.assertEqual(inline("return global_value + 1", **self.test_kwds), global_value + 1) |
|
|
|
|
|
def test_no_return(self): |
|
|
self.assertEqual(inline(""" |
|
|
a = 1 |
|
|
cdef double b = 2 |
|
|
cdef c = [] |
|
|
""", **self.test_kwds), dict(a=1, b=2.0, c=[])) |
|
|
|
|
|
def test_def_node(self): |
|
|
foo = inline("def foo(x): return x * x", **self.test_kwds)['foo'] |
|
|
self.assertEqual(foo(7), 49) |
|
|
|
|
|
def test_class_ref(self): |
|
|
class Type(object): |
|
|
pass |
|
|
tp = inline("Type")['Type'] |
|
|
self.assertEqual(tp, Type) |
|
|
|
|
|
def test_pure(self): |
|
|
import cython as cy |
|
|
b = inline(""" |
|
|
b = cy.declare(float, a) |
|
|
c = cy.declare(cy.pointer(cy.float), &b) |
|
|
return b |
|
|
""", a=3, **self.test_kwds) |
|
|
self.assertEqual(type(b), float) |
|
|
|
|
|
def test_compiler_directives(self): |
|
|
self.assertEqual( |
|
|
inline('return sum(x)', |
|
|
x=[1, 2, 3], |
|
|
cython_compiler_directives={'boundscheck': False}), |
|
|
6 |
|
|
) |
|
|
|
|
|
def test_lang_version(self): |
|
|
|
|
|
inline_divcode = "def f(int a, int b): return a/b" |
|
|
self.assertEqual( |
|
|
inline(inline_divcode, language_level=2)['f'](5,2), |
|
|
2 |
|
|
) |
|
|
self.assertEqual( |
|
|
inline(inline_divcode, language_level=3)['f'](5,2), |
|
|
2.5 |
|
|
) |
|
|
|
|
|
if has_numpy: |
|
|
|
|
|
def test_numpy(self): |
|
|
import numpy |
|
|
a = numpy.ndarray((10, 20)) |
|
|
a[0,0] = 10 |
|
|
self.assertEqual(safe_type(a), 'numpy.ndarray[numpy.float64_t, ndim=2]') |
|
|
self.assertEqual(inline("return a[0,0]", a=a, **self.test_kwds), 10.0) |
|
|
|
