def setUp(self): self.oil_r = ts_io.read_ts('oil', 'fpp2', as_pandas=False) self.oil_py = converters.ts_as_series(self.oil_r) self.aus_r = ts_io.read_ts('austourists', 'fpp2', as_pandas=False) self.aus_py = converters.ts_as_series(self.aus_r) self.austa_r = ts_io.read_ts('austa', 'fpp2', as_pandas=False) self.austa_py = converters.ts_as_series(self.austa_r) self.fc = importr('forecast')
def setUp(self): self.oil_r = ts_io.read_ts('oil', 'fpp', as_pandas=False) self.oil_py = converters.ts_as_series(self.oil_r) self.aus_r = ts_io.read_ts('austourists', 'fpp', as_pandas=False) self.aus_py = converters.ts_as_series(self.aus_r) self.austa_r = ts_io.read_ts('austa', 'fpp', as_pandas=False) self.austa_py = converters.ts_as_series(self.austa_r) self.fc = importr('forecast')
def test_ts_as_series(self): oil = converters.ts_as_series(self.oil_ts) self.assertEqual(list(oil.index), range(1965, 2011)) self.assertAlmostEqual(oil[1965], 111.0091, places=3) self.assertAlmostEqual(oil[2010], 467.7724, places=3) aus = converters.ts_as_series(self.aus_ts) self.assertAlmostEqual(aus[(1999, 1)], 30.0525, places=3) self.assertAlmostEqual(aus[(2010, 4)], 47.9137, places=3) self.assertEqual(aus[2010].shape, (4, )) self.assertEqual(type(aus.index), pandas.core.index.MultiIndex) self.assertEqual(aus.index[0], (1999, 1)) self.assertEqual(aus.index[-1], (2010, 4))
def test_tsclean(self): gold_py = converters.ts_as_series(converters.ts(self.gold)) clean_py = wrappers.tsclean(gold_py) self.assertFalse(clean_py.isnull().any()) clean_r = self.fc.tsclean(converters.ts(self.gold)) self.assertAlmostEqual(clean_py[770], clean_r.rx(770), places=3)
def test_tsclean(self): clean_gold = wrappers.tsclean(converters.ts_as_series(self.gold)) self.assertFalse(clean_gold.isnull().any()) self.assertAlmostEqual(clean_gold[56], 333.250, places=3) self.assertAlmostEqual(clean_gold[419],394.5, places=3) self.assertAlmostEqual(clean_gold[604], 476.6000, places=3)
def test_tsclean(self): gold_py = converters.ts_as_series(self.gold) clean_py = wrappers.tsclean(gold_py) self.assertFalse(clean_py.isnull().any()) clean_r = self.fc.tsclean(self.gold) self.assertAlmostEqual(clean_py[770], clean_r.rx(770), places=3)