Disable logging in rmgpy.tools.simulateTest

ReactionMechanismGenerator · Aug 5, 2019 · 2b2fe62 · 2b2fe62
1 parent 3ac99b4
commit 2b2fe62
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/rmgpy/tools/simulateTest.py b/rmgpy/tools/simulateTest.py
@@ -28,6 +28,7 @@
 #                                                                             #
 ###############################################################################
 
+import logging
 import unittest
 import os
 import os.path
@@ -36,8 +37,14 @@
 from rmgpy.tools.simulate import run_simulation
 import rmgpy
 
+
 class SimulateTest(unittest.TestCase):
 
+    def setUp(self):
+        """This method is run once before each unit test"""
+        # Disable logging
+        logging.disable(logging.WARNING)
+
     def test_minimal(self):
         """Test that we can simlulate a SimpleReactor with sensitivity"""
         folder = os.path.join(os.path.dirname(rmgpy.__file__), 'tools/data/sim/simple')
@@ -96,3 +103,6 @@ def test_mbSampled(self):
     def tearDown(self):
         import rmgpy.data.rmg
         rmgpy.data.rmg.database = None
+
+        # Reset logging
+        logging.disable(logging.NOTSET)