mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-28 02:02:39 +08:00
Rowan Cockett
58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
import unittest
|
|
from SimPEG import *
|
|
from SimPEG.Mesh import TensorMesh
|
|
from SimPEG.Utils import sdiag
|
|
import numpy as np
|
|
import scipy.sparse as sparse
|
|
|
|
TOLD = 1e-10
|
|
TOLI = 1e-3
|
|
numRHS = 5
|
|
|
|
def dotest(MYSOLVER, multi=False, A=None, **solverOpts):
|
|
if A is None:
|
|
h1 = np.ones(10)*100.
|
|
h2 = np.ones(10)*100.
|
|
h3 = np.ones(10)*100.
|
|
|
|
h = [h1,h2,h3]
|
|
|
|
M = TensorMesh(h)
|
|
|
|
D = M.faceDiv
|
|
G = -M.faceDiv.T
|
|
Msig = M.getFaceInnerProduct()
|
|
A = D*Msig*G
|
|
A[-1,-1] *= 1/M.vol[-1] # remove the constant null space from the matrix
|
|
else:
|
|
M = Mesh.TensorMesh([A.shape[0]])
|
|
|
|
Ainv = MYSOLVER(A, **solverOpts)
|
|
if multi:
|
|
e = np.ones(M.nC)
|
|
else:
|
|
e = np.ones((M.nC, numRHS))
|
|
rhs = A * e
|
|
x = Ainv * rhs
|
|
Ainv.clean()
|
|
return np.linalg.norm(e-x,np.inf)
|
|
|
|
class TestSolver(unittest.TestCase):
|
|
|
|
def test_direct_spsolve_1(self): self.assertLess(dotest(Solver, False),TOLD)
|
|
def test_direct_spsolve_M(self): self.assertLess(dotest(Solver, True),TOLD)
|
|
|
|
def test_direct_splu_1(self): self.assertLess(dotest(SolverLU, False),TOLD)
|
|
def test_direct_splu_M(self): self.assertLess(dotest(SolverLU, True),TOLD)
|
|
|
|
def test_iterative_diag_1(self): self.assertLess(dotest(SolverDiag, False, A=Utils.sdiag(np.random.rand(10)+1.0)),TOLI)
|
|
def test_iterative_diag_M(self): self.assertLess(dotest(SolverDiag, True, A=Utils.sdiag(np.random.rand(10)+1.0)),TOLI)
|
|
|
|
def test_iterative_cg_1(self): self.assertLess(dotest(SolverCG, False),TOLI)
|
|
def test_iterative_cg_M(self): self.assertLess(dotest(SolverCG, True),TOLI)
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|