Feature/issue184 loguru debug

logger_test.py

@@ -0,0 +1,23 @@
+"""This test is to test that logging works properly when importing torchquad."""
+
+import unittest
+from contextlib import contextmanager
+from loguru import logger
+
+@contextmanager
+def capture_logs(level = "INFO", format="{message}"):
+    """"Capture loguru-based logs."""
+    output = []
+    handler_id = logger.add(output.append, level = level, format = format)
+
+    yield output
+    logger.remove(handler_id)
+
+class TestLogger(unittest.TestCase):
+
+    def test_logging_with_tq(self):
+        import torchquad
+        with capture_logs() as cap_log:
+            logger.info("This message should print.")
+        self.assertEqual(cap_log,["This message should print.\n"])
+

torchquad/__init__.py

@@ -1,5 +1,7 @@
 import os
 from loguru import logger
+#Called to prevent library logs from being mixed with thoes of the user.
+logger.disable("torchquad") 
 
 # TODO: Currently this is the way to expose to the docs
 # hopefully changes with setup.py
@@ -46,5 +48,4 @@
     "_deployment_test",
 ]
 
-set_log_level(os.environ.get("TORCHQUAD_LOG_LEVEL", "WARNING"))
 logger.info("Initializing torchquad.")