from __future__ import annotations
from collections.abc import Iterable
from typing import Callable
from .pseudo_traceback import _build_pseudo_trace_str
should_use_color = False
COLOR_RED = "\x1b[31m"
COLOR_RESET = "\x1b[0m"
_failures: list[str] = []
_stop_on_fail = False
_default_max_fail = None
_default_max_report = None
_default_max_tb = 1
_max_fail: int | None = _default_max_fail
_max_report: int | None = _default_max_report
_max_tb = _default_max_tb
_num_failures = 0
_fail_function: Callable[[str], None] | None = None
_showlocals = False
def clear_failures() -> None:
# gets called at the beginning of each test function
global _failures, _num_failures
global _max_fail, _max_report, _max_tb
_failures = []
_num_failures = 0
_max_fail = _default_max_fail
_max_report = _default_max_report
_max_tb = _default_max_tb
def any_failures() -> bool:
return bool(get_failures())
def get_failures() -> list[str]:
return _failures
def log_failure(
msg: object = "", check_str: str = "", tb: Iterable[str] | None = None
) -> None:
global _num_failures
__tracebackhide__ = True
_num_failures += 1
msg = str(msg).strip()
if check_str:
msg = f"{msg}: {check_str}"
if (_max_report is None) or (_num_failures <= _max_report):
if _num_failures <= _max_tb:
pseudo_trace_str = _build_pseudo_trace_str(_showlocals,
tb,
should_use_color)
msg = f"{msg}\n{pseudo_trace_str}"
if should_use_color:
msg = f"{COLOR_RED}FAILURE: {COLOR_RESET}{msg}"
else:
msg = f"FAILURE: {msg}"
_failures.append(msg)
if _fail_function:
_fail_function(str(msg))
if _max_fail and (_num_failures >= _max_fail):
assert_msg = f"pytest-check max fail of {_num_failures} reached"
assert _num_failures < _max_fail, assert_msg
if _stop_on_fail:
assert False, "Stopping on first failure"
|