def test_from_info(self) :
arr = algebra.info_array(self.data)
mat_arr = algebra.make_mat(arr, (0,1), (0,2))
self.assertTrue(isinstance(mat_arr, algebra.mat))
mem = algebra.info_memmap(self.memmap_data)
vect_mem = algebra.make_vect(mem)
self.assertTrue(isinstance(vect_mem, algebra.vect))
def test_from_info(self):
arr = algebra.info_array(self.data)
mat_arr = algebra.make_mat(arr, (0, 1), (0, 2))
self.assertTrue(isinstance(mat_arr, algebra.mat))
mem = algebra.info_memmap(self.memmap_data)
vect_mem = algebra.make_vect(mem)
self.assertTrue(isinstance(vect_mem, algebra.vect))
def test_from_memmap(self) :
# Works if constructed from array.
data = npfor.open_memmap('temp.npy', mode='w+', shape=(4,3,3))
data[:] = 5.0
Mat = algebra.info_memmap(data, {'a': 'b'})
Mat.flush()
self.assertEqual(Mat.shape, (4, 3, 3))
self.assertEqual(Mat.info['a'], 'b')
self.assertTrue(sp.allclose(Mat, 5.0))
self.assertTrue(isinstance(Mat, sp.memmap))
del Mat
os.remove('temp.npy')
def test_from_memmap(self):
# Works if constructed from array.
data = npfor.open_memmap('temp.npy', mode='w+', shape=(4, 3, 3))
data[:] = 5.0
Mat = algebra.info_memmap(data, {'a': 'b'})
Mat.flush()
self.assertEqual(Mat.shape, (4, 3, 3))
self.assertEqual(Mat.info['a'], 'b')
self.assertTrue(sp.allclose(Mat, 5.0))
self.assertTrue(isinstance(Mat, sp.memmap))
del Mat
os.remove('temp.npy')
def test_assert_info(self) :
"""Test the assert_info function."""
# info_memaps should pass.
data = npfor.open_memmap('temp.npy', mode='w+', shape=(4,3,3))
data[:] = 5.0
Mat = algebra.info_memmap(data)
algebra.assert_info(Mat)
del Mat
os.remove('temp.npy')
# info_arrays should pass.
data = sp.empty((5, 6, 6))
data[:] = 4.0
Mat = algebra.info_array(data)
algebra.assert_info(Mat)
# arrays should fail.
self.assertRaises(TypeError, algebra.assert_info, data)
def test_assert_info(self):
"""Test the assert_info function."""
# info_memaps should pass.
data = npfor.open_memmap('temp.npy', mode='w+', shape=(4, 3, 3))
data[:] = 5.0
Mat = algebra.info_memmap(data)
algebra.assert_info(Mat)
del Mat
os.remove('temp.npy')
# info_arrays should pass.
data = sp.empty((5, 6, 6))
data[:] = 4.0
Mat = algebra.info_array(data)
algebra.assert_info(Mat)
# arrays should fail.
self.assertRaises(TypeError, algebra.assert_info, data)
