diff options
Diffstat (limited to 'numpy/matrixlib/tests/test_multiarray.py')
-rw-r--r-- | numpy/matrixlib/tests/test_multiarray.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/numpy/matrixlib/tests/test_multiarray.py b/numpy/matrixlib/tests/test_multiarray.py new file mode 100644 index 000000000..9f2dce7e4 --- /dev/null +++ b/numpy/matrixlib/tests/test_multiarray.py @@ -0,0 +1,16 @@ +import numpy as np +from numpy.testing import * + +class TestView(TestCase): + def test_type(self): + x = np.array([1,2,3]) + assert(isinstance(x.view(np.matrix),np.matrix)) + + def test_keywords(self): + x = np.array([(1,2)],dtype=[('a',np.int8),('b',np.int8)]) + # We must be specific about the endianness here: + y = x.view(dtype='<i2', type=np.matrix) + assert_array_equal(y,[[513]]) + + assert(isinstance(y,np.matrix)) + assert_equal(y.dtype, np.dtype('<i2')) |