mirror of
https://github.com/wassname/simpeg.git
synced 2026-06-27 22:23:39 +08:00
Rowan Cockett
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
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 Optimization
|
|
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 = Optimization.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 = Optimization.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 = Optimization.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 = Optimization.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 = Optimization.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()
|