def test_prophet_valid(self): pytest.importorskip("prophet") df = proc.prophet(df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.9) columns = {column for column in df.columns} assert columns == { DTTM_ALIAS, "a__yhat", "a__yhat_upper", "a__yhat_lower", "a", "b__yhat", "b__yhat_upper", "b__yhat_lower", "b", } assert df[DTTM_ALIAS].iloc[0].to_pydatetime() == datetime(2018, 12, 31) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 3, 31) assert len(df) == 7 df = proc.prophet(df=prophet_df, time_grain="P1M", periods=5, confidence_interval=0.9) assert df[DTTM_ALIAS].iloc[0].to_pydatetime() == datetime(2018, 12, 31) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 5, 31) assert len(df) == 9
def test_prophet_import(self): prophet = find_spec("prophet") if prophet is None: with pytest.raises(QueryObjectValidationError): proc.prophet( df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.9 )
def test_prophet_import(): dynamic_module = find_spec("prophet") if dynamic_module is None: with pytest.raises(QueryObjectValidationError): prophet(df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.9)
def test_prophet_incorrect_time_grain(): with pytest.raises(QueryObjectValidationError): prophet( df=prophet_df, time_grain="yearly", periods=10, confidence_interval=0.8, )
def test_prophet_incorrect_time_grain(): with pytest.raises(InvalidPostProcessingError): prophet( df=prophet_df, time_grain="yearly", periods=10, confidence_interval=0.8, )
def test_prophet_import(): dynamic_module = find_spec("prophet") if dynamic_module is None: with pytest.raises(InvalidPostProcessingError): prophet(df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.9)
def test_prophet_missing_temporal_column(): df = prophet_df.drop(DTTM_ALIAS, axis=1) with pytest.raises(QueryObjectValidationError): prophet( df=df, time_grain="P1M", periods=3, confidence_interval=0.9, )
def test_prophet_incorrect_confidence_interval(): with pytest.raises(QueryObjectValidationError): prophet( df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.0, ) with pytest.raises(QueryObjectValidationError): prophet( df=prophet_df, time_grain="P1M", periods=3, confidence_interval=1.0, )
def test_prophet_incorrect_confidence_interval(): with pytest.raises(InvalidPostProcessingError): prophet( df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.0, ) with pytest.raises(InvalidPostProcessingError): prophet( df=prophet_df, time_grain="P1M", periods=3, confidence_interval=1.0, )
def test_prophet_valid(): pytest.importorskip("prophet") df = prophet(df=prophet_df, time_grain="P1M", periods=3, confidence_interval=0.9) columns = {column for column in df.columns} assert columns == { DTTM_ALIAS, "a__yhat", "a__yhat_upper", "a__yhat_lower", "a", "b__yhat", "b__yhat_upper", "b__yhat_lower", "b", } assert df[DTTM_ALIAS].iloc[0].to_pydatetime() == datetime(2018, 12, 31) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 3, 31) assert len(df) == 7 df = prophet(df=prophet_df, time_grain="P1M", periods=5, confidence_interval=0.9) assert df[DTTM_ALIAS].iloc[0].to_pydatetime() == datetime(2018, 12, 31) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 5, 31) assert len(df) == 9 df = prophet( df=pd.DataFrame({ "__timestamp": [datetime(2022, 1, 2), datetime(2022, 1, 9)], "x": [1, 1], }), time_grain="P1W", periods=1, confidence_interval=0.9, ) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 1, 16) assert len(df) == 3 df = prophet( df=pd.DataFrame({ "__timestamp": [datetime(2022, 1, 2), datetime(2022, 1, 9)], "x": [1, 1], }), time_grain="1969-12-28T00:00:00Z/P1W", periods=1, confidence_interval=0.9, ) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 1, 16) assert len(df) == 3 df = prophet( df=pd.DataFrame({ "__timestamp": [datetime(2022, 1, 3), datetime(2022, 1, 10)], "x": [1, 1], }), time_grain="1969-12-29T00:00:00Z/P1W", periods=1, confidence_interval=0.9, ) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 1, 17) assert len(df) == 3 df = prophet( df=pd.DataFrame({ "__timestamp": [datetime(2022, 1, 8), datetime(2022, 1, 15)], "x": [1, 1], }), time_grain="P1W/1970-01-03T00:00:00Z", periods=1, confidence_interval=0.9, ) assert df[DTTM_ALIAS].iloc[-1].to_pydatetime() == datetime(2022, 1, 22) assert len(df) == 3