Improved logging

Author: lucasimi

tests/setup_logging.py

@@ -1,10 +1,11 @@
 import logging
 
 
-logging.basicConfig(
-    level=logging.INFO,
-    format='%(asctime)s %(module)s %(levelname)s: %(message)s',
-    handlers=[
-        logging.StreamHandler()  # Logs to console
-    ]
-)
+def setup_logging():
+    logging.basicConfig(
+        level=logging.INFO,
+        format='%(asctime)s %(module)s %(levelname)s: %(message)s',
+        handlers=[
+            logging.StreamHandler()  # Logs to console
+        ]
+    )

tests/test_bench_cover.py

@@ -10,7 +10,7 @@
 from tdamapper.utils.vptree_flat import VPTree as FVPT
 
 from tests.ball_tree import SkBallTree
-import tests.setup_logging
+from tests.setup_logging import setup_logging
 
 
 dist = euclidean()
@@ -25,6 +25,7 @@ def dataset(dim=10, num=1000):
 
 class TestVpSettings(unittest.TestCase):
 
+    setup_logging()
     logger = logging.getLogger(__name__)
 
     def cover(self, vpt, X, r):

tests/test_bench_metrics.py

@@ -9,7 +9,7 @@
 
 import tdamapper.utils.metrics as metrics
 
-import tests.setup_logging
+from tests.setup_logging import setup_logging
 
 
 @numba.njit(fastmath=True)
@@ -109,6 +109,7 @@ def run_bench(X):
 
 class TestBenchMetrics(unittest.TestCase):
 
+    setup_logging()
     logger = logging.getLogger(__name__)
 
     def test_bench(self):

tests/test_bench_vptree.py

@@ -10,7 +10,7 @@
 from tdamapper.utils.vptree_flat import VPTree as FVPT
 
 from tests.ball_tree import SkBallTree
-import tests.setup_logging
+from tests.setup_logging import setup_logging
 
 
 dist = euclidean()
@@ -25,12 +25,13 @@ def dataset(dim=10, num=1000):
 
 class TestBenchmark(unittest.TestCase):
 
+    setup_logging()
+    logger = logging.getLogger(__name__)
+
     eps = 0.25
 
     k = 5
 
-    logger = logging.getLogger(__name__)
-
     def test_bench(self):
         self.logger.info('==== Dataset random =============')
         self._test_compare(dataset())