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_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))
Ejemplo n.º 5
0
 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)