Use ARROW as data format in DB API cursor

googleapis · May 13, 2020 · 5b82c76 · 5b82c76
1 parent 530a5db
commit 5b82c76
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/google/cloud/bigquery/dbapi/cursor.py b/google/cloud/bigquery/dbapi/cursor.py
@@ -296,7 +296,7 @@ def _bqstorage_fetch(self, bqstorage_client):
         else:
             requested_session = bigquery_storage_v1.types.ReadSession(
                 table=table_reference.to_bqstorage(),
-                data_format=bigquery_storage_v1.enums.DataFormat.AVRO,
+                data_format=bigquery_storage_v1.enums.DataFormat.ARROW,
             )
             read_session = bqstorage_client.create_read_session(
                 parent="projects/{}".format(table_reference.project),

diff --git a/tests/system.py b/tests/system.py
@@ -1667,7 +1667,7 @@ def test_dbapi_fetch_w_bqstorage_client_small_result_set(self):
     @unittest.skipIf(
         bigquery_storage_v1 is None, "Requires `google-cloud-bigquery-storage`"
     )
-    @unittest.skipIf(fastavro is None, "Requires `fastavro`")
+    @unittest.skipIf(pyarrow is None, "Requires `pyarrow`")
     def test_dbapi_fetch_w_bqstorage_client_large_result_set(self):
         bqstorage_client = bigquery_storage_v1.BigQueryReadClient(
             credentials=Config.CLIENT._credentials
@@ -1717,7 +1717,7 @@ def test_dbapi_fetch_w_bqstorage_client_large_result_set(self):
     @unittest.skipIf(
         bigquery_storage_v1beta1 is None, "Requires `google-cloud-bigquery-storage`"
     )
-    @unittest.skipIf(fastavro is None, "Requires `fastavro`")
+    @unittest.skipIf(pyarrow is None, "Requires `pyarrow`")
     def test_dbapi_fetch_w_bqstorage_client_v1beta1_large_result_set(self):
         bqstorage_client = bigquery_storage_v1beta1.BigQueryStorageClient(
             credentials=Config.CLIENT._credentials