test_historical_medians.py 文件源码

def test_medians_for_gaps(self):
        df = pd.DataFrame({
            'date': list(pd.date_range(pd.datetime(2015, 6, 15),
                                       pd.datetime(2015, 6, 20))),
            'value': range(6)
        })
        df = df.iloc[[0, 2, 3, 4, 5]]
        guac = GuacMl(df, 'value')
        guac.make_time_series('date', prediction_length=1)
        medians = HistoricalMedians([3], guac.config, guac.logger)
        out = medians.execute(guac.data)
        self.assertTrue(np.isnan(out.df['value_median_3'].iloc[0]))
        self.assertEqual(out.df['value_median_3'].iloc[1], 0)
        self.assertEqual(out.df['value_median_3'].iloc[2], 1)
        self.assertEqual(out.df['value_median_3'].iloc[3], 2.5)
        self.assertEqual(out.df['value_median_3'].iloc[4], 3)