def test_make_mat(self) :
self.array = algebra.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, algebra.mat))
def test_make_mat(self):
self.array = algebra.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, algebra.mat))
def setUp(self) :
data = sp.arange(30)
data.shape = (5, 2, 3)
self.vect = algebra.info_array(data)
self.vect = algebra.vect_array(self.vect,
axis_names=('freq', 'a', 'b'))
data = sp.arange(120)
self.mat = algebra.info_array(data)
self.mat.shape = (5, 4, 6)
self.mat = algebra.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 = algebra.info_array(data)
self.vect = algebra.vect_array(self.vect,
axis_names=('freq', 'a', 'b'))
data = sp.arange(120, dtype=float)
self.mat = algebra.info_array(data)
self.mat.shape = (5, 4, 6)
self.mat = algebra.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.
algebra.save('temp.npy', self.vect)
new_vect = algebra.vect_array(algebra.load('temp.npy'))
self.assertTrue(sp.allclose(self.vect, new_vect))
self.assertEqual(self.vect.axes, new_vect.axes)
# For matricies.
algebra.save('temp.npy', self.mat)
new_mat = algebra.mat_array(algebra.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 = algebra.load('temp.npy')
new_mat.info['cols'] = (0,3)
self.assertRaises(ValueError, algebra.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.
algebra.save('temp.npy', self.vect)
new_vect = algebra.vect_array(algebra.load('temp.npy'))
self.assertTrue(sp.allclose(self.vect, new_vect))
self.assertEqual(self.vect.axes, new_vect.axes)
# For matricies.
algebra.save('temp.npy', self.mat)
new_mat = algebra.mat_array(algebra.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 = algebra.load('temp.npy')
new_mat.info['cols'] = (0, 3)
self.assertRaises(ValueError, algebra.mat_array, new_mat)
# Clean up
os.remove('temp.npy')
os.remove('temp.npy.meta')
