def test_1D_example():
"""
1D example: a half-circle using newton's method (secant method for estimating derivative)
Change sign of 'y' initial condition to solve for other half-circle
"""
fvarspecs = {
"y": "t*t+y*y-r*r",
"x": "t"
}
##fnspecs = {'myauxfn': (['t'], '.5*cos(3*t)')}
dsargs = args()
##dsargs.fnspecs = fnspecs
dsargs.varspecs = fvarspecs
dsargs.algparams = {'solvemethod': 'newton', 'atol': 1e-4}
dsargs.xdomain = {'y': [-2, 2]}
dsargs.ics = {'y': 0.75}
dsargs.tdomain = [0, 2]
dsargs.pars = {'r': 2}
dsargs.vars = ['y']
dsargs.checklevel = 1
dsargs.name = 'imptest'
ex1d = ImplicitFnGen(dsargs)
assert not ex1d.defined
traj1 = ex1d.compute('traj1')
assert ex1d.defined
assert allclose(traj1(1.5)['y'], 1.3228755105)
assert traj1.dimension == 2
def test_2D_example():
"""
2D example: a quarter-sphere with linear constraint y = 3z,
using MINPACK's fsolve
"""
radius = 2.
fvarspecs2d = {"y": "t*t+y*y+z*z-r*r", "z": "y-3.*z", "x": "t"}
args2d = {
'varspecs': fvarspecs2d,
'algparams': {
'solvemethod': 'fsolve',
'atol': 1e-3
},
'xdomain': {
'y': [-2, 2],
'z': [-2, 2]
},
'ics': {
'y': 0.75,
'z': 0.9
},
'tdomain': [-2, 2],
'pars': {
'r': radius
},
'vars': ['y', 'z'],
'checklevel': 1,
'name': 'imptest2d'
}
testimp2d = ImplicitFnGen(args2d)
traj2 = testimp2d.compute('traj2')
p = traj2(1.5)
assert allclose(norm(p), radius)
assert allclose(p('y') - 3 * p('z'), 0)
assert traj2.dimension == 3
assert isparameterized(traj2)
# Test bounds checking
with pytest.raises(ValueError):
traj2(3.)
# Test saving and loading
fname = 'temp_implicit2D.pkl'
saveObjects([testimp2d, traj2], fname, force=True)
impgen, imptraj = loadObjects(fname)
assert impgen.xdomain['y'] == [-2, 2]
assert allclose(imptraj(-0.4)['y'], 1.85903)
impgen.set(pars={'r': 10.}, xdomain={'y': [-10, 10]})
imptraj2 = impgen.compute('test2')
with pytest.raises(PyDSTool_BoundsError):
imptraj2(-0.4)
impgen.set(xdomain={'z': [-5, 5]})
imptraj2 = impgen.compute('test2')
assert allclose(imptraj2(-0.4)['y'], 9.47924)
os.remove(fname)
def test_1D_example():
"""
1D example: a half-circle using newton's method (secant method for estimating derivative)
Change sign of 'y' initial condition to solve for other half-circle
"""
fvarspecs = {"y": "t*t+y*y-r*r", "x": "t"}
##fnspecs = {'myauxfn': (['t'], '.5*cos(3*t)')}
dsargs = args()
##dsargs.fnspecs = fnspecs
dsargs.varspecs = fvarspecs
dsargs.algparams = {'solvemethod': 'newton', 'atol': 1e-4}
dsargs.xdomain = {'y': [-2, 2]}
dsargs.ics = {'y': 0.75}
dsargs.tdomain = [0, 2]
dsargs.pars = {'r': 2}
dsargs.vars = ['y']
dsargs.checklevel = 1
dsargs.name = 'imptest'
ex1d = ImplicitFnGen(dsargs)
assert not ex1d.defined
traj1 = ex1d.compute('traj1')
assert ex1d.defined
assert allclose(traj1(1.5)['y'], 1.3228755105)
assert traj1.dimension == 2
def test_saveload_implicitfngen():
"""Test pickling for saving and loading 'ImplicitFnGen'"""
argsi = {
'varspecs': {
"y": "t*t+y*y-r*r",
"x": "t"
},
'algparams': {'solvemethod': 'newton', 'atol': 1e-4},
'xdomain': {'y': [-2, 2]},
'ics': {'y': 0.75},
'tdomain': [-2, 0],
'pars': {'r': 2},
'vars': ['y'],
'checklevel': 2,
'name': 'imptest',
}
testimp = ImplicitFnGen(argsi)
traj1 = testimp.compute('traj1')
saveObjects([testimp, traj1], 'temp_objects.pkl', True)
objs_imp = loadObjects('temp_objects.pkl')
assert objs_imp[0].xdomain['y'] == [-2, 2]
assert traj1(-0.4) == objs_imp[1](-0.4)
os.remove('temp_objects.pkl')
def test_saveload_implicitfngen():
"""Test pickling for saving and loading 'ImplicitFnGen'"""
argsi = {
'varspecs': {
"y": "t*t+y*y-r*r",
"x": "t"
},
'algparams': {
'solvemethod': 'newton',
'atol': 1e-4
},
'xdomain': {
'y': [-2, 2]
},
'ics': {
'y': 0.75
},
'tdomain': [-2, 0],
'pars': {
'r': 2
},
'vars': ['y'],
'checklevel': 2,
'name': 'imptest',
}
testimp = ImplicitFnGen(argsi)
traj1 = testimp.compute('traj1')
saveObjects([testimp, traj1], 'temp_objects.pkl', True)
objs_imp = loadObjects('temp_objects.pkl')
assert objs_imp[0].xdomain['y'] == [-2, 2]
assert traj1(-0.4) == objs_imp[1](-0.4)
os.remove('temp_objects.pkl')
def test_2D_example():
"""
2D example: a quarter-sphere with linear constraint y = 3z,
using MINPACK's fsolve
"""
radius = 2.
fvarspecs2d = {
"y": "t*t+y*y+z*z-r*r",
"z": "y-3.*z",
"x": "t"
}
args2d = {
'varspecs': fvarspecs2d,
'algparams': {'solvemethod': 'fsolve', 'atol': 1e-3},
'xdomain': {
'y': [-2, 2],
'z': [-2, 2]
},
'ics': {'y': 0.75, 'z': 0.9},
'tdomain': [-2, 2],
'pars': {'r': radius},
'vars': ['y', 'z'],
'checklevel': 1,
'name': 'imptest2d'
}
testimp2d = ImplicitFnGen(args2d)
traj2 = testimp2d.compute('traj2')
p = traj2(1.5)
assert allclose(norm(p), radius)
assert allclose(p('y') - 3 * p('z'), 0)
assert traj2.dimension == 3
assert isparameterized(traj2)
# Test bounds checking
with pytest.raises(ValueError):
traj2(3.)
# Test saving and loading
fname = 'temp_implicit2D.pkl'
saveObjects([testimp2d, traj2], fname, force=True)
impgen, imptraj = loadObjects(fname)
assert impgen.xdomain['y'] == [-2, 2]
assert allclose(imptraj(-0.4)['y'], 1.85903)
impgen.set(pars={'r': 10.}, xdomain={'y': [-10, 10]})
imptraj2 = impgen.compute('test2')
with pytest.raises(PyDSTool_BoundsError):
imptraj2(-0.4)
impgen.set(xdomain={'z': [-5, 5]})
imptraj2 = impgen.compute('test2')
assert allclose(imptraj2(-0.4)['y'], 9.47924)
os.remove(fname)
