Initial work Mag inversion

simpegPF/Tests/test_forward_PFproblem.py

@@ -30,11 +30,11 @@ class MagFwdProblemTests(unittest.TestCase):
 
         data = PF.BaseMag.BaseMagData()
 
-        Inc = 90.
-        Dec = 0.
+        Inc = 45.
+        Dec = 45.
         Btot = 51000
 
-        b0 = PF.MagAnalytics.IDTtoxyz(Inc, Dec, Btot)        
+        b0 = PF.MagAnalytics.IDTtoxyz(Inc, Dec, Btot)
         data.setBackgroundField(Inc, Dec, Btot)
         xr = np.linspace(-300, 300, 41)
         yr = np.linspace(-300, 300, 41)
@@ -42,21 +42,25 @@ class MagFwdProblemTests(unittest.TestCase):
         Z = np.ones((xr.size, yr.size))*150
         rxLoc = np.c_[Utils.mkvc(X), Utils.mkvc(Y), Utils.mkvc(Z)]
         data.rxLoc = rxLoc
-            
+
         self.prob.pair(data)
         u = self.prob.fields(self.chi)
         B = u['B']
 
-        bxa,bya,bza = PF.MagAnalytics.MagSphereAnalFunA(rxLoc[:,0],rxLoc[:,1],rxLoc[:,2],100.,0.,0.,0.,0.01, b0,'secondary') 
-       
+        bxa,bya,bza = PF.MagAnalytics.MagSphereAnalFunA(rxLoc[:,0],rxLoc[:,1],rxLoc[:,2],100.,0.,0.,0.,0.01, b0,'secondary')
+
         dpred = data.projectFieldsAsVector(B)
         err = np.linalg.norm(dpred-np.r_[bxa, bya, bza])/np.linalg.norm(np.r_[bxa, bya, bza])
 
+        plt.plot(dpred)
+        plt.plot(np.r_[bxa, bya, bza])
+        plt.show()
+
         if err > 0.05:
             raise Exception('Anaytic test is failed T.T')
         else:
             print "Anaytic test is passed"
             pass
-    
+
 if __name__ == '__main__':
     unittest.main()