def test_load_from_cache(self): cache = FinancialDataTimeSeriesCache(gets_data=None, database=None) test_date, test_price = datetime.datetime(2012, 12, 3, tzinfo=pytz.UTC), 100 cache.get = lambda *args, **kwargs : [(test_date, test_price), (datetime.datetime(2012, 12, 4, tzinfo=pytz.UTC), 101), (datetime.datetime(2012, 12, 5, tzinfo=pytz.UTC), 102), ] symbol = 'ABC' df = cache.load_from_cache(start=datetime.datetime(2012, 11, 30, tzinfo=pytz.UTC), end=datetime.datetime(2013, 1, 1, tzinfo=pytz.UTC), stocks=[symbol]) self.assertIsInstance(df, pd.DataFrame) self.assertIn(symbol, df.keys()) self.assertEqual(df[symbol][test_date], test_price)
def test_load_from_cache(self): cache = FinancialDataTimeSeriesCache(gets_data=None, database=None) test_date, test_price = datetime.datetime(2012, 12, 3, tzinfo=pytz.UTC), 100 cache.get = lambda *args, **kwargs: [ (test_date, test_price), (datetime.datetime(2012, 12, 4, tzinfo=pytz.UTC), 101), (datetime.datetime(2012, 12, 5, tzinfo=pytz.UTC), 102), ] symbol = 'ABC' df = cache.load_from_cache(start=datetime.datetime(2012, 11, 30, tzinfo=pytz.UTC), end=datetime.datetime(2013, 1, 1, tzinfo=pytz.UTC), stocks=[symbol]) self.assertIsInstance(df, pd.DataFrame) self.assertIn(symbol, df.keys()) self.assertEqual(df[symbol][test_date], test_price)