Tested ingestion after refactoring

This commit is contained in:
fredfortier
2017-10-05 21:03:39 -04:00
parent 874968bbbb
commit 50f075792c
3 changed files with 18 additions and 7 deletions
+8 -4
View File
@@ -141,12 +141,16 @@ class Exchange:
return symbols
def get_assets(self, symbols):
def get_assets(self, symbols=None):
assets = []
for symbol in symbols:
asset = self.get_asset(symbol)
assets.append(asset)
if symbols is not None:
for symbol in symbols:
asset = self.get_asset(symbol)
assets.append(asset)
else:
for key in self.assets:
assets.append(self.assets[key])
return assets
+8 -2
View File
@@ -39,8 +39,14 @@ class ExchangeBundle:
self._reader = None
def get_assets(self, include_symbols, exclude_symbols):
# TODO: filter assets
return self.exchange.assets
# TODO: filter exclude symbols assets
if include_symbols is not None:
include_symbols_list = include_symbols.split(',')
return self.exchange.get_assets(include_symbols_list)
else:
return self.exchange.get_assets()
def get_adj_dates(self, start, end):
now = pd.Timestamp.utcnow()
+2 -1
View File
@@ -18,10 +18,11 @@ class ExchangeBundleTestCase:
exchange_bundle = ExchangeBundle(
exchange_name=exchange_name,
data_frequency='minute',
include_symbols=None,
include_symbols='neo_btc',
exclude_symbols=None,
start=start,
end=end,
show_progress=True
)
exchange_bundle.ingest()
pass