* Added testcase wpylib.math.fitting.linear.Test_2() against numpy.polyfit().
This commit is contained in:
@@ -118,3 +118,36 @@ def Test_1():
|
|||||||
rslt = linregr2d_SZ(x,y,dy)
|
rslt = linregr2d_SZ(x,y,dy)
|
||||||
print rslt
|
print rslt
|
||||||
return rslt
|
return rslt
|
||||||
|
|
||||||
|
|
||||||
|
def Test_2():
|
||||||
|
"""Testcase 2.
|
||||||
|
Similar to testcase 1 but with all uncertainties == 1.
|
||||||
|
|
||||||
|
>>> wpylib.math.fitting.linear.Test_2()
|
||||||
|
polyfit result = -0.809999999961 -1392.318265
|
||||||
|
{'a': -1392.3182649999987,
|
||||||
|
'b': -0.81000000006627304,
|
||||||
|
'fit_method': 'linregr2d_SZ',
|
||||||
|
'fit_model': 'linear',
|
||||||
|
'sigma': 1.2247448713915881,
|
||||||
|
'sigma_a': 1.2247448713915885,
|
||||||
|
'sigma_b': 185.16401995451022}
|
||||||
|
|
||||||
|
This is to be compared with the polyfit output.
|
||||||
|
"""
|
||||||
|
from wpylib.text_tools import make_matrix as mtx
|
||||||
|
M = mtx("""
|
||||||
|
# Source: Co+ QMC/CAS(8,11)d26 cc-pwCVQZ-DK result dated 20121015
|
||||||
|
0.01 -1392.32619 1.0
|
||||||
|
0.005 -1392.32284 1.0
|
||||||
|
0.0025 -1392.31994 1.0
|
||||||
|
""")
|
||||||
|
x = M[:,0]
|
||||||
|
y = M[:,1]
|
||||||
|
dy = M[:,2]
|
||||||
|
rslt = linregr2d_SZ(x,y,dy)
|
||||||
|
|
||||||
|
polyfit_result = numpy.polyfit(x,y,deg=1,full=False)
|
||||||
|
print "polyfit result = ", polyfit_result[0], polyfit_result[1]
|
||||||
|
return rslt
|
||||||
|
|||||||
Reference in New Issue
Block a user