From b0f2202b54772da35d6ac840dfd0024a8576a983 Mon Sep 17 00:00:00 2001 From: Frederic Fortier Date: Thu, 18 Jan 2018 14:08:11 -0500 Subject: [PATCH] BUG: fixed bundle test suite --- tests/exchange/test_suites/test_suite_bundle.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/tests/exchange/test_suites/test_suite_bundle.py b/tests/exchange/test_suites/test_suite_bundle.py index e490fc22..e9468668 100644 --- a/tests/exchange/test_suites/test_suite_bundle.py +++ b/tests/exchange/test_suites/test_suite_bundle.py @@ -21,10 +21,11 @@ pd.set_option('display.max_colwidth', 1000) class TestSuiteBundle(WithLogger, ZiplineTestCase): @staticmethod - def get_data_portal(exchange_names): + def get_data_portal(exchanges): open_calendar = get_calendar('OPEN') - asset_finder = ExchangeAssetFinder() + asset_finder = ExchangeAssetFinder(exchanges) + exchange_names = [exchange.name for exchange in exchanges] data_portal = DataPortalExchangeBacktest( exchange_names=exchange_names, asset_finder=asset_finder, @@ -107,9 +108,7 @@ class TestSuiteBundle(WithLogger, ZiplineTestCase): # ) # Type: list[Exchange] exchanges = [get_exchange('bitfinex', skip_init=True)] - data_portal = TestSuiteBundle.get_data_portal( - [exchange.name for exchange in exchanges] - ) + data_portal = TestSuiteBundle.get_data_portal(exchanges) for exchange in exchanges: exchange.init()