wassname/simpeg
1
0
Fork 0
mirror of https://github.com/wassname/simpeg.git synced 2026-09-09 11:34:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

start of getting any field from any formulation

Browse Source
This commit is contained in:
Lindsey
2015-12-08 18:11:01 -08:00
parent 7c0be4ec50
commit 884d27b541
6 changed files with 235 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/em/fdem/forward/test_FDEM_forward.py
+61 -40
View File
@@ -7,26 +7,28 @@ from SimPEG.EM.Utils.testingUtils import getFDEMProblem
testEB = True
testHJ = True
testEJ = False
verbose = False
TOL = 1e-5
TOLEBHJ = 1e-5
TOLEJHB = 1e-1
FLR = 1e-20 # "zero", so if residual below this --> pass regardless of order
CONDUCTIVITY = 1e1
MU = mu_0
freq = 1e-1
addrandoms = True
addrandoms = False
SrcList = ['RawVec', 'MagDipole_Bfield', 'MagDipole', 'CircularLoop']
def crossCheckTest(fdemType, comp):
def crossCheckTest(fdemType1, fdemType2, comp, TOL=TOLEBHJ):
l2norm = lambda r: np.sqrt(r.dot(r))
prb1 = getFDEMProblem(fdemType, comp, SrcList, freq, verbose)
prb1 = getFDEMProblem(fdemType1, comp, SrcList, freq, verbose)
mesh = prb1.mesh
print 'Cross Checking Forward: %s formulation - %s' % (fdemType, comp)
print 'Cross Checking Forward: %s formulation - %s' % (fdemType1, comp)
m = np.log(np.ones(mesh.nC)*CONDUCTIVITY)
mu = np.log(np.ones(mesh.nC)*MU)
@@ -42,16 +44,8 @@ def crossCheckTest(fdemType, comp):
if verbose:
print ' Problem 1 solved'
if fdemType == 'e':
prb2 = getFDEMProblem('b', comp, SrcList, freq, verbose)
elif fdemType == 'b':
prb2 = getFDEMProblem('e', comp, SrcList, freq, verbose)
elif fdemType == 'j':
prb2 = getFDEMProblem('h', comp, SrcList, freq, verbose)
elif fdemType == 'h':
prb2 = getFDEMProblem('j', comp, SrcList, freq, verbose)
else:
raise NotImplementedError()
prb2 = getFDEMProblem(fdemType2, comp, SrcList, freq, verbose)
# prb2.mu = mu
survey2 = prb2.survey
@@ -71,57 +65,84 @@ def crossCheckTest(fdemType, comp):
class FDEM_CrossCheck(unittest.TestCase):
if testEB:
def test_EB_CrossCheck_exr_Eform(self):
self.assertTrue(crossCheckTest('e', 'exr'))
self.assertTrue(crossCheckTest('e', 'b', 'exr'))
def test_EB_CrossCheck_eyr_Eform(self):
self.assertTrue(crossCheckTest('e', 'eyr'))
self.assertTrue(crossCheckTest('e', 'b', 'eyr'))
def test_EB_CrossCheck_ezr_Eform(self):
self.assertTrue(crossCheckTest('e', 'ezr'))
self.assertTrue(crossCheckTest('e', 'b', 'ezr'))
def test_EB_CrossCheck_exi_Eform(self):
self.assertTrue(crossCheckTest('e', 'exi'))
self.assertTrue(crossCheckTest('e', 'b', 'exi'))
def test_EB_CrossCheck_eyi_Eform(self):
self.assertTrue(crossCheckTest('e', 'eyi'))
self.assertTrue(crossCheckTest('e', 'b', 'eyi'))
def test_EB_CrossCheck_ezi_Eform(self):
self.assertTrue(crossCheckTest('e', 'ezi'))
self.assertTrue(crossCheckTest('e', 'b', 'ezi'))
def test_EB_CrossCheck_bxr_Eform(self):
self.assertTrue(crossCheckTest('e', 'bxr'))
self.assertTrue(crossCheckTest('e', 'b', 'bxr'))
def test_EB_CrossCheck_byr_Eform(self):
self.assertTrue(crossCheckTest('e', 'byr'))
self.assertTrue(crossCheckTest('e', 'b', 'byr'))
def test_EB_CrossCheck_bzr_Eform(self):
self.assertTrue(crossCheckTest('e', 'bzr'))
self.assertTrue(crossCheckTest('e', 'b', 'bzr'))
def test_EB_CrossCheck_bxi_Eform(self):
self.assertTrue(crossCheckTest('e', 'bxi'))
self.assertTrue(crossCheckTest('e', 'b', 'bxi'))
def test_EB_CrossCheck_byi_Eform(self):
self.assertTrue(crossCheckTest('e', 'byi'))
self.assertTrue(crossCheckTest('e', 'b', 'byi'))
def test_EB_CrossCheck_bzi_Eform(self):
self.assertTrue(crossCheckTest('e', 'bzi'))
self.assertTrue(crossCheckTest('e', 'b', 'bzi'))
if testHJ:
def test_HJ_CrossCheck_jxr_Jform(self):
self.assertTrue(crossCheckTest('j', 'jxr'))
self.assertTrue(crossCheckTest('j', 'h', 'jxr'))
def test_HJ_CrossCheck_jyr_Jform(self):
self.assertTrue(crossCheckTest('j', 'jyr'))
self.assertTrue(crossCheckTest('j', 'h', 'jyr'))
def test_HJ_CrossCheck_jzr_Jform(self):
self.assertTrue(crossCheckTest('j', 'jzr'))
self.assertTrue(crossCheckTest('j', 'h', 'jzr'))
def test_HJ_CrossCheck_jxi_Jform(self):
self.assertTrue(crossCheckTest('j', 'jxi'))
self.assertTrue(crossCheckTest('j', 'h', 'jxi'))
def test_HJ_CrossCheck_jyi_Jform(self):
self.assertTrue(crossCheckTest('j', 'jyi'))
self.assertTrue(crossCheckTest('j', 'h', 'jyi'))
def test_HJ_CrossCheck_jzi_Jform(self):
self.assertTrue(crossCheckTest('j', 'jzi'))
self.assertTrue(crossCheckTest('j', 'h', 'jzi'))
def test_HJ_CrossCheck_hxr_Jform(self):
self.assertTrue(crossCheckTest('j', 'hxr'))
self.assertTrue(crossCheckTest('j', 'h', 'hxr'))
def test_HJ_CrossCheck_hyr_Jform(self):
self.assertTrue(crossCheckTest('j', 'hyr'))
self.assertTrue(crossCheckTest('j', 'h', 'hyr'))
def test_HJ_CrossCheck_hzr_Jform(self):
self.assertTrue(crossCheckTest('j', 'hzr'))
self.assertTrue(crossCheckTest('j', 'h', 'hzr'))
def test_HJ_CrossCheck_hxi_Jform(self):
self.assertTrue(crossCheckTest('j', 'hxi'))
self.assertTrue(crossCheckTest('j', 'h', 'hxi'))
def test_HJ_CrossCheck_hyi_Jform(self):
self.assertTrue(crossCheckTest('j', 'hyi'))
self.assertTrue(crossCheckTest('j', 'h', 'hyi'))
def test_HJ_CrossCheck_hzi_Jform(self):
self.assertTrue(crossCheckTest('j', 'hzi'))
self.assertTrue(crossCheckTest('j', 'h', 'hzi'))
if testEJ:
# def test_EJ_CrossCheck_jxr_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'jxr'))
# def test_EJ_CrossCheck_jyr_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'jyr'))
# def test_EJ_CrossCheck_jzr_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'jzr'))
# def test_EJ_CrossCheck_jxi_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'jxi'))
# def test_EJ_CrossCheck_jyi_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'jyi'))
# def test_EJ_CrossCheck_jzi_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'jzi'))
def test_EJ_CrossCheck_jxr_Jform(self):
self.assertTrue(crossCheckTest('e', 'j', 'exr', TOL=TOLEJHB))
# def test_EJ_CrossCheck_jyr_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'eyr'))
# def test_EJ_CrossCheck_jzr_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'ezr'))
# def test_EJ_CrossCheck_jxi_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'exi'))
# def test_EJ_CrossCheck_jyi_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'eyi'))
# def test_EJ_CrossCheck_jzi_Jform(self):
# self.assertTrue(crossCheckTest('e', 'j', 'ezi'))
if __name__ == '__main__':
unittest.main()