correct fixtures

Author: Linchin

samples/pyarrow/append_rows_with_arrow.py

@@ -30,7 +30,7 @@ def bqstorage_write_client():
     return bigquery_storage_v1.BigQueryWriteClient()
 
 
-def make_table(project_id, dataset, bq_client):
+def make_table(project_id, dataset_id, bq_client):
     schema = [
         bigquery.SchemaField("bool_col", enums.SqlTypeNames.BOOLEAN),
         bigquery.SchemaField("int64_col", enums.SqlTypeNames.INT64),
@@ -59,7 +59,7 @@ def make_table(project_id, dataset, bq_client):
         ),
     ]
     table_id = "append_rows_w_arrow_test"
-    table_id_full = f"{project_id}.{dataset}.{table_id}"
+    table_id_full = f"{project_id}.{dataset_id}.{table_id}"
     bq_table = bigquery.Table(table_id_full, schema=schema)
     created_table = bq_client.create_table(bq_table)
 
@@ -167,8 +167,8 @@ def append_rows(bqstorage_write_client, table):
         print(e)
 
 
-def main(project_id, dataset_id):
+def main(project_id, dataset):
     write_client = bqstorage_write_client()
     bq_client = bigquery.Client()
-    table = make_table(project_id, dataset_id, bq_client)
+    table = make_table(project_id, dataset.dataset_id, bq_client)
     append_rows(write_client, table)

samples/pyarrow/append_rows_with_arrow_test.py

@@ -15,7 +15,7 @@
 from . import append_rows_with_arrow
 
 
-def test_read_query_results(capsys, project_id, dataset_id):
-    append_rows_with_arrow.main(project_id, dataset_id)
+def test_append_rows_with_arrow(capsys, project_id, dataset):
+    append_rows_with_arrow.main(project_id, dataset)
     out, _ = capsys.readouterr()
     assert "append_result" in out