diff options
| author | Jarrod Millman <millman@berkeley.edu> | 2007-12-29 02:21:56 +0000 |
|---|---|---|
| committer | Jarrod Millman <millman@berkeley.edu> | 2007-12-29 02:21:56 +0000 |
| commit | 72297ed57ac666c6ef50d5c88ef486a94f7fc080 (patch) | |
| tree | 9ee064e4147d76e2b154e35103a7f46a6b3081c6 /numpy/tests | |
| parent | 423b1953b1742a2b71cc766ff08a919707f19e1c (diff) | |
| download | numpy-72297ed57ac666c6ef50d5c88ef486a94f7fc080.tar.gz | |
more janitorial work
Diffstat (limited to 'numpy/tests')
| -rw-r--r-- | numpy/tests/test_ctypeslib.py | 54 |
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() |
