def test_single():
data = fetch.ohlcv('AAPL')
open = Open()(data)
high = High()(data)
low = Low()(data)
close = Close()(data)
volume = Volume()(data)
assert np.all(open == data['open'])
assert np.all(high == data['high'])
assert np.all(low == data['low'])
assert np.all(close == data['close'])
assert np.all(volume == data['volume'])
def test_multi():
data = fetch.ohlcv('AAPL TSLA')
open = Open()(data)
high = High()(data)
low = Low()(data)
close = Close()(data)
volume = Volume()(data)
for symbol in data.columns.levels[0]:
assert np.all(open[symbol] == data[symbol, 'open'])
assert np.all(high[symbol] == data[symbol, 'high'])
assert np.all(low[symbol] == data[symbol, 'low'])
assert np.all(close[symbol] == data[symbol, 'close'])
assert np.all(volume[symbol] == data[symbol, 'volume'])
def test_list(self):
fetched = fetch.ohlcv(['AAPL', 'TSLA'], period='1mo', interval='1d')
self.assert_monthish(fetched)
self.assert_ohlcv(fetched)
def test_spaced(self):
fetched = fetch.ohlcv('AAPL TSLA', period='1mo', interval='1d')
self.assert_monthish(fetched)
self.assert_ohlcv(fetched)
def test_single(self):
fetched = fetch.ohlcv('AAPL', period='1mo', interval='1d')
self.assert_monthish(fetched)
self.assert_ohlcv(fetched)
def test_multi():
data = fetch.ohlcv('AAPL TSLA')
log_close = Close().log()(data)
for symbol in data.columns.levels[0]:
assert_single(data[symbol], log_close[symbol])
def test_single():
data = fetch.ohlcv('AAPL')
log_close = Close().log()(data)
assert_single(data, log_close)