fix: add setup_logging in test

Electrostatics · Nov 30, 2024 · 2cec21d · 2cec21d
1 parent 666ea2a
commit 2cec21d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/common.py b/tests/common.py
@@ -212,6 +212,8 @@ def run_pdb2pqr_for_tests(
     compare_resnames=False,
 ):
     """Basic code for invoking PDB2PQR."""
+    from pdb2pqr import io
+
     if output_pqr is None:
         hash_str = f"{args}{input_pdb}"
         hash_ = hashlib.sha1(hash_str.encode("UTF-8")).hexdigest()
@@ -220,6 +222,7 @@ def run_pdb2pqr_for_tests(
     _LOGGER.debug(f"Writing output to {output_pqr}")
     arg_str = f"{args} {input_pdb} {output_pqr}"
     args = PARSER.parse_args(arg_str.split())
+    io.setup_logger(args.output_pqr, args.log_level)
     _ = main_driver(args)
     if expected_pqr is not None:
         compare_pqr(