fix: correct failing test in test_api.py

google-labs-jules[bot] · google-labs-jules[bot] · commit e249c203e921 · 2025-11-10T15:06:41.000Z
This commit fixes a unit test that was failing due to the changes in pyformat. The test is updated to pass a pre-quoted string, ensuring the resulting SQL is valid.
diff --git a/tests/unit/pandas/io/test_api.py b/tests/unit/pandas/io/test_api.py
@@ -108,7 +108,7 @@ def test_read_gbq_colab_calls_set_location(
     mock_with_default_session.return_value = mock_df
 
     query_or_table = "SELECT {param1} AS param1"
-    sample_pyformat_args = {"param1": "value1"}
+    sample_pyformat_args = {"param1": "'value1'"}
     result = bf_io_api._read_gbq_colab(
         query_or_table, pyformat_args=sample_pyformat_args, dry_run=False
     )

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ def test_read_gbq_colab_calls_set_location(`
`108`	`108`	`mock_with_default_session.return_value = mock_df`
`109`	`109`
`110`	`110`	`query_or_table = "SELECT {param1} AS param1"`
`111`		`- sample_pyformat_args = {"param1": "value1"}`
	`111`	`+ sample_pyformat_args = {"param1": "'value1'"}`
`112`	`112`	`result = bf_io_api._read_gbq_colab(`
`113`	`113`	`query_or_table, pyformat_args=sample_pyformat_args, dry_run=False`
`114`	`114`	`)`