Make debugging failing tests (check and update expect output values) easier 🔥 (huggingface#40727)

* fix

* fix

---------

Co-authored-by: ydshieh <[email protected]>

Author: ydshieh

conftest.py

@@ -16,6 +16,7 @@
 # by pytest before any tests are run
 
 import doctest
+import os
 import sys
 import warnings
 from os.path import abspath, dirname, join
@@ -27,6 +28,7 @@
     HfDoctestModule,
     HfDocTestParser,
     is_torch_available,
+    patch_testing_methods_to_collect_info,
     patch_torch_compile_force_graph,
 )
 
@@ -145,3 +147,8 @@ def check_output(self, want, got, optionflags):
     # patch `torch.compile`: if `TORCH_COMPILE_FORCE_FULLGRAPH=1` (or values considered as true, e.g. yes, y, etc.),
     # the patched version will always run with `fullgraph=True`.
     patch_torch_compile_force_graph()
+
+
+
+if os.environ.get("PATCH_TESTING_METHODS_TO_COLLECT_OUTPUTS", "").lower() in ("yes", "true", "on", "y", "1"):
+    patch_testing_methods_to_collect_info()