renaming to ensure capitals

2026-07-23 13:10:51 +08:00 · 2014-01-16 13:22:46 -08:00
parent 7432591450
commit fa8a5cd7cb
48 changed files with 0 additions and 0 deletions
@@ -0,0 +1,64 @@
+import unittest
+from SimPEG import Solver
+from SimPEG.Mesh import TensorMesh
+from SimPEG.Utils import sdiag
+import numpy as np
+import scipy.sparse as sp
+from SimPEG import Inverse
+from SimPEG.Tests import getQuadratic, Rosenbrock
+
+TOL = 1e-2
+
+class TestOptimizers(unittest.TestCase):
+
+    def setUp(self):
+        self.A = sp.identity(2).tocsr()
+        self.b = np.array([-5,-5])
+
+    def test_GN_Rosenbrock(self):
+        GN = Inverse.GaussNewton()
+        xopt = GN.minimize(Rosenbrock,np.array([0,0]))
+        x_true = np.array([1.,1.])
+        print 'xopt: ', xopt
+        print 'x_true: ', x_true
+        self.assertTrue(np.linalg.norm(xopt-x_true,2) < TOL, True)
+
+    def test_GN_quadratic(self):
+        GN = Inverse.GaussNewton()
+        xopt = GN.minimize(getQuadratic(self.A,self.b),np.array([0,0]))
+        x_true = np.array([5.,5.])
+        print 'xopt: ', xopt
+        print 'x_true: ', x_true
+        self.assertTrue(np.linalg.norm(xopt-x_true,2) < TOL, True)
+
+    def test_ProjGradient_quadraticBounded(self):
+        PG = Inverse.ProjectedGradient(debug=True)
+        PG.lower, PG.upper = -2, 2
+        xopt = PG.minimize(getQuadratic(self.A,self.b),np.array([0,0]))
+        x_true = np.array([2.,2.])
+        print 'xopt: ', xopt
+        print 'x_true: ', x_true
+        self.assertTrue(np.linalg.norm(xopt-x_true,2) < TOL, True)
+
+    def test_ProjGradient_quadratic1Bound(self):
+        myB = np.array([-5,1])
+        PG = Inverse.ProjectedGradient()
+        PG.lower, PG.upper = -2, 2
+        xopt = PG.minimize(getQuadratic(self.A,myB),np.array([0,0]))
+        x_true = np.array([2.,-1.])
+        print 'xopt: ', xopt
+        print 'x_true: ', x_true
+        self.assertTrue(np.linalg.norm(xopt-x_true,2) < TOL, True)
+
+    def test_NewtonRoot(self):
+        fun = lambda x, return_g=True: np.sin(x) if not return_g else ( np.sin(x), sdiag( np.cos(x) ) )
+        x = np.array([np.pi-0.3, np.pi+0.1, 0])
+        xopt = Inverse.NewtonRoot(comments=False).root(fun,x)
+        x_true = np.array([np.pi,np.pi,0])
+        print 'Newton Root Finding'
+        print 'xopt: ', xopt
+        print 'x_true: ', x_true
+        self.assertTrue(np.linalg.norm(xopt-x_true,2) < TOL, True)
+
+if __name__ == '__main__':
+    unittest.main()