Skip to content

Commit eec722a

Browse files
committed
test: update testcase to fix the call stack management
1 parent e682e2d commit eec722a

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/unit/session/test_read_gbq_colab.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,17 @@
2727

2828
def test_read_gbq_colab_includes_label():
2929
"""Make sure we can tell direct colab usage apart from regular read_gbq usage."""
30+
import bigframes.core.log_adapter as log_adapter
31+
32+
# Clear any existing call stack and API methods
33+
log_adapter._call_stack.clear()
34+
log_adapter.get_and_reset_api_methods()
35+
3036
session = mocks.create_bigquery_session()
37+
38+
# Ensure call stack is empty before calling the method
39+
log_adapter._call_stack.clear()
40+
3141
_ = session._read_gbq_colab("SELECT 'read-gbq-colab-test'")
3242
configs = session._job_configs # type: ignore
3343

0 commit comments

Comments
 (0)