From 4b87b4b45f2c6b32e2f39422f63bfd07ad6a98d8 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 13:12:15 -0800 Subject: [PATCH 1/9] attempt to split up travis testing for em --- tests/em/fdem/__init__.py | 11 +++++++++++ tests/em/{ => fdem}/test_FDEM.py | 0 tests/em/{ => fdem}/test_FDEMCasing.py | 0 tests/em/{ => fdem}/test_FDEM_analytics.py | 0 tests/em/tdem/__init__.py | 11 +++++++++++ tests/em/{ => tdem}/test_TDEM_b_DerivAdjoint.py | 0 .../{ => tdem}/test_TDEM_b_MultiSrc_DerivAdjoint.py | 0 tests/em/{ => tdem}/test_TDEM_combos.py | 0 tests/em/{ => tdem}/test_TDEM_forward_Analytic.py | 0 9 files changed, 22 insertions(+) create mode 100644 tests/em/fdem/__init__.py rename tests/em/{ => fdem}/test_FDEM.py (100%) rename tests/em/{ => fdem}/test_FDEMCasing.py (100%) rename tests/em/{ => fdem}/test_FDEM_analytics.py (100%) create mode 100644 tests/em/tdem/__init__.py rename tests/em/{ => tdem}/test_TDEM_b_DerivAdjoint.py (100%) rename tests/em/{ => tdem}/test_TDEM_b_MultiSrc_DerivAdjoint.py (100%) rename tests/em/{ => tdem}/test_TDEM_combos.py (100%) rename tests/em/{ => tdem}/test_TDEM_forward_Analytic.py (100%) diff --git a/tests/em/fdem/__init__.py b/tests/em/fdem/__init__.py new file mode 100644 index 00000000..38d84328 --- /dev/null +++ b/tests/em/fdem/__init__.py @@ -0,0 +1,11 @@ +if __name__ == '__main__': + import os + import glob + import unittest + test_file_strings = glob.glob('test_*.py') + module_strings = [str[0:len(str)-3] for str in test_file_strings] + suites = [unittest.defaultTestLoader.loadTestsFromName(str) for str + in module_strings] + testSuite = unittest.TestSuite(suites) + + unittest.TextTestRunner(verbosity=2).run(testSuite) diff --git a/tests/em/test_FDEM.py b/tests/em/fdem/test_FDEM.py similarity index 100% rename from tests/em/test_FDEM.py rename to tests/em/fdem/test_FDEM.py diff --git a/tests/em/test_FDEMCasing.py b/tests/em/fdem/test_FDEMCasing.py similarity index 100% rename from tests/em/test_FDEMCasing.py rename to tests/em/fdem/test_FDEMCasing.py diff --git a/tests/em/test_FDEM_analytics.py b/tests/em/fdem/test_FDEM_analytics.py similarity index 100% rename from tests/em/test_FDEM_analytics.py rename to tests/em/fdem/test_FDEM_analytics.py diff --git a/tests/em/tdem/__init__.py b/tests/em/tdem/__init__.py new file mode 100644 index 00000000..38d84328 --- /dev/null +++ b/tests/em/tdem/__init__.py @@ -0,0 +1,11 @@ +if __name__ == '__main__': + import os + import glob + import unittest + test_file_strings = glob.glob('test_*.py') + module_strings = [str[0:len(str)-3] for str in test_file_strings] + suites = [unittest.defaultTestLoader.loadTestsFromName(str) for str + in module_strings] + testSuite = unittest.TestSuite(suites) + + unittest.TextTestRunner(verbosity=2).run(testSuite) diff --git a/tests/em/test_TDEM_b_DerivAdjoint.py b/tests/em/tdem/test_TDEM_b_DerivAdjoint.py similarity index 100% rename from tests/em/test_TDEM_b_DerivAdjoint.py rename to tests/em/tdem/test_TDEM_b_DerivAdjoint.py diff --git a/tests/em/test_TDEM_b_MultiSrc_DerivAdjoint.py b/tests/em/tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py similarity index 100% rename from tests/em/test_TDEM_b_MultiSrc_DerivAdjoint.py rename to tests/em/tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py diff --git a/tests/em/test_TDEM_combos.py b/tests/em/tdem/test_TDEM_combos.py similarity index 100% rename from tests/em/test_TDEM_combos.py rename to tests/em/tdem/test_TDEM_combos.py diff --git a/tests/em/test_TDEM_forward_Analytic.py b/tests/em/tdem/test_TDEM_forward_Analytic.py similarity index 100% rename from tests/em/test_TDEM_forward_Analytic.py rename to tests/em/tdem/test_TDEM_forward_Analytic.py From 10bd87ac023e4516de88b030b064210c6b948a75 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 13:14:44 -0800 Subject: [PATCH 2/9] move tdem, fdem tests up a directory --- tests/{em => em_examples}/__init__.py | 0 tests/{em => em_examples}/test_Examples.py | 0 tests/{em/fdem => em_fdem}/__init__.py | 0 tests/{em/fdem => em_fdem}/test_FDEM.py | 0 tests/{em/fdem => em_fdem}/test_FDEMCasing.py | 0 tests/{em/fdem => em_fdem}/test_FDEM_analytics.py | 0 tests/{em/tdem => em_tdem}/__init__.py | 0 tests/{em/tdem => em_tdem}/test_TDEM_b_DerivAdjoint.py | 0 tests/{em/tdem => em_tdem}/test_TDEM_b_MultiSrc_DerivAdjoint.py | 0 tests/{em/tdem => em_tdem}/test_TDEM_combos.py | 0 tests/{em/tdem => em_tdem}/test_TDEM_forward_Analytic.py | 0 11 files changed, 0 insertions(+), 0 deletions(-) rename tests/{em => em_examples}/__init__.py (100%) rename tests/{em => em_examples}/test_Examples.py (100%) rename tests/{em/fdem => em_fdem}/__init__.py (100%) rename tests/{em/fdem => em_fdem}/test_FDEM.py (100%) rename tests/{em/fdem => em_fdem}/test_FDEMCasing.py (100%) rename tests/{em/fdem => em_fdem}/test_FDEM_analytics.py (100%) rename tests/{em/tdem => em_tdem}/__init__.py (100%) rename tests/{em/tdem => em_tdem}/test_TDEM_b_DerivAdjoint.py (100%) rename tests/{em/tdem => em_tdem}/test_TDEM_b_MultiSrc_DerivAdjoint.py (100%) rename tests/{em/tdem => em_tdem}/test_TDEM_combos.py (100%) rename tests/{em/tdem => em_tdem}/test_TDEM_forward_Analytic.py (100%) diff --git a/tests/em/__init__.py b/tests/em_examples/__init__.py similarity index 100% rename from tests/em/__init__.py rename to tests/em_examples/__init__.py diff --git a/tests/em/test_Examples.py b/tests/em_examples/test_Examples.py similarity index 100% rename from tests/em/test_Examples.py rename to tests/em_examples/test_Examples.py diff --git a/tests/em/fdem/__init__.py b/tests/em_fdem/__init__.py similarity index 100% rename from tests/em/fdem/__init__.py rename to tests/em_fdem/__init__.py diff --git a/tests/em/fdem/test_FDEM.py b/tests/em_fdem/test_FDEM.py similarity index 100% rename from tests/em/fdem/test_FDEM.py rename to tests/em_fdem/test_FDEM.py diff --git a/tests/em/fdem/test_FDEMCasing.py b/tests/em_fdem/test_FDEMCasing.py similarity index 100% rename from tests/em/fdem/test_FDEMCasing.py rename to tests/em_fdem/test_FDEMCasing.py diff --git a/tests/em/fdem/test_FDEM_analytics.py b/tests/em_fdem/test_FDEM_analytics.py similarity index 100% rename from tests/em/fdem/test_FDEM_analytics.py rename to tests/em_fdem/test_FDEM_analytics.py diff --git a/tests/em/tdem/__init__.py b/tests/em_tdem/__init__.py similarity index 100% rename from tests/em/tdem/__init__.py rename to tests/em_tdem/__init__.py diff --git a/tests/em/tdem/test_TDEM_b_DerivAdjoint.py b/tests/em_tdem/test_TDEM_b_DerivAdjoint.py similarity index 100% rename from tests/em/tdem/test_TDEM_b_DerivAdjoint.py rename to tests/em_tdem/test_TDEM_b_DerivAdjoint.py diff --git a/tests/em/tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py b/tests/em_tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py similarity index 100% rename from tests/em/tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py rename to tests/em_tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py diff --git a/tests/em/tdem/test_TDEM_combos.py b/tests/em_tdem/test_TDEM_combos.py similarity index 100% rename from tests/em/tdem/test_TDEM_combos.py rename to tests/em_tdem/test_TDEM_combos.py diff --git a/tests/em/tdem/test_TDEM_forward_Analytic.py b/tests/em_tdem/test_TDEM_forward_Analytic.py similarity index 100% rename from tests/em/tdem/test_TDEM_forward_Analytic.py rename to tests/em_tdem/test_TDEM_forward_Analytic.py From 63a2ace3db4cbe72a47ce0f9a0ba0a2c175f437b Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 13:17:04 -0800 Subject: [PATCH 3/9] em directory with tests, faded, them and travis update --- .travis.yml | 4 +++- tests/{em_examples => em/examples}/__init__.py | 0 tests/{em_examples => em/examples}/test_Examples.py | 0 tests/{em_fdem => em/fdem}/__init__.py | 0 tests/{em_fdem => em/fdem}/test_FDEM.py | 0 tests/{em_fdem => em/fdem}/test_FDEMCasing.py | 0 tests/{em_fdem => em/fdem}/test_FDEM_analytics.py | 0 tests/{em_tdem => em/tdem}/__init__.py | 0 tests/{em_tdem => em/tdem}/test_TDEM_b_DerivAdjoint.py | 0 .../{em_tdem => em/tdem}/test_TDEM_b_MultiSrc_DerivAdjoint.py | 0 tests/{em_tdem => em/tdem}/test_TDEM_combos.py | 0 tests/{em_tdem => em/tdem}/test_TDEM_forward_Analytic.py | 0 12 files changed, 3 insertions(+), 1 deletion(-) rename tests/{em_examples => em/examples}/__init__.py (100%) rename tests/{em_examples => em/examples}/test_Examples.py (100%) rename tests/{em_fdem => em/fdem}/__init__.py (100%) rename tests/{em_fdem => em/fdem}/test_FDEM.py (100%) rename tests/{em_fdem => em/fdem}/test_FDEMCasing.py (100%) rename tests/{em_fdem => em/fdem}/test_FDEM_analytics.py (100%) rename tests/{em_tdem => em/tdem}/__init__.py (100%) rename tests/{em_tdem => em/tdem}/test_TDEM_b_DerivAdjoint.py (100%) rename tests/{em_tdem => em/tdem}/test_TDEM_b_MultiSrc_DerivAdjoint.py (100%) rename tests/{em_tdem => em/tdem}/test_TDEM_combos.py (100%) rename tests/{em_tdem => em/tdem}/test_TDEM_forward_Analytic.py (100%) diff --git a/.travis.yml b/.travis.yml index 0705506f..c4bd33ef 100644 --- a/.travis.yml +++ b/.travis.yml @@ -5,7 +5,9 @@ python: sudo: false env: - - TEST_DIR=tests/em + - TEST_DIR=tests/em/examples + - TEST_DIR=tests/em/fdem + - TEST_DIR=tests/em/tdem - TEST_DIR=tests/mesh - TEST_DIR=tests/flow - TEST_DIR=tests/utils diff --git a/tests/em_examples/__init__.py b/tests/em/examples/__init__.py similarity index 100% rename from tests/em_examples/__init__.py rename to tests/em/examples/__init__.py diff --git a/tests/em_examples/test_Examples.py b/tests/em/examples/test_Examples.py similarity index 100% rename from tests/em_examples/test_Examples.py rename to tests/em/examples/test_Examples.py diff --git a/tests/em_fdem/__init__.py b/tests/em/fdem/__init__.py similarity index 100% rename from tests/em_fdem/__init__.py rename to tests/em/fdem/__init__.py diff --git a/tests/em_fdem/test_FDEM.py b/tests/em/fdem/test_FDEM.py similarity index 100% rename from tests/em_fdem/test_FDEM.py rename to tests/em/fdem/test_FDEM.py diff --git a/tests/em_fdem/test_FDEMCasing.py b/tests/em/fdem/test_FDEMCasing.py similarity index 100% rename from tests/em_fdem/test_FDEMCasing.py rename to tests/em/fdem/test_FDEMCasing.py diff --git a/tests/em_fdem/test_FDEM_analytics.py b/tests/em/fdem/test_FDEM_analytics.py similarity index 100% rename from tests/em_fdem/test_FDEM_analytics.py rename to tests/em/fdem/test_FDEM_analytics.py diff --git a/tests/em_tdem/__init__.py b/tests/em/tdem/__init__.py similarity index 100% rename from tests/em_tdem/__init__.py rename to tests/em/tdem/__init__.py diff --git a/tests/em_tdem/test_TDEM_b_DerivAdjoint.py b/tests/em/tdem/test_TDEM_b_DerivAdjoint.py similarity index 100% rename from tests/em_tdem/test_TDEM_b_DerivAdjoint.py rename to tests/em/tdem/test_TDEM_b_DerivAdjoint.py diff --git a/tests/em_tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py b/tests/em/tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py similarity index 100% rename from tests/em_tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py rename to tests/em/tdem/test_TDEM_b_MultiSrc_DerivAdjoint.py diff --git a/tests/em_tdem/test_TDEM_combos.py b/tests/em/tdem/test_TDEM_combos.py similarity index 100% rename from tests/em_tdem/test_TDEM_combos.py rename to tests/em/tdem/test_TDEM_combos.py diff --git a/tests/em_tdem/test_TDEM_forward_Analytic.py b/tests/em/tdem/test_TDEM_forward_Analytic.py similarity index 100% rename from tests/em_tdem/test_TDEM_forward_Analytic.py rename to tests/em/tdem/test_TDEM_forward_Analytic.py From ac9b50358fdfe0dae709d90d67de0c125b092857 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 13:30:37 -0800 Subject: [PATCH 4/9] split fdem tests into forward and inverse --- .travis.yml | 3 +- .../fdem/{ => forward}/test_FDEM_analytics.py | 0 .../test_FDEM_casing.py} | 0 tests/em/fdem/forward/test_FDEM_forward.py | 211 ++++++++++++++++++ .../test_FDEM_derivs.py} | 52 ----- 5 files changed, 213 insertions(+), 53 deletions(-) rename tests/em/fdem/{ => forward}/test_FDEM_analytics.py (100%) rename tests/em/fdem/{test_FDEMCasing.py => forward/test_FDEM_casing.py} (100%) create mode 100644 tests/em/fdem/forward/test_FDEM_forward.py rename tests/em/fdem/{test_FDEM.py => inverse/test_FDEM_derivs.py} (93%) diff --git a/.travis.yml b/.travis.yml index c4bd33ef..90177108 100644 --- a/.travis.yml +++ b/.travis.yml @@ -6,7 +6,8 @@ sudo: false env: - TEST_DIR=tests/em/examples - - TEST_DIR=tests/em/fdem + - TEST_DIR=tests/em/fdem/forward + - TEST_DIR=tests/em/fdem/inverse - TEST_DIR=tests/em/tdem - TEST_DIR=tests/mesh - TEST_DIR=tests/flow diff --git a/tests/em/fdem/test_FDEM_analytics.py b/tests/em/fdem/forward/test_FDEM_analytics.py similarity index 100% rename from tests/em/fdem/test_FDEM_analytics.py rename to tests/em/fdem/forward/test_FDEM_analytics.py diff --git a/tests/em/fdem/test_FDEMCasing.py b/tests/em/fdem/forward/test_FDEM_casing.py similarity index 100% rename from tests/em/fdem/test_FDEMCasing.py rename to tests/em/fdem/forward/test_FDEM_casing.py diff --git a/tests/em/fdem/forward/test_FDEM_forward.py b/tests/em/fdem/forward/test_FDEM_forward.py new file mode 100644 index 00000000..af07e8fc --- /dev/null +++ b/tests/em/fdem/forward/test_FDEM_forward.py @@ -0,0 +1,211 @@ +import unittest +from SimPEG import * +from SimPEG import EM +import sys +from scipy.constants import mu_0 + +testCrossCheck = True +testEB = True +testHJ = True + + +verbose = False + +TOL = 1e-5 +FLR = 1e-20 # "zero", so if residual below this --> pass regardless of order +CONDUCTIVITY = 1e1 +MU = mu_0 +freq = 1e-1 +addrandoms = True + +SrcType = 'RawVec' #or 'MAgDipole_Bfield', 'CircularLoop', 'RawVec' + + +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.array([np.linspace(-30,-15,3),np.linspace(15,30,3)]) #don't sample right by the source + XYZ = Utils.ndgrid(x,x,np.r_[0.]) + Rx0 = EM.FDEM.RxFDEM(XYZ, comp) + + if SrcType is 'MagDipole': + Src = EM.FDEM.SrcFDEM_MagDipole([Rx0], freq=freq, loc=np.r_[0.,0.,0.]) + elif SrcType is 'MagDipole_Bfield': + Src = EM.FDEM.SrcFDEM_MagDipole_Bfield([Rx0], freq=freq, loc=np.r_[0.,0.,0.]) + elif SrcType is 'CircularLoop': + Src2 = EM.FDEM.SrcFDEM_CircularLoop([Rx0], freq=freq, loc=np.r_[0.,0.,0.]) + + if verbose: + print ' Fetching %s problem' % (fdemType) + + if fdemType == 'e': + if SrcType is 'RawVec': + S_m = np.zeros(mesh.nF) + S_e = np.zeros(mesh.nE) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) + + survey = EM.FDEM.SurveyFDEM([Src]) + prb = EM.FDEM.ProblemFDEM_e(mesh, mapping=mapping) + + elif fdemType == 'b': + if SrcType is 'RawVec': + S_m = np.zeros(mesh.nF) + S_e = np.zeros(mesh.nE) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) + + survey = EM.FDEM.SurveyFDEM([Src]) + prb = EM.FDEM.ProblemFDEM_b(mesh, mapping=mapping) + + elif fdemType == 'j': + if SrcType is 'RawVec': + S_m = np.zeros(mesh.nE) + S_e = np.zeros(mesh.nF) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) + + survey = EM.FDEM.SurveyFDEM([Src]) + prb = EM.FDEM.ProblemFDEM_j(mesh, mapping=mapping) + + elif fdemType == 'h': + if SrcType is 'RawVec': + S_m = np.zeros(mesh.nE) + S_e = np.zeros(mesh.nF) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) + + survey = EM.FDEM.SurveyFDEM([Src]) + prb = EM.FDEM.ProblemFDEM_h(mesh, mapping=mapping) + + else: + raise NotImplementedError() + prb.pair(survey) + + try: + from pymatsolver import MumpsSolver + prb.Solver = MumpsSolver + except ImportError, e: + pass + + return prb + + +def crossCheckTest(fdemType, comp): + + l2norm = lambda r: np.sqrt(r.dot(r)) + + prb1 = getProblem(fdemType, comp) + mesh = prb1.mesh + print 'Cross Checking Forward: %s formulation - %s' % (fdemType, comp) + m = np.log(np.ones(mesh.nC)*CONDUCTIVITY) + mu = np.log(np.ones(mesh.nC)*MU) + + if addrandoms is True: + m = m + np.random.randn(mesh.nC)*np.log(CONDUCTIVITY)*1e-1 + mu = mu + np.random.randn(mesh.nC)*MU*1e-1 + + # prb1.PropMap.PropModel.mu = mu + # prb1.PropMap.PropModel.mui = 1./mu + survey1 = prb1.survey + d1 = survey1.dpred(m) + + if verbose: + print ' Problem 1 solved' + + if fdemType == 'e': + prb2 = getProblem('b', comp) + elif fdemType == 'b': + prb2 = getProblem('e', comp) + elif fdemType == 'j': + prb2 = getProblem('h', comp) + elif fdemType == 'h': + prb2 = getProblem('j', comp) + else: + raise NotImplementedError() + + # prb2.mu = mu + survey2 = prb2.survey + d2 = survey2.dpred(m) + + if verbose: + print ' Problem 2 solved' + + r = d2-d1 + l2r = l2norm(r) + + tol = np.max([TOL*(10**int(np.log10(l2norm(d1)))),FLR]) + print l2norm(d1), l2norm(d2), l2r , tol, l2r < tol + return l2r < tol + + +if testCrossCheck: + if testEB: + def test_EB_CrossCheck_exr_Eform(self): + self.assertTrue(crossCheckTest('e', 'exr')) + def test_EB_CrossCheck_eyr_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyr')) + def test_EB_CrossCheck_ezr_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezr')) + def test_EB_CrossCheck_exi_Eform(self): + self.assertTrue(crossCheckTest('e', 'exi')) + def test_EB_CrossCheck_eyi_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyi')) + def test_EB_CrossCheck_ezi_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezi')) + + def test_EB_CrossCheck_bxr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxr')) + def test_EB_CrossCheck_byr_Eform(self): + self.assertTrue(crossCheckTest('e', 'byr')) + def test_EB_CrossCheck_bzr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzr')) + def test_EB_CrossCheck_bxi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxi')) + def test_EB_CrossCheck_byi_Eform(self): + self.assertTrue(crossCheckTest('e', 'byi')) + def test_EB_CrossCheck_bzi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzi')) + + if testHJ: + def test_HJ_CrossCheck_jxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxr')) + def test_HJ_CrossCheck_jyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyr')) + def test_HJ_CrossCheck_jzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzr')) + def test_HJ_CrossCheck_jxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxi')) + def test_HJ_CrossCheck_jyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyi')) + def test_HJ_CrossCheck_jzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzi')) + + def test_HJ_CrossCheck_hxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxr')) + def test_HJ_CrossCheck_hyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyr')) + def test_HJ_CrossCheck_hzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzr')) + def test_HJ_CrossCheck_hxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxi')) + def test_HJ_CrossCheck_hyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyi')) + def test_HJ_CrossCheck_hzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzi')) + +if __name__ == '__main__': + unittest.main() + diff --git a/tests/em/fdem/test_FDEM.py b/tests/em/fdem/inverse/test_FDEM_derivs.py similarity index 93% rename from tests/em/fdem/test_FDEM.py rename to tests/em/fdem/inverse/test_FDEM_derivs.py index 23bfa0d2..8053c5b7 100644 --- a/tests/em/fdem/test_FDEM.py +++ b/tests/em/fdem/inverse/test_FDEM_derivs.py @@ -5,7 +5,6 @@ import sys from scipy.constants import mu_0 testDerivs = True -testCrossCheck = True testAdjoint = True testEB = True testHJ = True @@ -149,57 +148,6 @@ def derivTest(fdemType, comp): return Tests.checkDerivative(fun, x0, num=3, plotIt=False, eps=FLR) -def crossCheckTest(fdemType, comp): - - l2norm = lambda r: np.sqrt(r.dot(r)) - - prb1 = getProblem(fdemType, comp) - mesh = prb1.mesh - print 'Cross Checking Forward: %s formulation - %s' % (fdemType, comp) - m = np.log(np.ones(mesh.nC)*CONDUCTIVITY) - mu = np.log(np.ones(mesh.nC)*MU) - - if addrandoms is True: - m = m + np.random.randn(mesh.nC)*np.log(CONDUCTIVITY)*1e-1 - mu = mu + np.random.randn(mesh.nC)*MU*1e-1 - - # prb1.PropMap.PropModel.mu = mu - # prb1.PropMap.PropModel.mui = 1./mu - survey1 = prb1.survey - d1 = survey1.dpred(m) - - if verbose: - print ' Problem 1 solved' - - if fdemType == 'e': - prb2 = getProblem('b', comp) - elif fdemType == 'b': - prb2 = getProblem('e', comp) - elif fdemType == 'j': - prb2 = getProblem('h', comp) - elif fdemType == 'h': - prb2 = getProblem('j', comp) - else: - raise NotImplementedError() - - # prb2.mu = mu - survey2 = prb2.survey - d2 = survey2.dpred(m) - - if verbose: - print ' Problem 2 solved' - - r = d2-d1 - l2r = l2norm(r) - - tol = np.max([TOL*(10**int(np.log10(l2norm(d1)))),FLR]) - print l2norm(d1), l2norm(d2), l2r , tol, l2r < tol - return l2r < tol - - - - - class FDEM_DerivTests(unittest.TestCase): if testDerivs: From 8ce9699b8a57517be04be84eec3dc24d23248cec Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 13:34:01 -0800 Subject: [PATCH 5/9] removed cross-check test from derive test --- tests/em/fdem/inverse/test_FDEM_derivs.py | 56 ----------------------- 1 file changed, 56 deletions(-) diff --git a/tests/em/fdem/inverse/test_FDEM_derivs.py b/tests/em/fdem/inverse/test_FDEM_derivs.py index 8053c5b7..179ff069 100644 --- a/tests/em/fdem/inverse/test_FDEM_derivs.py +++ b/tests/em/fdem/inverse/test_FDEM_derivs.py @@ -366,61 +366,5 @@ class FDEM_DerivTests(unittest.TestCase): self.assertTrue(adjointTest('h', 'jzi')) - if testCrossCheck: - if testEB: - def test_EB_CrossCheck_exr_Eform(self): - self.assertTrue(crossCheckTest('e', 'exr')) - def test_EB_CrossCheck_eyr_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyr')) - def test_EB_CrossCheck_ezr_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezr')) - def test_EB_CrossCheck_exi_Eform(self): - self.assertTrue(crossCheckTest('e', 'exi')) - def test_EB_CrossCheck_eyi_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyi')) - def test_EB_CrossCheck_ezi_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezi')) - - def test_EB_CrossCheck_bxr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxr')) - def test_EB_CrossCheck_byr_Eform(self): - self.assertTrue(crossCheckTest('e', 'byr')) - def test_EB_CrossCheck_bzr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzr')) - def test_EB_CrossCheck_bxi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxi')) - def test_EB_CrossCheck_byi_Eform(self): - self.assertTrue(crossCheckTest('e', 'byi')) - def test_EB_CrossCheck_bzi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzi')) - - if testHJ: - def test_HJ_CrossCheck_jxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxr')) - def test_HJ_CrossCheck_jyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyr')) - def test_HJ_CrossCheck_jzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzr')) - def test_HJ_CrossCheck_jxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxi')) - def test_HJ_CrossCheck_jyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyi')) - def test_HJ_CrossCheck_jzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzi')) - - def test_HJ_CrossCheck_hxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxr')) - def test_HJ_CrossCheck_hyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyr')) - def test_HJ_CrossCheck_hzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzr')) - def test_HJ_CrossCheck_hxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxi')) - def test_HJ_CrossCheck_hyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyi')) - def test_HJ_CrossCheck_hzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzi')) - - if __name__ == '__main__': unittest.main() From 6669c3d3aaf2c668b5d7183491c6440a101c6619 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 14:09:27 -0800 Subject: [PATCH 6/9] fixed alignment --- tests/em/fdem/forward/test_FDEM_forward.py | 104 ++++++++++----------- 1 file changed, 51 insertions(+), 53 deletions(-) diff --git a/tests/em/fdem/forward/test_FDEM_forward.py b/tests/em/fdem/forward/test_FDEM_forward.py index af07e8fc..b8e05859 100644 --- a/tests/em/fdem/forward/test_FDEM_forward.py +++ b/tests/em/fdem/forward/test_FDEM_forward.py @@ -8,7 +8,6 @@ testCrossCheck = True testEB = True testHJ = True - verbose = False TOL = 1e-5 @@ -152,60 +151,59 @@ def crossCheckTest(fdemType, comp): if testCrossCheck: - if testEB: - def test_EB_CrossCheck_exr_Eform(self): - self.assertTrue(crossCheckTest('e', 'exr')) - def test_EB_CrossCheck_eyr_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyr')) - def test_EB_CrossCheck_ezr_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezr')) - def test_EB_CrossCheck_exi_Eform(self): - self.assertTrue(crossCheckTest('e', 'exi')) - def test_EB_CrossCheck_eyi_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyi')) - def test_EB_CrossCheck_ezi_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezi')) + if testEB: + def test_EB_CrossCheck_exr_Eform(self): + self.assertTrue(crossCheckTest('e', 'exr')) + def test_EB_CrossCheck_eyr_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyr')) + def test_EB_CrossCheck_ezr_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezr')) + def test_EB_CrossCheck_exi_Eform(self): + self.assertTrue(crossCheckTest('e', 'exi')) + def test_EB_CrossCheck_eyi_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyi')) + def test_EB_CrossCheck_ezi_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezi')) - def test_EB_CrossCheck_bxr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxr')) - def test_EB_CrossCheck_byr_Eform(self): - self.assertTrue(crossCheckTest('e', 'byr')) - def test_EB_CrossCheck_bzr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzr')) - def test_EB_CrossCheck_bxi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxi')) - def test_EB_CrossCheck_byi_Eform(self): - self.assertTrue(crossCheckTest('e', 'byi')) - def test_EB_CrossCheck_bzi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzi')) + def test_EB_CrossCheck_bxr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxr')) + def test_EB_CrossCheck_byr_Eform(self): + self.assertTrue(crossCheckTest('e', 'byr')) + def test_EB_CrossCheck_bzr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzr')) + def test_EB_CrossCheck_bxi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxi')) + def test_EB_CrossCheck_byi_Eform(self): + self.assertTrue(crossCheckTest('e', 'byi')) + def test_EB_CrossCheck_bzi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzi')) - if testHJ: - def test_HJ_CrossCheck_jxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxr')) - def test_HJ_CrossCheck_jyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyr')) - def test_HJ_CrossCheck_jzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzr')) - def test_HJ_CrossCheck_jxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxi')) - def test_HJ_CrossCheck_jyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyi')) - def test_HJ_CrossCheck_jzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzi')) + if testHJ: + def test_HJ_CrossCheck_jxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxr')) + def test_HJ_CrossCheck_jyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyr')) + def test_HJ_CrossCheck_jzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzr')) + def test_HJ_CrossCheck_jxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxi')) + def test_HJ_CrossCheck_jyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyi')) + def test_HJ_CrossCheck_jzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzi')) - def test_HJ_CrossCheck_hxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxr')) - def test_HJ_CrossCheck_hyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyr')) - def test_HJ_CrossCheck_hzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzr')) - def test_HJ_CrossCheck_hxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxi')) - def test_HJ_CrossCheck_hyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyi')) - def test_HJ_CrossCheck_hzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzi')) + def test_HJ_CrossCheck_hxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxr')) + def test_HJ_CrossCheck_hyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyr')) + def test_HJ_CrossCheck_hzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzr')) + def test_HJ_CrossCheck_hxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxi')) + def test_HJ_CrossCheck_hyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyi')) + def test_HJ_CrossCheck_hzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzi')) if __name__ == '__main__': - unittest.main() - + unittest.main() \ No newline at end of file From a4a2c400a7e5473633c1aa57afb40d6fe23e6a5f Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 14:12:12 -0800 Subject: [PATCH 7/9] fixed cross check test --- tests/em/fdem/forward/test_FDEM_forward.py | 103 +++++++++++---------- 1 file changed, 52 insertions(+), 51 deletions(-) diff --git a/tests/em/fdem/forward/test_FDEM_forward.py b/tests/em/fdem/forward/test_FDEM_forward.py index b8e05859..b9fcf74e 100644 --- a/tests/em/fdem/forward/test_FDEM_forward.py +++ b/tests/em/fdem/forward/test_FDEM_forward.py @@ -150,60 +150,61 @@ def crossCheckTest(fdemType, comp): return l2r < tol -if testCrossCheck: - if testEB: - def test_EB_CrossCheck_exr_Eform(self): - self.assertTrue(crossCheckTest('e', 'exr')) - def test_EB_CrossCheck_eyr_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyr')) - def test_EB_CrossCheck_ezr_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezr')) - def test_EB_CrossCheck_exi_Eform(self): - self.assertTrue(crossCheckTest('e', 'exi')) - def test_EB_CrossCheck_eyi_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyi')) - def test_EB_CrossCheck_ezi_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezi')) +class FDEM_CrossCheck(unittest.TestCase): + if testCrossCheck: + if testEB: + def test_EB_CrossCheck_exr_Eform(self): + self.assertTrue(crossCheckTest('e', 'exr')) + def test_EB_CrossCheck_eyr_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyr')) + def test_EB_CrossCheck_ezr_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezr')) + def test_EB_CrossCheck_exi_Eform(self): + self.assertTrue(crossCheckTest('e', 'exi')) + def test_EB_CrossCheck_eyi_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyi')) + def test_EB_CrossCheck_ezi_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezi')) - def test_EB_CrossCheck_bxr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxr')) - def test_EB_CrossCheck_byr_Eform(self): - self.assertTrue(crossCheckTest('e', 'byr')) - def test_EB_CrossCheck_bzr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzr')) - def test_EB_CrossCheck_bxi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxi')) - def test_EB_CrossCheck_byi_Eform(self): - self.assertTrue(crossCheckTest('e', 'byi')) - def test_EB_CrossCheck_bzi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzi')) + def test_EB_CrossCheck_bxr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxr')) + def test_EB_CrossCheck_byr_Eform(self): + self.assertTrue(crossCheckTest('e', 'byr')) + def test_EB_CrossCheck_bzr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzr')) + def test_EB_CrossCheck_bxi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxi')) + def test_EB_CrossCheck_byi_Eform(self): + self.assertTrue(crossCheckTest('e', 'byi')) + def test_EB_CrossCheck_bzi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzi')) - if testHJ: - def test_HJ_CrossCheck_jxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxr')) - def test_HJ_CrossCheck_jyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyr')) - def test_HJ_CrossCheck_jzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzr')) - def test_HJ_CrossCheck_jxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxi')) - def test_HJ_CrossCheck_jyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyi')) - def test_HJ_CrossCheck_jzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzi')) + if testHJ: + def test_HJ_CrossCheck_jxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxr')) + def test_HJ_CrossCheck_jyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyr')) + def test_HJ_CrossCheck_jzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzr')) + def test_HJ_CrossCheck_jxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxi')) + def test_HJ_CrossCheck_jyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyi')) + def test_HJ_CrossCheck_jzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzi')) - def test_HJ_CrossCheck_hxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxr')) - def test_HJ_CrossCheck_hyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyr')) - def test_HJ_CrossCheck_hzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzr')) - def test_HJ_CrossCheck_hxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxi')) - def test_HJ_CrossCheck_hyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyi')) - def test_HJ_CrossCheck_hzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzi')) + def test_HJ_CrossCheck_hxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxr')) + def test_HJ_CrossCheck_hyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyr')) + def test_HJ_CrossCheck_hzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzr')) + def test_HJ_CrossCheck_hxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxi')) + def test_HJ_CrossCheck_hyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyi')) + def test_HJ_CrossCheck_hzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzi')) if __name__ == '__main__': unittest.main() \ No newline at end of file From 14f7cf4fff4565377245643595a02b22f4b71078 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 14:58:47 -0800 Subject: [PATCH 8/9] test 4 source types for cross check --- tests/em/fdem/forward/test_FDEM_forward.py | 67 +++++++++------------- 1 file changed, 28 insertions(+), 39 deletions(-) diff --git a/tests/em/fdem/forward/test_FDEM_forward.py b/tests/em/fdem/forward/test_FDEM_forward.py index b9fcf74e..4f2bfd1f 100644 --- a/tests/em/fdem/forward/test_FDEM_forward.py +++ b/tests/em/fdem/forward/test_FDEM_forward.py @@ -17,7 +17,7 @@ MU = mu_0 freq = 1e-1 addrandoms = True -SrcType = 'RawVec' #or 'MAgDipole_Bfield', 'CircularLoop', 'RawVec' +SrcList = ['RawVec', 'MagDipole_Bfield', 'MagDipole', 'CircularLoop'] def getProblem(fdemType, comp): @@ -35,58 +35,47 @@ def getProblem(fdemType, comp): XYZ = Utils.ndgrid(x,x,np.r_[0.]) Rx0 = EM.FDEM.RxFDEM(XYZ, comp) - if SrcType is 'MagDipole': - Src = EM.FDEM.SrcFDEM_MagDipole([Rx0], freq=freq, loc=np.r_[0.,0.,0.]) - elif SrcType is 'MagDipole_Bfield': - Src = EM.FDEM.SrcFDEM_MagDipole_Bfield([Rx0], freq=freq, loc=np.r_[0.,0.,0.]) - elif SrcType is 'CircularLoop': - Src2 = EM.FDEM.SrcFDEM_CircularLoop([Rx0], freq=freq, loc=np.r_[0.,0.,0.]) + Src = [] + + for SrcType in SrcList: + if SrcType is 'MagDipole': + Src.append(EM.FDEM.SrcFDEM_MagDipole([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) + elif SrcType is 'MagDipole_Bfield': + Src.append(EM.FDEM.SrcFDEM_MagDipole_Bfield([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) + elif SrcType is 'CircularLoop': + Src.append(EM.FDEM.SrcFDEM_CircularLoop([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) + elif SrcType is 'RawVec': + if fdemType is 'e' or fdemType is 'b': + S_m = np.zeros(mesh.nF) + S_e = np.zeros(mesh.nE) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + Src.append(EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e)) + + elif fdemType is 'h' or fdemType is 'j': + S_m = np.zeros(mesh.nE) + S_e = np.zeros(mesh.nF) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + Src.append(EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e)) if verbose: print ' Fetching %s problem' % (fdemType) if fdemType == 'e': - if SrcType is 'RawVec': - S_m = np.zeros(mesh.nF) - S_e = np.zeros(mesh.nE) - S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. - S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. - Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) - - survey = EM.FDEM.SurveyFDEM([Src]) + survey = EM.FDEM.SurveyFDEM(Src) prb = EM.FDEM.ProblemFDEM_e(mesh, mapping=mapping) elif fdemType == 'b': - if SrcType is 'RawVec': - S_m = np.zeros(mesh.nF) - S_e = np.zeros(mesh.nE) - S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. - S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. - Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) - - survey = EM.FDEM.SurveyFDEM([Src]) + survey = EM.FDEM.SurveyFDEM(Src) prb = EM.FDEM.ProblemFDEM_b(mesh, mapping=mapping) elif fdemType == 'j': - if SrcType is 'RawVec': - S_m = np.zeros(mesh.nE) - S_e = np.zeros(mesh.nF) - S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. - S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. - Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) - - survey = EM.FDEM.SurveyFDEM([Src]) + survey = EM.FDEM.SurveyFDEM(Src) prb = EM.FDEM.ProblemFDEM_j(mesh, mapping=mapping) elif fdemType == 'h': - if SrcType is 'RawVec': - S_m = np.zeros(mesh.nE) - S_e = np.zeros(mesh.nF) - S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. - S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. - Src = EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e) - - survey = EM.FDEM.SurveyFDEM([Src]) + survey = EM.FDEM.SurveyFDEM(Src) prb = EM.FDEM.ProblemFDEM_h(mesh, mapping=mapping) else: From 41970a221a536389605bb9c8c7db652c6f764640 Mon Sep 17 00:00:00 2001 From: Lindsey Heagy Date: Thu, 5 Nov 2015 15:44:35 -0800 Subject: [PATCH 9/9] split up adjoint and deriv test, pulled out getProblem call and put it in utils --- .travis.yml | 3 +- SimPEG/EM/Utils/testingUtils.py | 75 +++++++ tests/em/fdem/{ => forward}/__init__.py | 0 tests/em/fdem/forward/test_FDEM_forward.py | 184 ++++++------------ tests/em/fdem/inverse/adjoint/__init__.py | 11 ++ .../fdem/inverse/adjoint/test_FDEM_adjoint.py | 156 +++++++++++++++ tests/em/fdem/inverse/derivs/__init__.py | 11 ++ .../inverse/{ => derivs}/test_FDEM_derivs.py | 0 8 files changed, 311 insertions(+), 129 deletions(-) create mode 100644 SimPEG/EM/Utils/testingUtils.py rename tests/em/fdem/{ => forward}/__init__.py (100%) create mode 100644 tests/em/fdem/inverse/adjoint/__init__.py create mode 100644 tests/em/fdem/inverse/adjoint/test_FDEM_adjoint.py create mode 100644 tests/em/fdem/inverse/derivs/__init__.py rename tests/em/fdem/inverse/{ => derivs}/test_FDEM_derivs.py (100%) diff --git a/.travis.yml b/.travis.yml index 90177108..a64ddcc2 100644 --- a/.travis.yml +++ b/.travis.yml @@ -7,7 +7,8 @@ sudo: false env: - TEST_DIR=tests/em/examples - TEST_DIR=tests/em/fdem/forward - - TEST_DIR=tests/em/fdem/inverse + - TEST_DIR=tests/em/fdem/inverse/derivs + - TEST_DIR=tests/em/fdem/inverse/adjoint - TEST_DIR=tests/em/tdem - TEST_DIR=tests/mesh - TEST_DIR=tests/flow diff --git a/SimPEG/EM/Utils/testingUtils.py b/SimPEG/EM/Utils/testingUtils.py new file mode 100644 index 00000000..b4570c0d --- /dev/null +++ b/SimPEG/EM/Utils/testingUtils.py @@ -0,0 +1,75 @@ +import unittest +from SimPEG import * +from SimPEG import EM +import sys +from scipy.constants import mu_0 + +def getFDEMProblem(fdemType, comp, SrcList, freq, verbose=False): + 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.array([np.linspace(-30,-15,3),np.linspace(15,30,3)]) #don't sample right by the source + XYZ = Utils.ndgrid(x,x,np.r_[0.]) + Rx0 = EM.FDEM.RxFDEM(XYZ, comp) + + Src = [] + + for SrcType in SrcList: + if SrcType is 'MagDipole': + Src.append(EM.FDEM.SrcFDEM_MagDipole([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) + elif SrcType is 'MagDipole_Bfield': + Src.append(EM.FDEM.SrcFDEM_MagDipole_Bfield([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) + elif SrcType is 'CircularLoop': + Src.append(EM.FDEM.SrcFDEM_CircularLoop([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) + elif SrcType is 'RawVec': + if fdemType is 'e' or fdemType is 'b': + S_m = np.zeros(mesh.nF) + S_e = np.zeros(mesh.nE) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + Src.append(EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e)) + + elif fdemType is 'h' or fdemType is 'j': + S_m = np.zeros(mesh.nE) + S_e = np.zeros(mesh.nF) + S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. + S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. + Src.append(EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e)) + + if verbose: + print ' Fetching %s problem' % (fdemType) + + if fdemType == 'e': + survey = EM.FDEM.SurveyFDEM(Src) + prb = EM.FDEM.ProblemFDEM_e(mesh, mapping=mapping) + + elif fdemType == 'b': + survey = EM.FDEM.SurveyFDEM(Src) + prb = EM.FDEM.ProblemFDEM_b(mesh, mapping=mapping) + + elif fdemType == 'j': + survey = EM.FDEM.SurveyFDEM(Src) + prb = EM.FDEM.ProblemFDEM_j(mesh, mapping=mapping) + + elif fdemType == 'h': + survey = EM.FDEM.SurveyFDEM(Src) + prb = EM.FDEM.ProblemFDEM_h(mesh, mapping=mapping) + + else: + raise NotImplementedError() + prb.pair(survey) + + try: + from pymatsolver import MumpsSolver + prb.Solver = MumpsSolver + except ImportError, e: + pass + + return prb \ No newline at end of file diff --git a/tests/em/fdem/__init__.py b/tests/em/fdem/forward/__init__.py similarity index 100% rename from tests/em/fdem/__init__.py rename to tests/em/fdem/forward/__init__.py diff --git a/tests/em/fdem/forward/test_FDEM_forward.py b/tests/em/fdem/forward/test_FDEM_forward.py index 4f2bfd1f..437f3708 100644 --- a/tests/em/fdem/forward/test_FDEM_forward.py +++ b/tests/em/fdem/forward/test_FDEM_forward.py @@ -3,8 +3,8 @@ from SimPEG import * from SimPEG import EM import sys from scipy.constants import mu_0 +from SimPEG.EM.Utils.testingUtils import getFDEMProblem -testCrossCheck = True testEB = True testHJ = True @@ -20,82 +20,11 @@ addrandoms = True SrcList = ['RawVec', 'MagDipole_Bfield', 'MagDipole', 'CircularLoop'] -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.array([np.linspace(-30,-15,3),np.linspace(15,30,3)]) #don't sample right by the source - XYZ = Utils.ndgrid(x,x,np.r_[0.]) - Rx0 = EM.FDEM.RxFDEM(XYZ, comp) - - Src = [] - - for SrcType in SrcList: - if SrcType is 'MagDipole': - Src.append(EM.FDEM.SrcFDEM_MagDipole([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) - elif SrcType is 'MagDipole_Bfield': - Src.append(EM.FDEM.SrcFDEM_MagDipole_Bfield([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) - elif SrcType is 'CircularLoop': - Src.append(EM.FDEM.SrcFDEM_CircularLoop([Rx0], freq=freq, loc=np.r_[0.,0.,0.])) - elif SrcType is 'RawVec': - if fdemType is 'e' or fdemType is 'b': - S_m = np.zeros(mesh.nF) - S_e = np.zeros(mesh.nE) - S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. - S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. - Src.append(EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e)) - - elif fdemType is 'h' or fdemType is 'j': - S_m = np.zeros(mesh.nE) - S_e = np.zeros(mesh.nF) - S_m[Utils.closestPoints(mesh,[0.,0.,0.],'Ez') + np.sum(mesh.vnE[:1])] = 1. - S_e[Utils.closestPoints(mesh,[0.,0.,0.],'Fz') + np.sum(mesh.vnF[:1])] = 1. - Src.append(EM.FDEM.SrcFDEM_RawVec([Rx0], freq, S_m, S_e)) - - if verbose: - print ' Fetching %s problem' % (fdemType) - - if fdemType == 'e': - survey = EM.FDEM.SurveyFDEM(Src) - prb = EM.FDEM.ProblemFDEM_e(mesh, mapping=mapping) - - elif fdemType == 'b': - survey = EM.FDEM.SurveyFDEM(Src) - prb = EM.FDEM.ProblemFDEM_b(mesh, mapping=mapping) - - elif fdemType == 'j': - survey = EM.FDEM.SurveyFDEM(Src) - prb = EM.FDEM.ProblemFDEM_j(mesh, mapping=mapping) - - elif fdemType == 'h': - survey = EM.FDEM.SurveyFDEM(Src) - prb = EM.FDEM.ProblemFDEM_h(mesh, mapping=mapping) - - else: - raise NotImplementedError() - prb.pair(survey) - - try: - from pymatsolver import MumpsSolver - prb.Solver = MumpsSolver - except ImportError, e: - pass - - return prb - - def crossCheckTest(fdemType, comp): l2norm = lambda r: np.sqrt(r.dot(r)) - prb1 = getProblem(fdemType, comp) + prb1 = getFDEMProblem(fdemType, comp, SrcList, freq, verbose) mesh = prb1.mesh print 'Cross Checking Forward: %s formulation - %s' % (fdemType, comp) m = np.log(np.ones(mesh.nC)*CONDUCTIVITY) @@ -114,13 +43,13 @@ def crossCheckTest(fdemType, comp): print ' Problem 1 solved' if fdemType == 'e': - prb2 = getProblem('b', comp) + prb2 = getFDEMProblem('b', comp, SrcList, freq, verbose) elif fdemType == 'b': - prb2 = getProblem('e', comp) + prb2 = getFDEMProblem('e', comp, SrcList, freq, verbose) elif fdemType == 'j': - prb2 = getProblem('h', comp) + prb2 = getFDEMProblem('h', comp, SrcList, freq, verbose) elif fdemType == 'h': - prb2 = getProblem('j', comp) + prb2 = getFDEMProblem('j', comp, SrcList, freq, verbose) else: raise NotImplementedError() @@ -140,60 +69,59 @@ def crossCheckTest(fdemType, comp): class FDEM_CrossCheck(unittest.TestCase): - if testCrossCheck: - if testEB: - def test_EB_CrossCheck_exr_Eform(self): - self.assertTrue(crossCheckTest('e', 'exr')) - def test_EB_CrossCheck_eyr_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyr')) - def test_EB_CrossCheck_ezr_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezr')) - def test_EB_CrossCheck_exi_Eform(self): - self.assertTrue(crossCheckTest('e', 'exi')) - def test_EB_CrossCheck_eyi_Eform(self): - self.assertTrue(crossCheckTest('e', 'eyi')) - def test_EB_CrossCheck_ezi_Eform(self): - self.assertTrue(crossCheckTest('e', 'ezi')) + if testEB: + def test_EB_CrossCheck_exr_Eform(self): + self.assertTrue(crossCheckTest('e', 'exr')) + def test_EB_CrossCheck_eyr_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyr')) + def test_EB_CrossCheck_ezr_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezr')) + def test_EB_CrossCheck_exi_Eform(self): + self.assertTrue(crossCheckTest('e', 'exi')) + def test_EB_CrossCheck_eyi_Eform(self): + self.assertTrue(crossCheckTest('e', 'eyi')) + def test_EB_CrossCheck_ezi_Eform(self): + self.assertTrue(crossCheckTest('e', 'ezi')) - def test_EB_CrossCheck_bxr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxr')) - def test_EB_CrossCheck_byr_Eform(self): - self.assertTrue(crossCheckTest('e', 'byr')) - def test_EB_CrossCheck_bzr_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzr')) - def test_EB_CrossCheck_bxi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bxi')) - def test_EB_CrossCheck_byi_Eform(self): - self.assertTrue(crossCheckTest('e', 'byi')) - def test_EB_CrossCheck_bzi_Eform(self): - self.assertTrue(crossCheckTest('e', 'bzi')) + def test_EB_CrossCheck_bxr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxr')) + def test_EB_CrossCheck_byr_Eform(self): + self.assertTrue(crossCheckTest('e', 'byr')) + def test_EB_CrossCheck_bzr_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzr')) + def test_EB_CrossCheck_bxi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bxi')) + def test_EB_CrossCheck_byi_Eform(self): + self.assertTrue(crossCheckTest('e', 'byi')) + def test_EB_CrossCheck_bzi_Eform(self): + self.assertTrue(crossCheckTest('e', 'bzi')) - if testHJ: - def test_HJ_CrossCheck_jxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxr')) - def test_HJ_CrossCheck_jyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyr')) - def test_HJ_CrossCheck_jzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzr')) - def test_HJ_CrossCheck_jxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jxi')) - def test_HJ_CrossCheck_jyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jyi')) - def test_HJ_CrossCheck_jzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'jzi')) + if testHJ: + def test_HJ_CrossCheck_jxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxr')) + def test_HJ_CrossCheck_jyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyr')) + def test_HJ_CrossCheck_jzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzr')) + def test_HJ_CrossCheck_jxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jxi')) + def test_HJ_CrossCheck_jyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jyi')) + def test_HJ_CrossCheck_jzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'jzi')) - def test_HJ_CrossCheck_hxr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxr')) - def test_HJ_CrossCheck_hyr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyr')) - def test_HJ_CrossCheck_hzr_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzr')) - def test_HJ_CrossCheck_hxi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hxi')) - def test_HJ_CrossCheck_hyi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hyi')) - def test_HJ_CrossCheck_hzi_Jform(self): - self.assertTrue(crossCheckTest('j', 'hzi')) + def test_HJ_CrossCheck_hxr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxr')) + def test_HJ_CrossCheck_hyr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyr')) + def test_HJ_CrossCheck_hzr_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzr')) + def test_HJ_CrossCheck_hxi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hxi')) + def test_HJ_CrossCheck_hyi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hyi')) + def test_HJ_CrossCheck_hzi_Jform(self): + self.assertTrue(crossCheckTest('j', 'hzi')) if __name__ == '__main__': unittest.main() \ No newline at end of file diff --git a/tests/em/fdem/inverse/adjoint/__init__.py b/tests/em/fdem/inverse/adjoint/__init__.py new file mode 100644 index 00000000..38d84328 --- /dev/null +++ b/tests/em/fdem/inverse/adjoint/__init__.py @@ -0,0 +1,11 @@ +if __name__ == '__main__': + import os + import glob + import unittest + test_file_strings = glob.glob('test_*.py') + module_strings = [str[0:len(str)-3] for str in test_file_strings] + suites = [unittest.defaultTestLoader.loadTestsFromName(str) for str + in module_strings] + testSuite = unittest.TestSuite(suites) + + unittest.TextTestRunner(verbosity=2).run(testSuite) diff --git a/tests/em/fdem/inverse/adjoint/test_FDEM_adjoint.py b/tests/em/fdem/inverse/adjoint/test_FDEM_adjoint.py new file mode 100644 index 00000000..f77f2131 --- /dev/null +++ b/tests/em/fdem/inverse/adjoint/test_FDEM_adjoint.py @@ -0,0 +1,156 @@ +import unittest +from SimPEG import * +from SimPEG import EM +import sys +from scipy.constants import mu_0 +from SimPEG.EM.Utils.testingUtils import getFDEMProblem + +testEB = True +testHJ = True + +verbose = False + +TOL = 1e-5 +FLR = 1e-20 # "zero", so if residual below this --> pass regardless of order +CONDUCTIVITY = 1e1 +MU = mu_0 +freq = 1e-1 +addrandoms = True + +SrcType = 'RawVec' #or 'MAgDipole_Bfield', 'CircularLoop', 'RawVec' + +def adjointTest(fdemType, comp): + prb = getFDEMProblem(fdemType, comp, [SrcType], freq) + print 'Adjoint %s formulation - %s' % (fdemType, comp) + + m = np.log(np.ones(prb.mapping.nP)*CONDUCTIVITY) + mu = np.ones(prb.mesh.nC)*MU + + if addrandoms is True: + m = m + np.random.randn(prb.mapping.nP)*np.log(CONDUCTIVITY)*1e-1 + mu = mu + np.random.randn(prb.mesh.nC)*MU*1e-1 + + survey = prb.survey + # prb.PropMap.PropModel.mu = mu + # prb.PropMap.PropModel.mui = 1./mu + u = prb.fields(m) + + v = np.random.rand(survey.nD) + w = np.random.rand(prb.mesh.nC) + + vJw = v.dot(prb.Jvec(m, w, u)) + wJtv = w.dot(prb.Jtvec(m, v, u)) + tol = np.max([TOL*(10**int(np.log10(np.abs(vJw)))),FLR]) + print vJw, wJtv, vJw - wJtv, tol, np.abs(vJw - wJtv) < tol + return np.abs(vJw - wJtv) < tol + +class FDEM_AdjointTests(unittest.TestCase): + if testEB: + def test_Jtvec_adjointTest_exr_Eform(self): + self.assertTrue(adjointTest('e', 'exr')) + def test_Jtvec_adjointTest_eyr_Eform(self): + self.assertTrue(adjointTest('e', 'eyr')) + def test_Jtvec_adjointTest_ezr_Eform(self): + self.assertTrue(adjointTest('e', 'ezr')) + def test_Jtvec_adjointTest_exi_Eform(self): + self.assertTrue(adjointTest('e', 'exi')) + def test_Jtvec_adjointTest_eyi_Eform(self): + self.assertTrue(adjointTest('e', 'eyi')) + def test_Jtvec_adjointTest_ezi_Eform(self): + self.assertTrue(adjointTest('e', 'ezi')) + + def test_Jtvec_adjointTest_bxr_Eform(self): + self.assertTrue(adjointTest('e', 'bxr')) + def test_Jtvec_adjointTest_byr_Eform(self): + self.assertTrue(adjointTest('e', 'byr')) + def test_Jtvec_adjointTest_bzr_Eform(self): + self.assertTrue(adjointTest('e', 'bzr')) + def test_Jtvec_adjointTest_bxi_Eform(self): + self.assertTrue(adjointTest('e', 'bxi')) + def test_Jtvec_adjointTest_byi_Eform(self): + self.assertTrue(adjointTest('e', 'byi')) + def test_Jtvec_adjointTest_bzi_Eform(self): + self.assertTrue(adjointTest('e', 'bzi')) + + def test_Jtvec_adjointTest_exr_Bform(self): + self.assertTrue(adjointTest('b', 'exr')) + def test_Jtvec_adjointTest_eyr_Bform(self): + self.assertTrue(adjointTest('b', 'eyr')) + def test_Jtvec_adjointTest_ezr_Bform(self): + self.assertTrue(adjointTest('b', 'ezr')) + def test_Jtvec_adjointTest_exi_Bform(self): + self.assertTrue(adjointTest('b', 'exi')) + def test_Jtvec_adjointTest_eyi_Bform(self): + self.assertTrue(adjointTest('b', 'eyi')) + def test_Jtvec_adjointTest_ezi_Bform(self): + self.assertTrue(adjointTest('b', 'ezi')) + def test_Jtvec_adjointTest_bxr_Bform(self): + self.assertTrue(adjointTest('b', 'bxr')) + def test_Jtvec_adjointTest_byr_Bform(self): + self.assertTrue(adjointTest('b', 'byr')) + def test_Jtvec_adjointTest_bzr_Bform(self): + self.assertTrue(adjointTest('b', 'bzr')) + def test_Jtvec_adjointTest_bxi_Bform(self): + self.assertTrue(adjointTest('b', 'bxi')) + def test_Jtvec_adjointTest_byi_Bform(self): + self.assertTrue(adjointTest('b', 'byi')) + def test_Jtvec_adjointTest_bzi_Bform(self): + self.assertTrue(adjointTest('b', 'bzi')) + + + if testHJ: + def test_Jtvec_adjointTest_jxr_Jform(self): + self.assertTrue(adjointTest('j', 'jxr')) + def test_Jtvec_adjointTest_jyr_Jform(self): + self.assertTrue(adjointTest('j', 'jyr')) + def test_Jtvec_adjointTest_jzr_Jform(self): + self.assertTrue(adjointTest('j', 'jzr')) + def test_Jtvec_adjointTest_jxi_Jform(self): + self.assertTrue(adjointTest('j', 'jxi')) + def test_Jtvec_adjointTest_jyi_Jform(self): + self.assertTrue(adjointTest('j', 'jyi')) + def test_Jtvec_adjointTest_jzi_Jform(self): + self.assertTrue(adjointTest('j', 'jzi')) + + def test_Jtvec_adjointTest_hxr_Jform(self): + self.assertTrue(adjointTest('j', 'hxr')) + def test_Jtvec_adjointTest_hyr_Jform(self): + self.assertTrue(adjointTest('j', 'hyr')) + def test_Jtvec_adjointTest_hzr_Jform(self): + self.assertTrue(adjointTest('j', 'hzr')) + def test_Jtvec_adjointTest_hxi_Jform(self): + self.assertTrue(adjointTest('j', 'hxi')) + def test_Jtvec_adjointTest_hyi_Jform(self): + self.assertTrue(adjointTest('j', 'hyi')) + def test_Jtvec_adjointTest_hzi_Jform(self): + self.assertTrue(adjointTest('j', 'hzi')) + + def test_Jtvec_adjointTest_hxr_Hform(self): + self.assertTrue(adjointTest('h', 'hxr')) + def test_Jtvec_adjointTest_hyr_Hform(self): + self.assertTrue(adjointTest('h', 'hyr')) + def test_Jtvec_adjointTest_hzr_Hform(self): + self.assertTrue(adjointTest('h', 'hzr')) + def test_Jtvec_adjointTest_hxi_Hform(self): + self.assertTrue(adjointTest('h', 'hxi')) + def test_Jtvec_adjointTest_hyi_Hform(self): + self.assertTrue(adjointTest('h', 'hyi')) + def test_Jtvec_adjointTest_hzi_Hform(self): + self.assertTrue(adjointTest('h', 'hzi')) + + def test_Jtvec_adjointTest_hxr_Hform(self): + self.assertTrue(adjointTest('h', 'jxr')) + def test_Jtvec_adjointTest_hyr_Hform(self): + self.assertTrue(adjointTest('h', 'jyr')) + def test_Jtvec_adjointTest_hzr_Hform(self): + self.assertTrue(adjointTest('h', 'jzr')) + def test_Jtvec_adjointTest_hxi_Hform(self): + self.assertTrue(adjointTest('h', 'jxi')) + def test_Jtvec_adjointTest_hyi_Hform(self): + self.assertTrue(adjointTest('h', 'jyi')) + def test_Jtvec_adjointTest_hzi_Hform(self): + self.assertTrue(adjointTest('h', 'jzi')) + + +if __name__ == '__main__': + unittest.main() diff --git a/tests/em/fdem/inverse/derivs/__init__.py b/tests/em/fdem/inverse/derivs/__init__.py new file mode 100644 index 00000000..38d84328 --- /dev/null +++ b/tests/em/fdem/inverse/derivs/__init__.py @@ -0,0 +1,11 @@ +if __name__ == '__main__': + import os + import glob + import unittest + test_file_strings = glob.glob('test_*.py') + module_strings = [str[0:len(str)-3] for str in test_file_strings] + suites = [unittest.defaultTestLoader.loadTestsFromName(str) for str + in module_strings] + testSuite = unittest.TestSuite(suites) + + unittest.TextTestRunner(verbosity=2).run(testSuite) diff --git a/tests/em/fdem/inverse/test_FDEM_derivs.py b/tests/em/fdem/inverse/derivs/test_FDEM_derivs.py similarity index 100% rename from tests/em/fdem/inverse/test_FDEM_derivs.py rename to tests/em/fdem/inverse/derivs/test_FDEM_derivs.py