def test_lowess_3(self): "Tests lowess on typical data. part #3." (X, Y) = self.data YS = [14.811,12.115, 8.984, 9.676,10.000,11.346,11.346,11.346, 11.346,11.346,11.346,11.346,11.346,11.346,11.346,13.000, 6.734, 5.744, 5.415,18.998 ] Z = lowess(X, Y, span=0.25, nsteps=2, delta=0) assert_almost_equal(Z.outputs.fitted_values, YS, decimal=3) assert_almost_equal(Z.outputs.fitted_residuals+Z.outputs.fitted_values, Z.inputs.y, decimal=3)
def test_lowess_2(self): "Tests lowess on typical data. part #2." (X, Y) = self.data YS = [13.659,12.347,11.034, 9.722,10.511,11.300,11.300,11.300, 11.300,11.300,11.300,11.300,11.300,11.300,11.300,13.000, 6.440, 5.596, 5.456,18.998] Z = lowess(X, Y, span=0.25, nsteps=0, delta=3) assert_almost_equal(Z.outputs.fitted_values, YS, decimal=3) assert_almost_equal(Z.outputs.fitted_residuals+Z.outputs.fitted_values, Z.inputs.y, decimal=3)
def test_lowess_4(self): "Tests lowess on masked data." X = masked_values([ 1, 2, 3, 4, 5,-999, 6, 6, 6, 6, -999,-999, 6, 6, 6, 6, 6, 6, 8,-999,10,12,14,50,-999],-999) Y = marray([18, 2,15, 6,10,-999, 4, 16,11, 7, -999,-999, 3,14,17,20,12, 9,13,-999, 1, 8, 5,19,-999]) YS = [14.811,12.115, 8.984, 9.676,10.000,11.346,11.346,11.346, 11.346,11.346,11.346,11.346,11.346,11.346,11.346,13.000, 6.734, 5.744, 5.415,18.998 ] Z = lowess(X, Y, span=0.25, nsteps=2, delta=0) assert_almost_equal(Z.outputs.fitted_values.compressed(), YS, decimal=3) assert_almost_equal(Z.outputs.fitted_residuals + Z.outputs.fitted_values, Z.inputs.y, decimal=3)