Updates to DCProblem and testing.

SimPEG/tests/test_forward_problem.py

@@ -2,6 +2,7 @@ import numpy as np
 import unittest
 from SimPEG.mesh import TensorMesh
 from SimPEG.forward import Problem
+from SimPEG.regularization import Regularization
 from TestUtils import checkDerivative
 from scipy.sparse.linalg import dsolve
 
@@ -15,7 +16,7 @@ class ProblemTests(unittest.TestCase):
         c = np.array([1, 4])
         self.mesh2 = TensorMesh([a, b], np.array([3, 5]))
         self.p2 = Problem(self.mesh2)
-
+        self.reg = Regularization(self.mesh2)
 
     def test_modelTransform(self):
         print 'SimPEG.forward.Problem: Testing Model Transform'
@@ -23,6 +24,13 @@ class ProblemTests(unittest.TestCase):
         passed = checkDerivative(lambda m : [self.p2.modelTransform(m), self.p2.modelTransformDeriv(m)], m, plotIt=False)
         self.assertTrue(passed)
 
+    def test_regularization(self):
+        derChk = lambda m: [self.reg.modelObj(m), self.reg.modelObjDeriv(m)]
+        mSynth = np.random.randn(self.mesh2.nC)
+        checkDerivative(derChk, mSynth, plotIt=False)
+
+
+
 
 if __name__ == '__main__':
     unittest.main()