From ed208eb34b91cf6186fff1b6bfebbc931e893f6a Mon Sep 17 00:00:00 2001 From: beam Date: Thu, 29 Oct 2026 08:33:56 +0000 Subject: [PATCH] beam: guard module-level sys.exit in two test files so pytest collection succeeds --- wizard-work/beam/test_bisect_bounds.py | 3 ++- wizard-work/beam/test_safe_eval.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/wizard-work/beam/test_bisect_bounds.py b/wizard-work/beam/test_bisect_bounds.py index 4f1188a..4552b0c 100644 --- a/wizard-work/beam/test_bisect_bounds.py +++ b/wizard-work/beam/test_bisect_bounds.py @@ -50,4 +50,5 @@ for _ in range(2000): check("randomized", ok) print("ALL" if ok else "SOME FAILED") -sys.exit(0 if ok else 1) +if __name__ == "__main__": + sys.exit(0 if ok else 1) diff --git a/wizard-work/beam/test_safe_eval.py b/wizard-work/beam/test_safe_eval.py index cd57ada..c136c16 100644 --- a/wizard-work/beam/test_safe_eval.py +++ b/wizard-work/beam/test_safe_eval.py @@ -31,7 +31,7 @@ for e in errs: try: evaluate(e) print("FAIL no error:", repr(e)) - sys.exit(1) + raise AssertionError("test failed") except ValueError: print("ERR-OK", repr(e)) @@ -39,7 +39,7 @@ for e in errs: for bad in ["__builtins__", "eval('1')", "().__class__"]: try: evaluate(bad) - print("FAIL no error:", repr(bad)); sys.exit(1) + print("FAIL no error:", repr(bad)); raise AssertionError("test failed") except ValueError: print("INJ-OK", repr(bad))