Chris@87: from __future__ import division, absolute_import, print_function
Chris@87: 
Chris@87: import numpy as np
Chris@87: from numpy.testing import *
Chris@87: 
Chris@87: class TestView(TestCase):
Chris@87:     def test_type(self):
Chris@87:         x = np.array([1, 2, 3])
Chris@87:         assert_(isinstance(x.view(np.matrix), np.matrix))
Chris@87: 
Chris@87:     def test_keywords(self):
Chris@87:         x = np.array([(1, 2)], dtype=[('a', np.int8), ('b', np.int8)])
Chris@87:         # We must be specific about the endianness here:
Chris@87:         y = x.view(dtype='<i2', type=np.matrix)
Chris@87:         assert_array_equal(y, [[513]])
Chris@87: 
Chris@87:         assert_(isinstance(y, np.matrix))
Chris@87:         assert_equal(y.dtype, np.dtype('<i2'))