17 lines
512 B
Python
17 lines
512 B
Python
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'))
|