def test_stl_2(self):
"Tests a robust STL."
(co2_data, co2_results, parameters) = self.d
co2_fitted = stl(co2_data, robust=True, **parameters).outputs
assert_almost_equal(co2_fitted.seasonal, co2_results[4], 6)
assert_almost_equal(co2_fitted.trend, co2_results[5], 6)
assert_almost_equal(co2_fitted.weights, co2_results[6], 6)
def test_stl_1(self):
"Tests a classic STL."
(co2_data, co2_results, parameters) = self.d
co2_fitted = stl(co2_data, robust=False, **parameters).outputs
assert_almost_equal(co2_fitted.seasonal, co2_results[0], 6)
assert_almost_equal(co2_fitted.trend, co2_results[1], 6)
assert_almost_equal(co2_fitted.weights, co2_results[2], 6)
