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)