mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-29 04:10:59 +08:00
rowanc1
d58cb21123
Speedups are from brackets mat-vec prods vs mat-mat Memory is from not setting all 'b' fields to zero in Gvec
178 lines
6.1 KiB
Python
178 lines
6.1 KiB
Python
import unittest
|
|
from SimPEG import *
|
|
import simpegEM as EM
|
|
import sys
|
|
|
|
TOL = 1e-4
|
|
CONDUCTIVITY = 1e3
|
|
|
|
def getProblem(fdemType, comp):
|
|
cs = 5.
|
|
ncx, ncy, ncz = 6, 6, 6
|
|
npad = 3
|
|
hx = [(cs,npad,-1.3), (cs,ncx), (cs,npad,1.3)]
|
|
hy = [(cs,npad,-1.3), (cs,ncy), (cs,npad,1.3)]
|
|
hz = [(cs,npad,-1.3), (cs,ncz), (cs,npad,1.3)]
|
|
mesh = Mesh.TensorMesh([hx,hy,hz],['C','C','C'])
|
|
|
|
mapping = Maps.ExpMap(mesh)
|
|
|
|
x = np.linspace(-30,30,6)
|
|
XYZ = Utils.ndgrid(x,x,np.r_[0])
|
|
Rx0 = EM.FDEM.RxFDEM(XYZ, comp)
|
|
Tx0 = EM.FDEM.TxFDEM(np.r_[4.,2.,2.], 'VMD', 1e-2, [Rx0])
|
|
|
|
survey = EM.FDEM.SurveyFDEM([Tx0])
|
|
|
|
if fdemType == 'e':
|
|
prb = EM.FDEM.ProblemFDEM_e(mesh, mapping=mapping)
|
|
elif fdemType == 'b':
|
|
prb = EM.FDEM.ProblemFDEM_b(mesh, mapping=mapping)
|
|
else:
|
|
raise NotImplementedError()
|
|
prb.pair(survey)
|
|
|
|
try:
|
|
from pymatsolver import MumpsSolver
|
|
prb.Solver = MumpsSolver
|
|
except ImportError, e:
|
|
pass
|
|
|
|
return prb
|
|
|
|
def adjointTest(fdemType, comp):
|
|
prb = getProblem(fdemType, comp)
|
|
print 'Adjoint %s formulation - %s' % (fdemType, comp)
|
|
|
|
m = np.log(np.ones(prb.mesh.nC)*CONDUCTIVITY)
|
|
survey = prb.survey
|
|
|
|
v = np.random.rand(survey.nD)
|
|
w = np.random.rand(prb.mapping.nP)
|
|
|
|
u = prb.fields(m)
|
|
vJw = v.dot(prb.Jvec(m, w, u=u))
|
|
wJtv = w.dot(prb.Jtvec(m, v, u=u))
|
|
tol = TOL*(10**int(np.log10(np.abs(vJw))))
|
|
print vJw, wJtv, vJw - wJtv, tol, np.abs(vJw - wJtv) < tol
|
|
return np.abs(vJw - wJtv) < tol
|
|
|
|
def derivTest(fdemType, comp):
|
|
prb = getProblem(fdemType, comp)
|
|
print '%s formulation - %s' % (fdemType, comp)
|
|
x0 = np.log(np.ones(prb.mesh.nC)*CONDUCTIVITY)
|
|
survey = prb.survey
|
|
def fun(x):
|
|
return survey.dpred(x), lambda x: prb.Jvec(x0, x)
|
|
return Tests.checkDerivative(fun, x0, num=3, plotIt=False, eps=1e-25)
|
|
|
|
|
|
class FDEM_DerivTests(unittest.TestCase):
|
|
|
|
def test_Jvec_exr_Eform(self):
|
|
self.assertTrue(derivTest('e', 'exr'))
|
|
def test_Jvec_exr_Bform(self):
|
|
self.assertTrue(derivTest('b', 'exr'))
|
|
def test_Jvec_eyr_Eform(self):
|
|
self.assertTrue(derivTest('e', 'eyr'))
|
|
def test_Jvec_eyr_Bform(self):
|
|
self.assertTrue(derivTest('b', 'eyr'))
|
|
def test_Jvec_ezr_Eform(self):
|
|
self.assertTrue(derivTest('e', 'ezr'))
|
|
def test_Jvec_ezr_Bform(self):
|
|
self.assertTrue(derivTest('b', 'ezr'))
|
|
def test_Jvec_exi_Eform(self):
|
|
self.assertTrue(derivTest('e', 'exi'))
|
|
def test_Jvec_exi_Bform(self):
|
|
self.assertTrue(derivTest('b', 'exi'))
|
|
def test_Jvec_eyi_Eform(self):
|
|
self.assertTrue(derivTest('e', 'eyi'))
|
|
def test_Jvec_eyi_Bform(self):
|
|
self.assertTrue(derivTest('b', 'eyi'))
|
|
def test_Jvec_ezi_Eform(self):
|
|
self.assertTrue(derivTest('e', 'ezi'))
|
|
def test_Jvec_ezi_Bform(self):
|
|
self.assertTrue(derivTest('b', 'ezi'))
|
|
|
|
def test_Jvec_bxr_Eform(self):
|
|
self.assertTrue(derivTest('e', 'bxr'))
|
|
def test_Jvec_bxr_Bform(self):
|
|
self.assertTrue(derivTest('b', 'bxr'))
|
|
def test_Jvec_byr_Eform(self):
|
|
self.assertTrue(derivTest('e', 'byr'))
|
|
def test_Jvec_byr_Bform(self):
|
|
self.assertTrue(derivTest('b', 'byr'))
|
|
def test_Jvec_bzr_Eform(self):
|
|
self.assertTrue(derivTest('e', 'bzr'))
|
|
def test_Jvec_bzr_Bform(self):
|
|
self.assertTrue(derivTest('b', 'bzr'))
|
|
def test_Jvec_bxi_Eform(self):
|
|
self.assertTrue(derivTest('e', 'bxi'))
|
|
def test_Jvec_bxi_Bform(self):
|
|
self.assertTrue(derivTest('b', 'bxi'))
|
|
def test_Jvec_byi_Eform(self):
|
|
self.assertTrue(derivTest('e', 'byi'))
|
|
def test_Jvec_byi_Bform(self):
|
|
self.assertTrue(derivTest('b', 'byi'))
|
|
def test_Jvec_bzi_Eform(self):
|
|
self.assertTrue(derivTest('e', 'bzi'))
|
|
def test_Jvec_bzi_Bform(self):
|
|
self.assertTrue(derivTest('b', 'bzi'))
|
|
|
|
|
|
|
|
def test_Jtvec_adjointTest_exr_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'exr'))
|
|
def test_Jtvec_adjointTest_exr_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'exr'))
|
|
def test_Jtvec_adjointTest_eyr_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'eyr'))
|
|
def test_Jtvec_adjointTest_eyr_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'eyr'))
|
|
def test_Jtvec_adjointTest_ezr_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'ezr'))
|
|
def test_Jtvec_adjointTest_ezr_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'ezr'))
|
|
def test_Jtvec_adjointTest_exi_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'exi'))
|
|
def test_Jtvec_adjointTest_exi_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'exi'))
|
|
def test_Jtvec_adjointTest_eyi_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'eyi'))
|
|
def test_Jtvec_adjointTest_eyi_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'eyi'))
|
|
def test_Jtvec_adjointTest_ezi_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'ezi'))
|
|
def test_Jtvec_adjointTest_ezi_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'ezi'))
|
|
|
|
def test_Jtvec_adjointTest_bxr_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'bxr'))
|
|
def test_Jtvec_adjointTest_bxr_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'bxr'))
|
|
def test_Jtvec_adjointTest_byr_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'byr'))
|
|
def test_Jtvec_adjointTest_byr_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'byr'))
|
|
def test_Jtvec_adjointTest_bzr_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'bzr'))
|
|
def test_Jtvec_adjointTest_bzr_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'bzr'))
|
|
def test_Jtvec_adjointTest_bxi_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'bxi'))
|
|
def test_Jtvec_adjointTest_bxi_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'bxi'))
|
|
def test_Jtvec_adjointTest_byi_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'byi'))
|
|
def test_Jtvec_adjointTest_byi_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'byi'))
|
|
def test_Jtvec_adjointTest_bzi_Eform(self):
|
|
self.assertTrue(adjointTest('e', 'bzi'))
|
|
def test_Jtvec_adjointTest_bzi_Bform(self):
|
|
self.assertTrue(adjointTest('b', 'bzi'))
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|