def test__get_quotes(self): start = dt.datetime(2006, 1, 1, tzinfo = pytz.utc) end = dt.datetime(2010, 12, 31, tzinfo = pytz.utc) feeds = DataFeed() data = feeds.quotes(tickers, start_date = start, end_date = end) assert(data.index.tzinfo) assert(isinstance(data, pd.DataFrame))
def test__get_tickers(self): feeds = DataFeed() tickers = feeds.random_stocks(6) assert(len(tickers) == 6)