def test_creating_prediction_df_basic(self):
df_builder = data_builder.DataFrameBuilder({})
y_df = df_builder.create_df(self.df, "y")
np.testing.assert_array_equal(["ds"], y_df.columns)
y1_df = df_builder.create_df(self.df, "y1")
np.testing.assert_array_equal(["ds"], y1_df.columns)
def test_creating_training_df_basic(self):
df_builder = data_builder.DataFrameBuilder({})
y_df = df_builder.create_df(self.df, "y", train=True)
np.testing.assert_array_equal(["ds", "y"], y_df.columns)
y1_df = df_builder.create_df(self.df, "y1", train=True)
np.testing.assert_array_equal(["ds", "y"], y1_df.columns)
def test_regressors_prediction_df(self):
regressors = {"y": ["y1"]}
df_builder = data_builder.DataFrameBuilder(regressors)
y_df = df_builder.create_df(self.df, "y")
np.testing.assert_array_equal(["ds", "y1"], y_df.columns)
y1_df = df_builder.create_df(self.df, "y1")
np.testing.assert_array_equal(["ds"], y1_df.columns)
def test_regressors_training_df(self):
regressors = {"y": ["y1"]}
df_builder = data_builder.DataFrameBuilder(regressors)
y_df = df_builder.create_df(self.df, "y", train=True)
np.testing.assert_array_equal(["ds", "y", "y1"], y_df.columns)
y1_df = df_builder.create_df(self.df, "y1", train=True)
np.testing.assert_array_equal(["ds", "y"], y1_df.columns)
def test_capacity_floor_prediction_df(self):
self.df["floor_y"] = 50
self.df["cap_y"] = 500
self.df["cap_y1"] = 500
df_builder = data_builder.DataFrameBuilder({})
y_df = df_builder.create_df(self.df, "y")
np.testing.assert_array_equal(["ds", "cap", "floor"], y_df.columns)
y1_df = df_builder.create_df(self.df, "y1")
np.testing.assert_array_equal(["ds", "cap"], y1_df.columns)
def test_add_regressor(self):
df_builder = data_builder.DataFrameBuilder({})
y_df = df_builder.create_df(self.df, "y")
np.testing.assert_array_equal(["ds"], y_df.columns)
df_builder.add_regressor("y1", ["y"])
yr_df = df_builder.create_df(self.df, "y")
np.testing.assert_array_equal(["ds", "y1"], yr_df.columns)
y1_df = df_builder.create_df(self.df, "y1")
np.testing.assert_array_equal(["ds"], y1_df.columns)