Fix import bug of sys

simpegMT/ProblemMT.py

@@ -1,7 +1,7 @@
 from SimPEG import Survey, Problem, Utils, Models, np, sp, SolverLU as SimpegSolver
 from scipy.constants import mu_0
 from SurveyMT import SurveyMT, FieldsMT
-import multiprocessing
+import multiprocessing, sys
 
 def omega(freq):
     """Change frequency to angular frequency, omega"""

simpegMT/Tests/test_ApparentResistivityLayerpy

@@ -0,0 +1,48 @@
+import unittest
+from SimPEG import *
+import simpegMT as MT
+
+TOL = 1e-6
+
+def appResPhs(freq,z):
+    app_res = ((1./(8e-7*np.pi**2))/freq)*np.abs(z)**2
+    app_phs = np.arctan2(-z.imag,z.real)*(180/np.pi)
+    return app_res, app_phs
+
+def appResNorm(sigmaHalf):
+    nFreq = 26
+
+    m1d = Mesh.TensorMesh([[(100,5,1.5),(100.,10),(100,5,1.5)]], x0=['C'])
+    sigma = np.zeros(m1d.nC) + sigmaHalf
+    sigma[m1d.gridCC[:]>200] = 1e-8
+
+    # Calculate the analytic fields
+    freqs = np.logspace(4,-4,nFreq)
+    Z = []
+    for freq in freqs:
+        Ed, Eu, Hd, Hu = MT.Utils.getEHfields(m1d,sigma,freq,np.array([200]))
+        Z.append((Ed + Eu)/(Hd + Hu))
+
+    Zarr = np.concatenate(Z)
+
+    app_r, app_p = appResPhs(freqs,Zarr)
+
+    return np.linalg.norm(np.abs(app_r - np.ones(nFreq)/sigmaHalf)) / np.log10(sigmaHalf)
+
+
+class TestAnalytics(unittest.TestCase):
+
+    def setUp(self):
+        pass
+    def test_appRes2en1(self):self.assertLess(appResNorm(2e-1), TOL)
+    def test_appRes2en2(self):self.assertLess(appResNorm(2e-2), TOL)
+    def test_appRes2en3(self):self.assertLess(appResNorm(2e-3), TOL)
+    def test_appRes2en4(self):self.assertLess(appResNorm(2e-4), TOL)
+    def test_appRes2en5(self):self.assertLess(appResNorm(2e-5), TOL)
+    def test_appRes2en6(self):self.assertLess(appResNorm(2e-6), TOL)
+
+
+
+
+if __name__ == '__main__':
+    unittest.main()

simpegMT/Utils/MT1Dsolutions.py

@@ -21,7 +21,7 @@ def get1DEfields(m1d,sigma,freq,sourceAmp=1.0):
 
     # Set the boundary conditions
     Ed, Eu, Hd, Hu = getEHfields(m1d,sigma,freq,m1d.vectorNx)
-    Etot = Ed + Eu
+    Etot = (Ed + Eu).conj()
     if sourceAmp is not None:
         Etot = ((Etot/Etot[-1])*sourceAmp) # Scale the fields to be equal to sourceAmp at the top
     ## Note: need to use conjugate of the analytic solution. It is derived with e^iwt