Skip to content

Commit 783731b

Browse files
committed
Hide model logs in regression test output
1 parent b67dc7e commit 783731b

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

tests/regression/test_process_input_files.py

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,24 @@
3636
}
3737

3838

39+
class ProcessModelFilter(logging.Filter):
40+
def __init__(self, name=""):
41+
self.name = name or "process.models"
42+
43+
def filter(self, record):
44+
return 0 if record.levelno < logging.CRITICAL else 1
45+
46+
47+
@pytest.fixture
48+
def hide_model_logs():
49+
filter_ = ProcessModelFilter()
50+
for handler in logging.getLogger().handlers:
51+
handler.addFilter(filter_)
52+
yield
53+
for handler in logging.getLogger().handlers:
54+
handler.removeFilter(filter_)
55+
56+
3957
@dataclass
4058
class MFileVariableDifference:
4159
name: str
@@ -245,6 +263,7 @@ def test_input_file(
245263
tracked_regression_test_assets,
246264
reg_tolerance: float,
247265
opt_params_only: bool,
266+
hide_model_logs,
248267
):
249268
"""Tests each input file in the 'input_files' directory.
250269

0 commit comments

Comments
 (0)