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
|
from numpy.testing import *
set_package_path()
import numpy as N
restore_path()
class test_ndpointer(NumpyTestCase):
def check_dtype(self):
dt = N.intc
p = N.ndpointer(dtype=dt)
self.assert_(p.from_param(N.array([1], dt)))
dt = '<i4'
p = N.ndpointer(dtype=dt)
self.assert_(p.from_param(N.array([1], dt)))
dt = N.dtype('>i4')
p = N.ndpointer(dtype=dt)
p.from_param(N.array([1], dt))
self.assertRaises(TypeError, p.from_param,
N.array([1], dt.newbyteorder('swap')))
dtnames = ['x', 'y']
dtformats = [N.intc, N.float64]
dtdescr = {'names' : dtnames, 'formats' : dtformats}
dt = N.dtype(dtdescr)
p = N.ndpointer(dtype=dt)
self.assert_(p.from_param(N.zeros((10,), dt)))
samedt = N.dtype(dtdescr)
p = N.ndpointer(dtype=samedt)
self.assert_(p.from_param(N.zeros((10,), dt)))
dt2 = N.dtype(dtdescr, align=True)
if dt.itemsize != dt2.itemsize:
self.assertRaises(TypeError, p.from_param, N.zeros((10,), dt2))
else:
self.assert_(p.from_param(N.zeros((10,), dt2)))
def check_ndim(self):
p = N.ndpointer(ndim=0)
self.assert_(p.from_param(N.array(1)))
self.assertRaises(TypeError, p.from_param, N.array([1]))
p = N.ndpointer(ndim=1)
self.assertRaises(TypeError, p.from_param, N.array(1))
self.assert_(p.from_param(N.array([1])))
p = N.ndpointer(ndim=2)
self.assert_(p.from_param(N.array([[1]])))
def check_shape(self):
p = N.ndpointer(shape=(1,2))
self.assert_(p.from_param(N.array([[1,2]])))
self.assertRaises(TypeError, p.from_param, N.array([[1],[2]]))
p = N.ndpointer(shape=())
self.assert_(p.from_param(N.array(1)))
def check_flags(self):
x = N.array([[1,2,3]], order='F')
p = N.ndpointer(flags='FORTRAN')
self.assert_(p.from_param(x))
p = N.ndpointer(flags='CONTIGUOUS')
self.assertRaises(TypeError, p.from_param, x)
p = N.ndpointer(flags=x.flags.num)
self.assert_(p.from_param(x))
self.assertRaises(TypeError, p.from_param, N.array([[1,2,3]]))
if __name__ == "__main__":
NumpyTest().run()
|
