recursion test prints more info

Author: TrevorBergeron

bigframes/pandas/__init__.py

@@ -20,6 +20,7 @@
 import datetime
 import inspect
 import sys
+import traceback
 import typing
 from typing import Literal, Optional, Sequence, Union
 
@@ -369,6 +370,19 @@ def reset_session():
 # https://github.com/python/cpython/issues/112282
 sys.setrecursionlimit(max(10000000, sys.getrecursionlimit()))
 
+
+original_setrecursionlimit = sys.setrecursionlimit
+print(f"recursion limit set to {sys.getrecursionlimit()}")
+
+
+def patched_setrecursionlimit(n):
+    print(f"\n[DEBUG] Recursion limit being set to {n} by:")
+    traceback.print_stack()
+    original_setrecursionlimit(n)
+
+
+sys.setrecursionlimit = patched_setrecursionlimit
+
 if resource is not None:
     soft_limit, hard_limit = resource.getrlimit(resource.RLIMIT_STACK)
     if soft_limit < hard_limit or hard_limit == resource.RLIM_INFINITY:

tests/system/small/test_dataframe.py

@@ -5755,13 +5755,15 @@ def test_df_dot_operator_series(
 
 
 def test_recursion_limit(scalars_df_index):
+    import sys
+
+    print(f"doing recursion test, recursion limit set to {sys.getrecursionlimit()}")
     scalars_df_index = scalars_df_index[["int64_too", "int64_col", "float64_col"]]
     for i in range(400):
         scalars_df_index = scalars_df_index + 4
     try:
         scalars_df_index.to_pandas()
     except Exception:
-        import sys
 
         try:
             import resource