use older pyarrow.record_batch constructor

tswast · tswast · commit eb22364b9c89 · 2025-02-04T11:33:19.000-06:00
diff --git a/tests/unit/test__pandas_helpers.py b/tests/unit/test__pandas_helpers.py
@@ -1874,10 +1874,10 @@ def test__download_table_bqstorage_shuts_down_workers(
         ]
     )
     arrow_rows = pyarrow.record_batch(
-        {
-            "int_col": [0, 1, 2],
-            "str_col": ["a", "b", "c"],
-        },
+        [
+            pyarrow.array([0, 1, 2], type=pyarrow.int64()),
+            pyarrow.array(["a", "b", "c"], type=pyarrow.string()),
+        ],
         schema=arrow_schema,
     )
     session = google.cloud.bigquery_storage_v1.types.ReadSession()

Original file line number	Diff line number	Diff line change
`@@ -1874,10 +1874,10 @@ def test__download_table_bqstorage_shuts_down_workers(`
`1874`	`1874`	`]`
`1875`	`1875`	`)`
`1876`	`1876`	`arrow_rows = pyarrow.record_batch(`
`1877`		`- {`
`1878`		`- "int_col": [0, 1, 2],`
`1879`		`- "str_col": ["a", "b", "c"],`
`1880`		`- },`
	`1877`	`+ [`
	`1878`	`+ pyarrow.array([0, 1, 2], type=pyarrow.int64()),`
	`1879`	`+ pyarrow.array(["a", "b", "c"], type=pyarrow.string()),`
	`1880`	`+ ],`
`1881`	`1881`	`schema=arrow_schema,`
`1882`	`1882`	`)`
`1883`	`1883`	`session = google.cloud.bigquery_storage_v1.types.ReadSession()`