def test_noinfo2(self) :
""" Run AR all equations under no info """
ss = Chou2006()
ss.exptype = "noinfo"
ss.equations = [1,2,3,4]
ar = ARSolver(ss)
ar.solve(maxiter=10000,tol=10e-7)
eq_(len(ar.all_exp_art),1)
print ar.all_exp_art[0]['eqns']
eq_(len(ar.all_exp_art[0]['eqns']),4)
art = ar.all_exp_art[0]['eqns'][0]
sol = {'alpha': 12,
'beta': 10,
'g': np.array([ 0,0,-0.8,0]),
'h': np.array([ 0.5,0,0,0])}
def test_fullinfo2(self) :
""" RUN AR (slow) eqn 2 under full info """
ss = Chou2006()
ss.exptype = "fullinfo"
ss.equations = [2]
ar = ARSolver(ss)
ar.solve(maxiter=10000,tol=10e-6)
assert len(ar.all_exp_art) == 1
assert len(ar.all_exp_art[0]['eqns']) == 1
art = ar.all_exp_art[0]['eqns'][0]
sol = {'alpha': 7.8801765469678156,
'beta': 2.8893110127454427,
'g': np.array([ 0.51119746]),
'h': np.array([ 0.76771705])}
params = art.params[-1]
assert np.allclose(sol['alpha'],params['alpha'],atol=1)
assert np.allclose(sol['beta'],params['beta'],atol=1.5)
assert np.allclose(sol['g'],params['g'],atol=0.5)
assert np.allclose(sol['h'],params['h'],atol=0.5)
def test_noinfo1(self) :
""" Run AR eqn 1 under no info """
ss = Chou2006()
ss.exptype = "noinfo"
ss.equations = [1]
ar = ARSolver(ss)
ar.solve(maxiter=10000,tol=10e-7)
assert len(ar.all_exp_art) == 1
assert len(ar.all_exp_art[0]['eqns']) == 1
art = ar.all_exp_art[0]['eqns'][0]
sol = {'alpha': 12,
'beta': 10,
'g': np.array([ 0,0,-0.8,0]),
'h': np.array([ 0.5,0,0,0])}
params = art.params[-1]
assert np.allclose(sol['alpha'],params['alpha'],atol=0.5)
assert np.allclose(sol['beta'],params['beta'],atol=0.5)
assert np.allclose(sol['g'],params['g'],atol=0.2)
assert np.allclose(sol['h'],params['h'],atol=0.2)
def test_fullinfo1(self) :
""" RUN AR (slow) eqn 1 under full info """
ss = Chou2006()
ss.exptype = "fullinfo"
ss.equations = [1]
ar = ARSolver(ss)
ar.solve(maxiter=10000,tol=10e-6)
assert len(ar.all_exp_art) == 1
assert len(ar.all_exp_art[0]['eqns']) == 1
art = ar.all_exp_art[0]['eqns'][0]
sol = {'alpha': 11.177018643003306,
'beta': 8.7027543289527056,
'g': np.array([-1.0236953]),
'h': np.array([ 0.62268492])}
params = art.params[-1]
assert np.allclose(sol['alpha'],params['alpha'],atol=1)
assert np.allclose(sol['beta'],params['beta'],atol=1.5)
assert np.allclose(sol['g'],params['g'],atol=0.5)
assert np.allclose(sol['h'],params['h'],atol=0.5)
