Source code for pynleq2.tests.test_solve
# -*- coding: utf-8 -*-
from __future__ import (absolute_import, division, print_function)
from pynleq2 import solve
[docs]def test_solve():
def f(x, i):
return [x[0] + (x[0] - x[1])**3/2 - 1,
(x[1] - x[0])**3/2 + x[1]], i
def j(x):
return [
[
1 + 3/2 * (x[0] - x[1])**(3-1),
-3/2 * (x[0] - x[1])**(3-1)
],
[
-3/2 * (x[1] - x[0])**(3-1),
1 + 3/2 * (x[1] - x[0])**(3 - 1)
]
]
x, ierr = solve(f, j, [0, 1])
assert abs(x[0] - 0.8411639) < 2e-7
assert abs(x[1] - 0.1588361) < 2e-7