def _validate_training_df()

in orbit/forecaster/forecaster.py [0:0]

• 17 lines of code
• 5 McCabe index (conditional complexity)

    def _validate_training_df(self, df):
        df_columns = df.columns

        # validate date_col
        if self.date_col not in df_columns:
            raise ForecasterException(
                "DataFrame does not contain `date_col`: {}".format(self.date_col)
            )

        # validate ordering of time series
        date_array = pd.to_datetime(df[self.date_col]).reset_index(drop=True)
        if not is_ordered_datetime(date_array):
            raise ForecasterException("Datetime index must be ordered and not repeat")

        if not is_even_gap_datetime(date_array):
            warnings.warn("Datetime index is not evenly distributed")

        # validate response variable is in df
        if self.response_col not in df_columns:
            raise ForecasterException(
                "DataFrame does not contain `response_col`: {}".format(
                    self.response_col
                )
            )