diff options
author | Alan McIntyre <alan.mcintyre@local> | 2008-07-22 23:02:29 +0000 |
---|---|---|
committer | Alan McIntyre <alan.mcintyre@local> | 2008-07-22 23:02:29 +0000 |
commit | 681aa536342db1332a5af9b632495ddc883ecba9 (patch) | |
tree | 75fee25252914fcb9a10f465dcd60fc8f2fd33ee /numpy/ma/tests/test_subclassing.py | |
parent | 6dd88dc3b6c94d54585f639c533c00bf0928e13a (diff) | |
download | numpy-681aa536342db1332a5af9b632495ddc883ecba9.tar.gz |
Standardize NumPy import as "import numpy as np".
Diffstat (limited to 'numpy/ma/tests/test_subclassing.py')
-rw-r--r-- | numpy/ma/tests/test_subclassing.py | 26 |
1 files changed, 13 insertions, 13 deletions
diff --git a/numpy/ma/tests/test_subclassing.py b/numpy/ma/tests/test_subclassing.py index 4d9842506..eff7b268d 100644 --- a/numpy/ma/tests/test_subclassing.py +++ b/numpy/ma/tests/test_subclassing.py @@ -10,25 +10,25 @@ __version__ = '1.0' __revision__ = "$Revision: 3473 $" __date__ = '$Date: 2007-10-29 17:18:13 +0200 (Mon, 29 Oct 2007) $' -import numpy as N +import numpy as np import numpy.core.numeric as numeric from numpy.testing import * from numpy.ma.testutils import * from numpy.ma.core import * -class SubArray(N.ndarray): - """Defines a generic N.ndarray subclass, that stores some metadata +class SubArray(np.ndarray): + """Defines a generic np.ndarray subclass, that stores some metadata in the dictionary `info`.""" def __new__(cls,arr,info={}): - x = N.asanyarray(arr).view(cls) + x = np.asanyarray(arr).view(cls) x.info = info return x def __array_finalize__(self, obj): self.info = getattr(obj,'info',{}) return def __add__(self, other): - result = N.ndarray.__add__(self, other) + result = np.ndarray.__add__(self, other) result.info.update({'added':result.info.pop('added',0)+1}) return result @@ -52,13 +52,13 @@ class MSubArray(SubArray,MaskedArray): msubarray = MSubArray -class MMatrix(MaskedArray, N.matrix,): +class MMatrix(MaskedArray, np.matrix,): def __new__(cls, data, mask=nomask): - mat = N.matrix(data) + mat = np.matrix(data) _data = MaskedArray.__new__(cls, data=mat, mask=mask) return _data def __array_finalize__(self,obj): - N.matrix.__array_finalize__(self, obj) + np.matrix.__array_finalize__(self, obj) MaskedArray.__array_finalize__(self,obj) return def _get_series(self): @@ -74,7 +74,7 @@ class TestSubclassing(TestCase): def test_data_subclassing(self): "Tests whether the subclass is kept." - x = N.arange(5) + x = np.arange(5) m = [0,0,1,0,0] xsub = SubArray(x) xmsub = masked_array(xsub, mask=m) @@ -84,14 +84,14 @@ class TestSubclassing(TestCase): def test_maskedarray_subclassing(self): "Tests subclassing MaskedArray" - x = N.arange(5) + x = np.arange(5) mx = mmatrix(x,mask=[0,1,0,0,0]) - assert isinstance(mx._data, N.matrix) + assert isinstance(mx._data, np.matrix) "Tests masked_unary_operation" assert isinstance(add(mx,mx), mmatrix) assert isinstance(add(mx,x), mmatrix) assert_equal(add(mx,x), mx+x) - assert isinstance(add(mx,mx)._data, N.matrix) + assert isinstance(add(mx,mx)._data, np.matrix) assert isinstance(add.outer(mx,mx), mmatrix) "Tests masked_binary_operation" assert isinstance(hypot(mx,mx), mmatrix) @@ -126,7 +126,7 @@ class TestSubclassing(TestCase): def test_subclasspreservation(self): "Checks that masked_array(...,subok=True) preserves the class." - x = N.arange(5) + x = np.arange(5) m = [0,0,1,0,0] xinfo = [(i,j) for (i,j) in zip(x,m)] xsub = MSubArray(x, mask=m, info={'xsub':xinfo}) |