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)