From 679e45ee4d9b30417f5403d504eac588623930c9 Mon Sep 17 00:00:00 2001
From: Thomas Li <47963215+lithomas1@users.noreply.github.com>
Date: Sat, 14 Sep 2024 08:20:30 -0400
Subject: [PATCH] fix typing and tune tests for copy on write

---
 pandas/io/gbq.py                | 6 +++---
 pandas/tests/io/test_parquet.py | 5 ++++-
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/pandas/io/gbq.py b/pandas/io/gbq.py
index 350002bf461ff..24e4e0b7cef0a 100644
--- a/pandas/io/gbq.py
+++ b/pandas/io/gbq.py
@@ -11,7 +11,7 @@
 from pandas.util._exceptions import find_stack_level
 
 if TYPE_CHECKING:
-    import google.auth
+    from google.auth.credentials import Credentials
 
     from pandas import DataFrame
 
@@ -37,7 +37,7 @@ def read_gbq(
     dialect: str | None = None,
     location: str | None = None,
     configuration: dict[str, Any] | None = None,
-    credentials: google.auth.credentials.Credentials | None = None,
+    credentials: Credentials | None = None,
     use_bqstorage_api: bool | None = None,
     max_results: int | None = None,
     progress_bar_type: str | None = None,
@@ -230,7 +230,7 @@ def to_gbq(
     table_schema: list[dict[str, str]] | None = None,
     location: str | None = None,
     progress_bar: bool = True,
-    credentials: google.auth.credentials.Credentials | None = None,
+    credentials: Credentials | None = None,
 ) -> None:
     warnings.warn(
         "to_gbq is deprecated and will be removed in a future version. "
diff --git a/pandas/tests/io/test_parquet.py b/pandas/tests/io/test_parquet.py
index d030ec12f882d..fd3b74dfa1a17 100644
--- a/pandas/tests/io/test_parquet.py
+++ b/pandas/tests/io/test_parquet.py
@@ -1315,7 +1315,10 @@ def test_empty_dataframe(self, fp):
         check_round_trip(df, fp, expected=expected)
 
     @pytest.mark.xfail(
-        reason="fastparquet bug, see https://github.com/dask/fastparquet/issues/929"
+        not using_copy_on_write()
+        and _HAVE_FASTPARQUET
+        and Version(fastparquet.__version__) > Version("2022.12"),
+        reason="fastparquet bug, see https://github.com/dask/fastparquet/issues/929",
     )
     def test_timezone_aware_index(self, fp, timezone_aware_date_list):
         idx = 5 * [timezone_aware_date_list]