From cbc2ed2aaf232be472b3e4bfe6d27ad1d81875d5 Mon Sep 17 00:00:00 2001 From: Frederic Fortier Date: Mon, 5 Mar 2018 21:35:38 -0500 Subject: [PATCH] BLD: fixes small issues when testing --- tests/exchange/test_bundle.py | 16 ++++++++-------- tests/exchange/test_suites/test_suite_bundle.py | 4 ++-- 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/tests/exchange/test_bundle.py b/tests/exchange/test_bundle.py index 4779d25d..ea739249 100644 --- a/tests/exchange/test_bundle.py +++ b/tests/exchange/test_bundle.py @@ -48,13 +48,13 @@ class TestExchangeBundle: exchange = get_exchange(exchange_name) exchange_bundle = ExchangeBundle(exchange_name) assets = [ - exchange.get_asset('eng_eth') + exchange.get_asset('bat_btc') ] - # start = pd.to_datetime('2018-02-01', utc=True) - # end = pd.to_datetime('2018-02-28', utc=True) - start = None - end = None + start = pd.to_datetime('2018-03-01', utc=True) + end = pd.to_datetime('2018-03-04', utc=True) + # start = None + # end = None log.info('ingesting exchange bundle {}'.format(exchange_name)) exchange_bundle.ingest( @@ -62,8 +62,8 @@ class TestExchangeBundle: include_symbols=','.join([asset.symbol for asset in assets]), # include_symbols=None, exclude_symbols=None, - start=None, - end=None, + start=start, + end=end, show_progress=False, show_breakdown=False ) @@ -83,7 +83,7 @@ class TestExchangeBundle: dx = get_df_from_arrays(arrays[0], periods) set_print_settings() print('found {} rows for last ingestion:\n{}\n{}'.format( - len(dx), dx.head(1000), dx.tail(1000) + len(dx), dx.head(10), dx.tail(10) )) pass diff --git a/tests/exchange/test_suites/test_suite_bundle.py b/tests/exchange/test_suites/test_suite_bundle.py index 3c828729..b256c441 100644 --- a/tests/exchange/test_suites/test_suite_bundle.py +++ b/tests/exchange/test_suites/test_suite_bundle.py @@ -215,8 +215,8 @@ class TestSuiteBundle: # assets = select_random_assets( # exchange.assets, asset_population # ) - assets = [exchange.get_asset('eng_eth')] - end_dt = None + assets = [exchange.get_asset('bat_btc')] + end_dt = pd.to_datetime('2018-03-03', utc=True) for asset in assets: attribute = 'end_{}'.format(data_frequency) asset_end_dt = getattr(asset, attribute)