summaryrefslogtreecommitdiff
path: root/numpy/matrixlib/tests/test_multiarray.py
blob: bed05561542f91245caa02b6c026c2ee771dac38 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from __future__ import division, absolute_import, print_function

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'))