We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e682e2d commit eec722aCopy full SHA for eec722a
tests/unit/session/test_read_gbq_colab.py
@@ -27,7 +27,17 @@
27
28
def test_read_gbq_colab_includes_label():
29
"""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
36
session = mocks.create_bigquery_session()
37
38
+ # Ensure call stack is empty before calling the method
39
40
41
_ = session._read_gbq_colab("SELECT 'read-gbq-colab-test'")
42
configs = session._job_configs # type: ignore
43
0 commit comments