summaryrefslogtreecommitdiff
path: root/numpy/tests
diff options
context:
space:
mode:
authorJarrod Millman <millman@berkeley.edu>2007-12-29 02:21:56 +0000
committerJarrod Millman <millman@berkeley.edu>2007-12-29 02:21:56 +0000
commit72297ed57ac666c6ef50d5c88ef486a94f7fc080 (patch)
tree9ee064e4147d76e2b154e35103a7f46a6b3081c6 /numpy/tests
parent423b1953b1742a2b71cc766ff08a919707f19e1c (diff)
downloadnumpy-72297ed57ac666c6ef50d5c88ef486a94f7fc080.tar.gz
more janitorial work
Diffstat (limited to 'numpy/tests')
-rw-r--r--numpy/tests/test_ctypeslib.py54
1 files changed, 26 insertions, 28 deletions
diff --git a/numpy/tests/test_ctypeslib.py b/numpy/tests/test_ctypeslib.py
index 875e1a1c7..5558445b1 100644
--- a/numpy/tests/test_ctypeslib.py
+++ b/numpy/tests/test_ctypeslib.py
@@ -1,63 +1,61 @@
-from numpy.testing import *
-set_package_path()
-import numpy as N
+import numpy as np
from numpy.ctypeslib import ndpointer
-restore_path()
+from numpy.testing import *
class TestNdpointer(NumpyTestCase):
def check_dtype(self):
- dt = N.intc
+ dt = np.intc
p = ndpointer(dtype=dt)
- self.assert_(p.from_param(N.array([1], dt)))
+ self.assert_(p.from_param(np.array([1], dt)))
dt = '<i4'
p = ndpointer(dtype=dt)
- self.assert_(p.from_param(N.array([1], dt)))
- dt = N.dtype('>i4')
+ self.assert_(p.from_param(np.array([1], dt)))
+ dt = np.dtype('>i4')
p = ndpointer(dtype=dt)
- p.from_param(N.array([1], dt))
+ p.from_param(np.array([1], dt))
self.assertRaises(TypeError, p.from_param,
- N.array([1], dt.newbyteorder('swap')))
+ np.array([1], dt.newbyteorder('swap')))
dtnames = ['x', 'y']
- dtformats = [N.intc, N.float64]
+ dtformats = [np.intc, np.float64]
dtdescr = {'names' : dtnames, 'formats' : dtformats}
- dt = N.dtype(dtdescr)
+ dt = np.dtype(dtdescr)
p = ndpointer(dtype=dt)
- self.assert_(p.from_param(N.zeros((10,), dt)))
- samedt = N.dtype(dtdescr)
+ self.assert_(p.from_param(np.zeros((10,), dt)))
+ samedt = np.dtype(dtdescr)
p = ndpointer(dtype=samedt)
- self.assert_(p.from_param(N.zeros((10,), dt)))
- dt2 = N.dtype(dtdescr, align=True)
+ self.assert_(p.from_param(np.zeros((10,), dt)))
+ dt2 = np.dtype(dtdescr, align=True)
if dt.itemsize != dt2.itemsize:
- self.assertRaises(TypeError, p.from_param, N.zeros((10,), dt2))
+ self.assertRaises(TypeError, p.from_param, np.zeros((10,), dt2))
else:
- self.assert_(p.from_param(N.zeros((10,), dt2)))
+ self.assert_(p.from_param(np.zeros((10,), dt2)))
def check_ndim(self):
p = ndpointer(ndim=0)
- self.assert_(p.from_param(N.array(1)))
- self.assertRaises(TypeError, p.from_param, N.array([1]))
+ self.assert_(p.from_param(np.array(1)))
+ self.assertRaises(TypeError, p.from_param, np.array([1]))
p = ndpointer(ndim=1)
- self.assertRaises(TypeError, p.from_param, N.array(1))
- self.assert_(p.from_param(N.array([1])))
+ self.assertRaises(TypeError, p.from_param, np.array(1))
+ self.assert_(p.from_param(np.array([1])))
p = ndpointer(ndim=2)
- self.assert_(p.from_param(N.array([[1]])))
+ self.assert_(p.from_param(np.array([[1]])))
def check_shape(self):
p = ndpointer(shape=(1,2))
- self.assert_(p.from_param(N.array([[1,2]])))
- self.assertRaises(TypeError, p.from_param, N.array([[1],[2]]))
+ self.assert_(p.from_param(np.array([[1,2]])))
+ self.assertRaises(TypeError, p.from_param, np.array([[1],[2]]))
p = ndpointer(shape=())
- self.assert_(p.from_param(N.array(1)))
+ self.assert_(p.from_param(np.array(1)))
def check_flags(self):
- x = N.array([[1,2,3]], order='F')
+ x = np.array([[1,2,3]], order='F')
p = ndpointer(flags='FORTRAN')
self.assert_(p.from_param(x))
p = ndpointer(flags='CONTIGUOUS')
self.assertRaises(TypeError, p.from_param, x)
p = ndpointer(flags=x.flags.num)
self.assert_(p.from_param(x))
- self.assertRaises(TypeError, p.from_param, N.array([[1,2,3]]))
+ self.assertRaises(TypeError, p.from_param, np.array([[1,2,3]]))
if __name__ == "__main__":
NumpyTest().run()