Skip to content

Commit

Permalink
Test for polyomial derivaties
Browse files Browse the repository at this point in the history
  • Loading branch information
jeffrey-hokanson committed Mar 26, 2019
1 parent e129a86 commit bbfaacb
Showing 1 changed file with 26 additions and 0 deletions.
26 changes: 26 additions & 0 deletions tests/test_poly.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,26 @@
import numpy as np

from psdr import PolynomialApproximation, PolynomialFunction, LegendreTensorBasis
from checkder import *

def test_poly_der(dimension = 3, degree = 5):
coef = np.random.randn(len(LegendreTensorBasis(dimension, degree)))
pf = PolynomialFunction(dimension, degree, coef)

x = np.random.randn(dimension)
print x
print pf.eval(x)
print pf.grad(x)

assert check_derivative(x, pf.eval, lambda x: pf.grad(x).flatten() ) < 1e-7

def test_poly_hess(dimension = 3, degree = 5):
coef = np.random.randn(len(LegendreTensorBasis(dimension, degree)))
pf = PolynomialFunction(dimension, degree, coef)

x = np.random.randn(dimension)
print x
print pf.eval(x)
print pf.hessian(x)

assert check_hessian(x, pf.eval, lambda x: pf.hessian(x).reshape(dimension, dimension) ) < 1e-5

0 comments on commit bbfaacb

Please sign in to comment.