mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-27 18:25:42 +08:00
remove tdem refactor test from this pr
This commit is contained in:
Lindsey Heagy
@@ -1,221 +0,0 @@
|
||||
import unittest
|
||||
from SimPEG import *
|
||||
from SimPEG import EM
|
||||
|
||||
plotIt = False
|
||||
|
||||
testDeriv = True
|
||||
testAdjoint = True
|
||||
|
||||
TOL = 1e-5
|
||||
|
||||
def setUp(prbtype='b', rxcomp='bz'):
|
||||
cs = 5.
|
||||
ncx = 20
|
||||
ncy = 15
|
||||
npad = 20
|
||||
hx = [(cs,ncx), (cs,npad,1.3)]
|
||||
hy = [(cs,npad,-1.3), (cs,ncy), (cs,npad,1.3)]
|
||||
mesh = Mesh.CylMesh([hx,1,hy], '00C')
|
||||
#
|
||||
active = mesh.vectorCCz<0.
|
||||
activeMap = Maps.InjectActiveCells(mesh, active, np.log(1e-8), nC=mesh.nCz)
|
||||
mapping = Maps.ExpMap(mesh) * Maps.SurjectVertical1D(mesh) * activeMap
|
||||
|
||||
rxOffset = 10.
|
||||
rx = EM.TDEM.Rx(np.array([[rxOffset, 0., -1e-2]]), np.logspace(-4,-3, 20), rxcomp) #,]
|
||||
src = EM.TDEM.Src.MagDipole([rx], loc=np.array([0., 0., 0.]))
|
||||
|
||||
survey = EM.TDEM.Survey([src])
|
||||
|
||||
if prbtype == 'b':
|
||||
prb = EM.TDEM.Problem_b(mesh, mapping=mapping)
|
||||
elif prbtype == 'e':
|
||||
prb = EM.TDEM.Problem_e(mesh, mapping=mapping)
|
||||
|
||||
prb.timeSteps = [(1e-05, 10), (5e-05, 10), (2.5e-4, 10)]
|
||||
# prb.timeSteps = [(1e-05, 10), (1e-05, 50), (1e-05, 50) ] #, (2.5e-4, 10)]
|
||||
|
||||
try:
|
||||
from pymatsolver import MumpsSolver
|
||||
prb.Solver = MumpsSolver
|
||||
except ImportError, e:
|
||||
prb.Solver = SolverLU
|
||||
|
||||
m = np.log(1e-1)*np.ones(prb.mapping.nP) + 1e-2*np.random.randn(prb.mapping.nP)
|
||||
|
||||
prb.pair(survey)
|
||||
mesh = mesh
|
||||
|
||||
return prb, m, mesh
|
||||
|
||||
|
||||
class TDEM_DerivTests(unittest.TestCase):
|
||||
|
||||
# ====== TEST A ========== #
|
||||
|
||||
def AderivTest(self, prbtype):
|
||||
prb, m0, mesh = setUp(prbtype)
|
||||
tInd = 2
|
||||
if prbtype == 'b':
|
||||
nu = mesh.nF
|
||||
elif prbtype == 'e':
|
||||
nu = mesh.nE
|
||||
v = np.random.rand(nu)
|
||||
|
||||
def AderivFun(m):
|
||||
prb.curModel = m
|
||||
A = prb.getAdiag(tInd)
|
||||
Av = A*v
|
||||
prb.curModel = m0
|
||||
ADeriv_dm = lambda dm: prb.getAdiagDeriv(tInd, v, dm)
|
||||
|
||||
return Av, ADeriv_dm
|
||||
|
||||
print '\n Testing ADeriv %s'%(prbtype)
|
||||
Tests.checkDerivative(AderivFun, m0, plotIt=False, num=4, eps=1e-20)
|
||||
|
||||
def A_adjointTest(self,prbtype):
|
||||
prb, m0, mesh = setUp(prbtype)
|
||||
tInd = 2
|
||||
|
||||
print '\n Testing A_adjoint'
|
||||
m = np.random.rand(prb.mapping.nP)
|
||||
if prbtype == 'b':
|
||||
nu = prb.mesh.nF
|
||||
elif prbtype == 'e':
|
||||
nu = prb.mesh.nE
|
||||
|
||||
v = np.random.rand(nu)
|
||||
u = np.random.rand(nu)
|
||||
prb.curModel = m0
|
||||
|
||||
tInd = 2 # not actually used
|
||||
V1 = v.dot(prb.getAdiagDeriv(tInd, u, m))
|
||||
V2 = m.dot(prb.getAdiagDeriv(tInd, u, v, adjoint=True))
|
||||
passed = np.abs(V1-V2) < TOL * (np.abs(V1) + np.abs(V2))/2.
|
||||
print 'AdjointTest %s'%(prbtype), V1, V2, passed
|
||||
self.assertTrue(passed)
|
||||
|
||||
def test_Aderiv_b(self):
|
||||
self.AderivTest('b')
|
||||
def test_Aderiv_e(self):
|
||||
self.AderivTest('e')
|
||||
|
||||
def test_Aadjoint_b(self):
|
||||
self.A_adjointTest('b')
|
||||
def test_Aadjoint_e(self):
|
||||
self.A_adjointTest('e')
|
||||
|
||||
# ====== TEST Fields Deriv Pieces ========== #
|
||||
|
||||
def test_eDeriv_m_adjoint(self):
|
||||
prb, m0, mesh = setUp()
|
||||
tInd = 0
|
||||
|
||||
v = np.random.rand(mesh.nF)
|
||||
|
||||
print '\n Testing eDeriv_m Adjoint'
|
||||
|
||||
prb, m0, mesh = setUp()
|
||||
f = prb.fields(m0)
|
||||
|
||||
m = np.random.rand(prb.mapping.nP)
|
||||
e = np.random.randn(prb.mesh.nE)
|
||||
V1 = e.dot(f._eDeriv_m(1, prb.survey.srcList[0], m))
|
||||
V2 = m.dot(f._eDeriv_m(1, prb.survey.srcList[0], e, adjoint=True))
|
||||
tol = TOL * (np.abs(V1) + np.abs(V2)) / 2.
|
||||
passed = np.abs(V1-V2) < tol
|
||||
|
||||
print ' ', V1, V2, np.abs(V1-V2), tol, passed
|
||||
self.assertTrue(passed)
|
||||
|
||||
def test_eDeriv_u_adjoint(self):
|
||||
print '\n Testing eDeriv_u Adjoint'
|
||||
|
||||
prb, m0, mesh = setUp()
|
||||
f = prb.fields(m0)
|
||||
|
||||
b = np.random.rand(prb.mesh.nF)
|
||||
e = np.random.randn(prb.mesh.nE)
|
||||
V1 = e.dot(f._eDeriv_u(1, prb.survey.srcList[0], b))
|
||||
V2 = b.dot(f._eDeriv_u(1, prb.survey.srcList[0], e, adjoint=True))
|
||||
tol = TOL * (np.abs(V1) + np.abs(V2)) / 2.
|
||||
passed = np.abs(V1-V2) < tol
|
||||
|
||||
print ' ', V1, V2, np.abs(V1-V2), tol, passed
|
||||
self.assertTrue(passed)
|
||||
|
||||
|
||||
# ====== TEST Jvec ========== #
|
||||
|
||||
if testDeriv:
|
||||
|
||||
def JvecTest(self, prbtype, rxcomp):
|
||||
prb, m, mesh = setUp(prbtype, rxcomp)
|
||||
|
||||
derChk = lambda m: [prb.survey.dpred(m), lambda mx: prb.Jvec(m, mx)]
|
||||
print '\n'
|
||||
print 'test_Jvec_%s_%s' %(prbtype, rxcomp)
|
||||
Tests.checkDerivative(derChk, m, plotIt=False, num=2, eps=1e-20)
|
||||
|
||||
# def test_Jvec_b_bx(self):
|
||||
# self.JvecTest('b','bx')
|
||||
|
||||
# def test_Jvec_b_bz(self):
|
||||
# self.JvecTest('b','bz')
|
||||
|
||||
# def test_Jvec_b_dbxdt(self):
|
||||
# self.JvecTest('b','dbxdt')
|
||||
|
||||
# def test_Jvec_b_dbzdt(self):
|
||||
# self.JvecTest('b','dbzdt')
|
||||
|
||||
# def test_Jvec_b_ey(self):
|
||||
# self.JvecTest('b','ey')
|
||||
|
||||
def test_Jvec_e_ey(self):
|
||||
self.JvecTest('e','ey')
|
||||
|
||||
|
||||
# ====== TEST Jtvec ========== #
|
||||
|
||||
if testAdjoint:
|
||||
|
||||
def JvecVsJtvecTest(self, prbtype='b', rxcomp='bz'):
|
||||
|
||||
print '\nAdjoint Testing Jvec, Jtvec %s' %(rxcomp)
|
||||
|
||||
prb, m0, mesh = setUp(prbtype, rxcomp)
|
||||
m = np.random.rand(prb.mapping.nP)
|
||||
d = np.random.randn(prb.survey.nD)
|
||||
V1 = d.dot(prb.Jvec(m0, m))
|
||||
V2 = m.dot(prb.Jtvec(m0, d))
|
||||
tol = TOL * (np.abs(V1) + np.abs(V2)) / 2.
|
||||
passed = np.abs(V1-V2) < tol
|
||||
|
||||
print ' ', V1, V2, np.abs(V1-V2), tol, passed
|
||||
self.assertTrue(passed)
|
||||
|
||||
# def test_Jvec_adjoint_b_bx(self):
|
||||
# self.JvecVsJtvecTest('b', 'bx')
|
||||
|
||||
# def test_Jvec_adjoint_b_bz(self):
|
||||
# self.JvecVsJtvecTest('b', 'bz')
|
||||
|
||||
# def test_Jvec_adjoint_b_dbxdt(self):
|
||||
# self.JvecVsJtvecTest('b', 'bx')
|
||||
|
||||
# def test_Jvec_adjoint_b_dbzdt(self):
|
||||
# self.JvecVsJtvecTest('b', 'bz')
|
||||
|
||||
# def test_Jvec_adjoint_b_ey(self):
|
||||
# self.JvecVsJtvecTest('b', 'ey')
|
||||
|
||||
def test_Jvec_adjoint_e_ey(self):
|
||||
self.JvecVsJtvecTest('e', 'ey')
|
||||
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user