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