def test_chebyshev_iteration_1(self):
''' Tests the pcd_shell operators produce correct output. '''
A = self.quad_mass_matrix
n = self.quad_mass_matrix.shape[0]
alpha = 1. / 4
beta = 9. / 4
x0 = np.zeros(n)
b1 = np.ones(n)
for i in range(0, n, 2):
b1[i] = 0.
A_petsc = LAT.dense_numpy_2_petsc4py(A)
x0_petsc = p4pyPETSc.Vec().createWithArray(x0)
b1_petsc = p4pyPETSc.Vec().createWithArray(b1)
solver = LS.ChebyshevSemiIteration(A_petsc, alpha, beta, True)
solver.apply(b1_petsc, x0_petsc, 20)
expected = np.load(
os.path.join(self._scriptdir, 'import_modules/sol_10.npy'))
actual = x0_petsc
assert np.allclose(expected, actual.getArray())
def test_chebyshev_iteration_2(self):
''' Tests the pcd_shell operators produce correct output. '''
A = np.diag(1. / np.diag(self.quad_mass_matrix)).dot(
self.quad_mass_matrix)
n = self.quad_mass_matrix.shape[0]
alpha = 1. / 4
beta = 9. / 4
x0 = np.zeros(n)
b1 = np.zeros(n)
for i in range(0, n):
b1[i] = i
A_petsc = LAT.dense_numpy_2_petsc4py(A)
x0_petsc = p4pyPETSc.Vec().createWithArray(x0)
b1_petsc = p4pyPETSc.Vec().createWithArray(b1)
solver = LS.ChebyshevSemiIteration(A_petsc,
alpha,
beta,
save_iterations=True)
solver.apply(b1_petsc, x0_petsc, 20)
expected = np.load(
os.path.join(self._scriptdir, 'import_modules/sol_20_lst.npy'))
for i, item in enumerate(expected):
assert np.allclose(item, solver.iteration_results[i], 1e-12)