diff --git a/tests/test_algorithm.py b/tests/test_algorithm.py index 4179637f..de9f12e0 100644 --- a/tests/test_algorithm.py +++ b/tests/test_algorithm.py @@ -25,7 +25,9 @@ import pandas as pd from zipline.utils.test_utils import ( nullctx, - setup_logger + setup_logger, + teardown_logger + ) import zipline.utils.factory as factory import zipline.utils.simfactory as simfactory @@ -139,6 +141,9 @@ class TestMiscellaneousAPI(TestCase): concurrent=True, ) + def tearDown(self): + teardown_logger(self) + def test_get_environment(self): expected_env = { 'arena': 'backtest', @@ -306,6 +311,9 @@ class TestTransformAlgorithm(TestCase): self.panel_source, self.panel = \ factory.create_test_panel_source(self.sim_params) + def tearDown(self): + teardown_logger(self) + def test_source_as_input(self): algo = TestRegisterTransformAlgorithm( sim_params=self.sim_params, @@ -427,7 +435,6 @@ class TestPositions(TestCase): def setUp(self): setup_logger(self) self.sim_params = factory.create_simulation_parameters(num_days=4) - setup_logger(self) trade_history = factory.create_trade_history( 1, @@ -441,6 +448,9 @@ class TestPositions(TestCase): self.df_source, self.df = \ factory.create_test_df_source(self.sim_params) + def tearDown(self): + teardown_logger(self); + def test_empty_portfolio(self): algo = EmptyPositionsAlgorithm(sim_params=self.sim_params) daily_stats = algo.run(self.df) @@ -488,6 +498,9 @@ class TestAlgoScript(TestCase): 'sid': 0, } + def tearDown(self): + teardown_logger(self); + def test_noop(self): algo = TradingAlgorithm(initialize=initialize_noop, handle_data=handle_data_noop) diff --git a/tests/test_batchtransform.py b/tests/test_batchtransform.py index 6b89c8e5..cc70adf1 100644 --- a/tests/test_batchtransform.py +++ b/tests/test_batchtransform.py @@ -22,7 +22,7 @@ import pandas as pd from datetime import datetime from unittest import TestCase -from zipline.utils.test_utils import setup_logger +from zipline.utils.test_utils import setup_logger, teardown_logger from zipline.sources.data_source import DataSource import zipline.utils.factory as factory @@ -127,6 +127,7 @@ class TestChangeOfSids(TestCase): class TestBatchTransformMinutely(TestCase): def setUp(self): + setup_logger(self) start = pd.datetime(1990, 1, 3, 0, 0, 0, 0, pytz.utc) end = pd.datetime(1990, 1, 8, 0, 0, 0, 0, pytz.utc) self.sim_params = factory.create_simulation_parameters( @@ -135,10 +136,12 @@ class TestBatchTransformMinutely(TestCase): ) self.sim_params.emission_rate = 'daily' self.sim_params.data_frequency = 'minute' - setup_logger(self) self.source, self.df = \ factory.create_test_df_source(bars='minute') + def tearDown(self): + teardown_logger(self); + def test_core(self): algo = BatchTransformAlgorithmMinute(sim_params=self.sim_params) algo.run(self.source) @@ -159,14 +162,17 @@ class TestBatchTransformMinutely(TestCase): class TestBatchTransform(TestCase): def setUp(self): + setup_logger(self) self.sim_params = factory.create_simulation_parameters( start=datetime(1990, 1, 1, tzinfo=pytz.utc), end=datetime(1990, 1, 8, tzinfo=pytz.utc) ) - setup_logger(self) self.source, self.df = \ factory.create_test_df_source(self.sim_params) + def tearDown(self): + teardown_logger(self); + def test_core_functionality(self): algo = BatchTransformAlgorithm(sim_params=self.sim_params) algo.run(self.source) diff --git a/tests/test_security_list.py b/tests/test_security_list.py index 8596444d..3a041d74 100644 --- a/tests/test_security_list.py +++ b/tests/test_security_list.py @@ -7,7 +7,7 @@ from zipline.algorithm import TradingAlgorithm from zipline.errors import TradingControlViolation from zipline.sources import SpecificEquityTrades from zipline.utils.test_utils import ( - setup_logger, security_list_copy, add_security_data) + setup_logger, teardown_logger, security_list_copy, add_security_data) from zipline.utils import factory from zipline.utils.security_list import ( SecurityListSet, load_from_directory) @@ -66,6 +66,9 @@ class SecurityListTestCase(TestCase): setup_logger(self) + def tearDown(self): + teardown_logger(self); + def test_iterate_over_rl(self): sim_params = factory.create_simulation_parameters( start=list(LEVERAGED_ETFS.keys())[0], num_days=4) diff --git a/tests/test_transforms_talib.py b/tests/test_transforms_talib.py index 268fa3a8..f0d1c17c 100644 --- a/tests/test_transforms_talib.py +++ b/tests/test_transforms_talib.py @@ -20,7 +20,7 @@ import pandas as pd from datetime import timedelta, datetime from unittest import TestCase, skip -from zipline.utils.test_utils import setup_logger +from zipline.utils.test_utils import setup_logger, teardown_logger import zipline.utils.factory as factory @@ -39,6 +39,9 @@ class TestTALIB(TestCase): self.source, self.panel = \ factory.create_test_panel_ohlc_source(sim_params) + def tearDown(self): + teardown_logger(self) + @skip def test_talib_with_default_params(self): BLACKLIST = ['make_transform', 'BatchTransform',