diff --git a/tests/exchange/test_suites/test_suite_bundle.py b/tests/exchange/test_suites/test_suite_bundle.py index 557d0744..ada0dcfd 100644 --- a/tests/exchange/test_suites/test_suite_bundle.py +++ b/tests/exchange/test_suites/test_suite_bundle.py @@ -12,6 +12,7 @@ from catalyst.exchange.utils.exchange_utils import get_candles_df from catalyst.exchange.utils.factory import get_exchange from catalyst.exchange.utils.test_utils import output_df, \ select_random_assets +from catalyst.exchange.utils.stats_utils import set_print_settings pd.set_option('display.expand_frame_repr', False) pd.set_option('precision', 8) @@ -58,6 +59,12 @@ class TestSuiteBundle: log_catcher = TestHandler() with log_catcher: + symbols = [asset.symbol for asset in assets] + print( + 'comparing data for {}/{} with {} timeframe until {}'.format( + exchange.name, symbols, freq, end_dt + ) + ) data['bundle'] = data_portal.get_history_window( assets=assets, end_dt=end_dt, @@ -66,6 +73,12 @@ class TestSuiteBundle: field='close', data_frequency=data_frequency, ) + set_print_settings() + print( + 'the bundle first / last row:\n{}'.format( + data['bundle'].iloc[[-1, 0]] + ) + ) candles = exchange.get_candles( end_dt=end_dt, freq=freq, @@ -79,6 +92,11 @@ class TestSuiteBundle: bar_count=bar_count, end_dt=end_dt, ) + print( + 'the exchange first / last row:\n{}'.format( + data['exchange'].iloc[[-1, 0]] + ) + ) for source in data: df = data[source] path, folder = output_df( @@ -126,6 +144,8 @@ class TestSuiteBundle: frequencies = exchange.get_candle_frequencies(data_frequency) freq = random.sample(frequencies, 1)[0] + freq = '15T' + bar_count = random.randint(1, 10) assets = select_random_assets(