mirror of
https://github.com/wassname/simpeg.git
synced 2026-07-19 11:28:00 +08:00
128 lines
4.5 KiB
Python
128 lines
4.5 KiB
Python
import unittest
|
|
from SimPEG import *
|
|
import SimPEG.EM.Static.DC as DC
|
|
|
|
|
|
class DCProblem_2DTestsCC(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
|
|
cs = 12.5
|
|
hx = [(cs,7, -1.3),(cs,61),(cs,7, 1.3)]
|
|
hy = [(cs,7, -1.3),(cs,20)]
|
|
mesh = Mesh.TensorMesh([hx, hy],x0="CN")
|
|
x = np.linspace(-135, 250., 20)
|
|
M = Utils.ndgrid(x-12.5, np.r_[0.])
|
|
N = Utils.ndgrid(x+12.5, np.r_[0.])
|
|
A0loc = np.r_[-150, 0.]
|
|
A1loc = np.r_[-130, 0.]
|
|
rxloc = [np.c_[M, np.zeros(20)], np.c_[N, np.zeros(20)]]
|
|
rx = DC.Rx.Dipole_ky(M, N)
|
|
src0 = DC.Src.Pole([rx], A0loc)
|
|
src1 = DC.Src.Pole([rx], A1loc)
|
|
survey = DC.Survey_ky([src0, src1])
|
|
problem = DC.Problem2D_CC(mesh, mapping=[('rho', Maps.IdentityMap(mesh))])
|
|
problem.pair(survey)
|
|
|
|
mSynth = np.ones(mesh.nC)*1.
|
|
survey.makeSyntheticData(mSynth)
|
|
|
|
# Now set up the problem to do some minimization
|
|
dmis = DataMisfit.l2_DataMisfit(survey)
|
|
reg = Regularization.Tikhonov(mesh)
|
|
opt = Optimization.InexactGaussNewton(maxIterLS=20, maxIter=10, tolF=1e-6, tolX=1e-6, tolG=1e-6, maxIterCG=6)
|
|
invProb = InvProblem.BaseInvProblem(dmis, reg, opt, beta=1e0)
|
|
inv = Inversion.BaseInversion(invProb)
|
|
|
|
self.inv = inv
|
|
self.reg = reg
|
|
self.p = problem
|
|
self.mesh = mesh
|
|
self.m0 = mSynth
|
|
self.survey = survey
|
|
self.dmis = dmis
|
|
|
|
def test_misfit(self):
|
|
derChk = lambda m: [self.survey.dpred(m), lambda mx: self.p.Jvec(self.m0, mx)]
|
|
passed = Tests.checkDerivative(derChk, self.m0, plotIt=False, num=3)
|
|
self.assertTrue(passed)
|
|
|
|
def test_adjoint(self):
|
|
# Adjoint Test
|
|
u = np.random.rand(self.mesh.nC*self.survey.nSrc)
|
|
v = np.random.rand(self.mesh.nC)
|
|
w = np.random.rand(self.survey.dobs.shape[0])
|
|
wtJv = w.dot(self.p.Jvec(self.m0, v))
|
|
vtJtw = v.dot(self.p.Jtvec(self.m0, w))
|
|
passed = np.abs(wtJv - vtJtw) < 1e-10
|
|
print 'Adjoint Test', np.abs(wtJv - vtJtw), passed
|
|
self.assertTrue(passed)
|
|
|
|
def test_dataObj(self):
|
|
derChk = lambda m: [self.dmis.eval(m), self.dmis.evalDeriv(m)]
|
|
passed = Tests.checkDerivative(derChk, self.m0, plotIt=False, num=3)
|
|
self.assertTrue(passed)
|
|
|
|
# class DCProblemTestsN(unittest.TestCase):
|
|
|
|
# def setUp(self):
|
|
|
|
# aSpacing=2.5
|
|
# nElecs=10
|
|
|
|
# surveySize = nElecs*aSpacing - aSpacing
|
|
# cs = surveySize/nElecs/4
|
|
|
|
# mesh = Mesh.TensorMesh([
|
|
# [(cs,10, -1.3),(cs,surveySize/cs),(cs,10, 1.3)],
|
|
# [(cs,3, -1.3),(cs,3,1.3)],
|
|
# # [(cs,5, -1.3),(cs,10)]
|
|
# ],'CN')
|
|
|
|
# srcList = DC.Utils.WennerSrcList(nElecs, aSpacing, in2D=True)
|
|
# survey = DC.Survey(srcList)
|
|
# problem = DC.Problem3D_N(mesh, mapping=[('rho', Maps.IdentityMap(mesh))])
|
|
# problem.pair(survey)
|
|
|
|
# mSynth = np.ones(mesh.nC)
|
|
# survey.makeSyntheticData(mSynth)
|
|
|
|
# # Now set up the problem to do some minimization
|
|
# dmis = DataMisfit.l2_DataMisfit(survey)
|
|
# reg = Regularization.Tikhonov(mesh)
|
|
# opt = Optimization.InexactGaussNewton(maxIterLS=20, maxIter=10, tolF=1e-6, tolX=1e-6, tolG=1e-6, maxIterCG=6)
|
|
# invProb = InvProblem.BaseInvProblem(dmis, reg, opt, beta=1e4)
|
|
# inv = Inversion.BaseInversion(invProb)
|
|
|
|
# self.inv = inv
|
|
# self.reg = reg
|
|
# self.p = problem
|
|
# self.mesh = mesh
|
|
# self.m0 = mSynth
|
|
# self.survey = survey
|
|
# self.dmis = dmis
|
|
|
|
# def test_misfit(self):
|
|
# derChk = lambda m: [self.survey.dpred(m), lambda mx: self.p.Jvec(self.m0, mx)]
|
|
# passed = Tests.checkDerivative(derChk, self.m0, plotIt=False)
|
|
# self.assertTrue(passed)
|
|
|
|
# def test_adjoint(self):
|
|
# # Adjoint Test
|
|
# u = np.random.rand(self.mesh.nC*self.survey.nSrc)
|
|
# v = np.random.rand(self.mesh.nC)
|
|
# w = np.random.rand(self.survey.dobs.shape[0])
|
|
# wtJv = w.dot(self.p.Jvec(self.m0, v))
|
|
# vtJtw = v.dot(self.p.Jtvec(self.m0, w))
|
|
# passed = np.abs(wtJv - vtJtw) < 1e-8
|
|
# print 'Adjoint Test', np.abs(wtJv - vtJtw), passed
|
|
# self.assertTrue(passed)
|
|
|
|
# def test_dataObj(self):
|
|
# derChk = lambda m: [self.dmis.eval(m), self.dmis.evalDeriv(m)]
|
|
# passed = Tests.checkDerivative(derChk, self.m0, plotIt=False)
|
|
# self.assertTrue(passed)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|