Pass custom logger to iotaa (#411)

src/uwtools/logging.py

@@ -7,6 +7,8 @@
 import sys
 from typing import Any
 
+import iotaa
+
 # The logging prefix
 #
 # [YYYY-MM-DDTHH:MM:HH]_CRITICAL_
@@ -67,3 +69,4 @@ def use_logger(logger: logging.Logger) -> None:
     :param logger: The logger to log to.
     """
     log.logger = logger
+    iotaa.logset(logger)

src/uwtools/tests/test_logging.py

@@ -54,9 +54,11 @@ def test_setup_logging_verbose():
 
 def test_use_logger():
     with patch.object(uwtools.logging, "log", uwtools.logging._Logger()):
-        # Initially, uwtools logging uses the Python root logger:
-        assert uwtools.logging.log.logger == logging.getLogger()
-        # But the logger can be swapped to use a logger of choice:
-        test_logger = logging.getLogger("test-logger")
-        uwtools.logging.use_logger(test_logger)
-        assert uwtools.logging.log.logger == test_logger
+        with patch.object(uwtools.logging.iotaa, "logset") as logset:
+            # Initially, uwtools logging uses the Python root logger:
+            assert uwtools.logging.log.logger == logging.getLogger()
+            # But the logger can be swapped to use a logger of choice:
+            test_logger = logging.getLogger("test-logger")
+            uwtools.logging.use_logger(test_logger)
+            assert uwtools.logging.log.logger == test_logger
+            logset.assert_called_once_with(test_logger)