import unittest
import SimPEG.DCIP as DC
from SimPEG import *

class IPforwardTests(unittest.TestCase):

    def test_IPforward(self):

        cs = 12.5
        nc = 200/cs+1
        hx = [(cs,7, -1.3),(cs,nc),(cs,7, 1.3)]
        hy = [(cs,7, -1.3),(cs,int(nc/2+1)),(cs,7, 1.3)]
        hz = [(cs,7, -1.3),(cs,int(nc/2+1))]
        mesh = Mesh.TensorMesh([hx, hy, hz], 'CCN')
        sighalf = 1e-2
        sigma = np.ones(mesh.nC)*sighalf
        p0 = np.r_[-50., 50., -50.]
        p1 = np.r_[ 50.,-50., -150.]
        blk_ind = Utils.ModelBuilder.getIndicesBlock(p0, p1, mesh.gridCC)
        sigma[blk_ind] = 1e-3
        eta = np.zeros_like(sigma)
        eta[blk_ind] = 0.1
        sigmaInf = sigma.copy()
        sigma0 = sigma*(1-eta)

        nElecs = 11
        x_temp = np.linspace(-100, 100, nElecs)
        aSpacing = x_temp[1]-x_temp[0]
        y_temp = 0.
        xyz = Utils.ndgrid(x_temp, np.r_[y_temp], np.r_[0.])
        srcList = DC.Utils.WennerSrcList(nElecs,aSpacing)
        survey = DC.SurveyDC(srcList)

        imap   = Maps.IdentityMap(mesh)
        problem = DC.ProblemDC_CC(mesh, mapping=imap)
        try:
            from pymatsolver import MumpsSolver
            problem.Solver = MumpsSolver
        except ImportError, e:
            problem.Solver = SolverLU

        problem.pair(survey)

        phi0 = survey.dpred(sigma0)
        phiInf = survey.dpred(sigmaInf)

        phiIP_true = phi0-phiInf

        surveyIP = DC.SurveyIP(srcList)
        problemIP = DC.ProblemIP(mesh, sigma=sigma)
        problemIP.pair(surveyIP)

        try:
            from pymatsolver import MumpsSolver
            problemIP.Solver = MumpsSolver
        except Exception, e:
            pass
        phiIP_approx = surveyIP.dpred(eta)

        err =  np.linalg.norm(phiIP_true-phiIP_approx) / np.linalg.norm(phiIP_true)

        self.assertTrue(err < 0.02)


if __name__ == '__main__':
    unittest.main()

@@ -1117,16 +1117,17 @@
 g=imap)%0A
+%0A
         
@@ -1187,33 +1187,25 @@
             
-problem.S
+s
 olver = Mump
@@ -1239,32 +1239,59 @@
 rtError, e:%0A    
+        solver = SolverLU%0A%0A
         problem.
@@ -1295,33 +1295,30 @@
 em.Solver = 
-S
+s
 olver
-LU%0A
 %0A        pro
@@ -1585,149 +1585,34 @@
     
-try:%0A            from pymatsolver import MumpsSolver%0A            problemIP.Solver = MumpsSolver%0A        except Exception, e:%0A            pass
+problemIP.Solver = solver%0A
 %0A   

clean up import solver