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