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

Added AhVec method and test.

Browse Source
This commit is contained in:
Dave Marchant
2014-02-11 22:51:10 -08:00
parent 88fb57bd9f
commit 6e6d90d362
3 changed files with 46 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
simpegEM/Tests/test_forward_EMproblem.py
+15 -5
View File
@@ -5,7 +5,7 @@ from scipy.constants import mu_0
from simpegEM.Utils.Ana import hzAnalyticDipoleT
class EMProblemTests(unittest.TestCase):
class TDEM_bTests(unittest.TestCase):
def setUp(self):
@@ -32,17 +32,27 @@ class EMProblemTests(unittest.TestCase):
self.sigma[mesh.vectorCCz<0] = 0.1
self.prb.pair(self.dat)
def test_analitic_b(self):
bz_calc = self.dat.dpred(self.sigma)
bz_ana = mu_0*hzAnalyticDipoleT(self.dat.rxLoc[0], self.prb.times, self.sigma[0])
diff = np.linalg.norm(bz_calc.flatten() - bz_ana.flatten())/np.linalg.norm(bz_ana.flatten())
print diff
self.assertTrue(diff<0.05)
def test_awesome(self):
self.assertTrue(True)
def test_AhVec(self):
"""
Test that fields and AhVec produce consistent results
"""
sigma = np.ones(self.prb.mesh.nCz)*1e-8
sigma[self.prb.mesh.vectorCCz<0] = 0.1
u = self.prb.fields(sigma)
Ahu = self.prb.AhVec(sigma, u)
self.assertTrue(np.linalg.norm(Ahu.get_b(0)-1/self.prb.getDt(0)*u.get_b(-1))/np.linalg.norm(u.get_b(0)) < 1.e-2)
self.assertTrue(np.linalg.norm(Ahu.get_e(0))/np.linalg.norm(u.get_e(0)) < 1.e-2)
for i in range(1,u.nTimes):
self.assertTrue(np.linalg.norm(Ahu.get_b(i))/np.linalg.norm(u.get_b(i)) < 1.e-2)
self.assertTrue(np.linalg.norm(Ahu.get_e(i))/np.linalg.norm(u.get_e(i)) < 1.e-2)