Updated doctests

master
pbrod 9 years ago
parent 6f67e46cb6
commit 33367e4c71

@ -19,6 +19,7 @@
# !/usr/bin/env python # !/usr/bin/env python
from __future__ import absolute_import from __future__ import absolute_import
import warnings # @UnusedImport import warnings # @UnusedImport
from functools import reduce
from numpy.polynomial import polyutils as pu from numpy.polynomial import polyutils as pu
from .plotbackend import plotbackend as plt from .plotbackend import plotbackend as plt
import numpy as np import numpy as np
@ -236,8 +237,8 @@ def polydeg(x, y):
>>> x = np.linspace(0,10,300) >>> x = np.linspace(0,10,300)
>>> y = np.sin(x ** 3 / 100) ** 2 + 0.05 * np.random.randn(x.size) >>> y = np.sin(x ** 3 / 100) ** 2 + 0.05 * np.random.randn(x.size)
>>> n = polydeg(x,y) >>> n = polydeg(x,y)
>>> n >>> 18 < n < 24
21 True
ys = orthofit(x,y,n); ys = orthofit(x,y,n);
plt.plot(x, y, '.', x, ys, 'k') plt.plot(x, y, '.', x, ys, 'k')
@ -266,7 +267,7 @@ def polydeg(x, y):
# required to take AIC noise into account and to ensure that this minimum # required to take AIC noise into account and to ensure that this minimum
# is a (likely) global minimum. # is a (likely) global minimum.
while nit < 6: while nit < 8:
p = orthofit(x, y, n) p = orthofit(x, y, n)
ys = orthoval(p, x) ys = orthoval(p, x)
# -- Akaike's Information Criterion # -- Akaike's Information Criterion

Loading…
Cancel
Save