diff --git a/afids_utils/tests/helpers.py b/afids_utils/tests/helpers.py index 96d0cd40..b7a38015 100644 --- a/afids_utils/tests/helpers.py +++ b/afids_utils/tests/helpers.py @@ -36,3 +36,10 @@ def inner(callable: _T, /) -> _T: )(callable) return inner + + +def slow_generation(callable: _T, /) -> _T: + """Disable health check for slow data generation tests""" + return settings( + suppress_health_check=[HealthCheck.too_slow], + )(callable)