def test_make_mat(self):
self.array = matrix.mat_array(self.array, row_axes=(0, 1),
col_axes=(0, 2))
self.assertEqual(self.array.info['type'], 'mat')
self.assertEqual(self.array.axes, (None, None, None))
self.assertEqual(self.array.rows, (0, 1))
self.assertEqual(self.array.cols, (0, 2))
self.assertTrue(isinstance(self.array, matrix.MatrixObject))
def test_make_mat(self):
self.array = matrix.mat_array(self.array,
row_axes=(0, 1),
col_axes=(0, 2))
self.assertEqual(self.array.info['type'], 'mat')
self.assertEqual(self.array.axes, (None, None, None))
self.assertEqual(self.array.rows, (0, 1))
self.assertEqual(self.array.cols, (0, 2))
self.assertTrue(isinstance(self.array, matrix.MatrixObject))
def setUp(self):
data = sp.arange(30, dtype=float)
data.shape = (5, 2, 3)
self.vect = info_header.InfoArray(data)
self.vect = vector.vect_array(self.vect,
axis_names=('freq', 'a', 'b'))
data = sp.arange(120, dtype=float)
self.mat = info_header.InfoArray(data)
self.mat.shape = (5, 4, 6)
self.mat = matrix.mat_array(self.mat,
row_axes=(0, 1), col_axes=(0, 2),
axis_names=('freq', 'mode1', 'mode2'))
def setUp(self):
data = sp.arange(30, dtype=float)
data.shape = (5, 2, 3)
self.vect = info_header.InfoArray(data)
self.vect = vector.vect_array(self.vect, axis_names=('freq', 'a', 'b'))
data = sp.arange(120, dtype=float)
self.mat = info_header.InfoArray(data)
self.mat.shape = (5, 4, 6)
self.mat = matrix.mat_array(self.mat,
row_axes=(0, 1),
col_axes=(0, 2),
axis_names=('freq', 'mode1', 'mode2'))
def test_to_from_file(self):
"""Test that vects and mats can be written to and from file and have
all thier properties preserved."""
# For vectors.
file_io.save('temp.npy', self.vect)
new_vect = vector.vect_array(file_io.load('temp.npy'))
self.assertTrue(sp.allclose(self.vect, new_vect))
self.assertEqual(self.vect.axes, new_vect.axes)
# For matricies.
file_io.save('temp.npy', self.mat)
new_mat = matrix.mat_array(file_io.load('temp.npy'))
self.assertTrue(sp.allclose(self.mat, new_mat))
self.assertEqual(self.mat.axes, new_mat.axes)
# Messing with stuf should raise exceptions.
new_mat = file_io.load('temp.npy')
new_mat.info['cols'] = (0, 3)
self.assertRaises(ValueError, matrix.mat_array, new_mat)
# Clean up
os.remove('temp.npy')
os.remove('temp.npy.meta')
def test_to_from_file(self):
"""Test that vects and mats can be written to and from file and have
all thier properties preserved."""
# For vectors.
file_io.save('temp.npy', self.vect)
new_vect = vector.vect_array(file_io.load('temp.npy'))
self.assertTrue(sp.allclose(self.vect, new_vect))
self.assertEqual(self.vect.axes, new_vect.axes)
# For matricies.
file_io.save('temp.npy', self.mat)
new_mat = matrix.mat_array(file_io.load('temp.npy'))
self.assertTrue(sp.allclose(self.mat, new_mat))
self.assertEqual(self.mat.axes, new_mat.axes)
# Messing with stuf should raise exceptions.
new_mat = file_io.load('temp.npy')
new_mat.info['cols'] = (0, 3)
self.assertRaises(ValueError, matrix.mat_array, new_mat)
# Clean up
os.remove('temp.npy')
os.remove('temp.npy.meta')
