Every setup_logger needs a teardown_logger

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)

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)

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)

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',