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)