| from Cython.TestUtils import CythonTest |
| import Cython.Compiler.Errors as Errors |
| from Cython.Compiler.Nodes import * |
| from Cython.Compiler.ParseTreeTransforms import * |
| from Cython.Compiler.Buffer import * |
|
|
|
|
| class TestMemviewParsing(CythonTest): |
|
|
| def parse(self, s): |
| return self.should_not_fail(lambda: self.fragment(s)).root |
|
|
| def not_parseable(self, expected_error, s): |
| e = self.should_fail(lambda: self.fragment(s), Errors.CompileError) |
| self.assertEqual(expected_error, e.message_only) |
|
|
| def test_default_1dim(self): |
| self.parse("cdef int[:] x") |
| self.parse("cdef short int[:] x") |
|
|
| def test_default_ndim(self): |
| self.parse("cdef int[:,:,:,:,:] x") |
| self.parse("cdef unsigned long int[:,:,:,:,:] x") |
| self.parse("cdef unsigned int[:,:,:,:,:] x") |
|
|
| def test_zero_offset(self): |
| self.parse("cdef long double[0:] x") |
| self.parse("cdef int[0:] x") |
|
|
| def test_zero_offset_ndim(self): |
| self.parse("cdef int[0:,0:,0:,0:] x") |
|
|
| def test_def_arg(self): |
| self.parse("def foo(int[:,:] x): pass") |
|
|
| def test_cdef_arg(self): |
| self.parse("cdef foo(int[:,:] x): pass") |
|
|
| def test_general_slice(self): |
| self.parse('cdef float[::ptr, ::direct & contig, 0::full & strided] x') |
|
|
| def test_non_slice_memview(self): |
| self.not_parseable("An axis specification in memoryview declaration does not have a ':'.", |
| "cdef double[:foo, bar] x") |
| self.not_parseable("An axis specification in memoryview declaration does not have a ':'.", |
| "cdef double[0:foo, bar] x") |
|
|
| def test_basic(self): |
| t = self.parse("cdef int[:] x") |
| memv_node = t.stats[0].base_type |
| self.assertTrue(isinstance(memv_node, MemoryViewSliceTypeNode)) |
|
|
| |
| |
|
|
| def disable_test_no_buf_arg(self): |
| self.not_parseable("Expected ']'", |
| "cdef extern foo(object[int, ndim=2])") |
|
|
| def disable_test_parse_sizeof(self): |
| self.parse("sizeof(int[NN])") |
| self.parse("sizeof(int[])") |
| self.parse("sizeof(int[][NN])") |
| self.not_parseable("Expected an identifier or literal", |
| "sizeof(int[:NN])") |
| self.not_parseable("Expected ']'", |
| "sizeof(foo[dtype=bar]") |
|
|
| if __name__ == '__main__': |
| import unittest |
| unittest.main() |
|
|
