feat(db-dtypes): support pandas 3.0 (#17177)

tswast · web-flow · commit 2086b34d8b34 · 2026-05-19T12:16:54.000-05:00
Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly: - [ ] Make sure to open an issue as a [bug/issue](https://github.com/googleapis/google-cloud-python/issues) before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [ ] Ensure the tests and linter pass - [ ] Code coverage does not decrease (if any source code was changed) - [ ] Appropriate docs were updated (if necessary) Fixes #15509 🦕
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -16,16 +16,17 @@
 # See https://pre-commit.com/hooks.html for more hooks
 repos:
 -   repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.0.1
+    rev: v6.0.0
     hooks:
     -   id: trailing-whitespace
     -   id: end-of-file-fixer
     -   id: check-yaml
 -   repo: https://github.com/psf/black
-    rev: 22.3.0
+    rev: 23.7.0
     hooks:
     - id: black
 -   repo: https://github.com/pycqa/flake8
-    rev: 3.9.2 # version-scanner: ignore
+    rev: 6.1.0 # version-scanner: ignore
     hooks:
     - id: flake8
+      args: [--config, packages/google-cloud-alloydb/.flake8]
diff --git a/packages/db-dtypes/setup.py b/packages/db-dtypes/setup.py
@@ -33,7 +33,7 @@
     "numpy >= 1.24.0, <= 2.2.6 ; python_version == '3.10'",
     "numpy >= 1.24.0 ; python_version != '3.10'",
     "packaging >= 24.2.0",
-    "pandas >= 1.5.3, < 3.0.0",
+    "pandas >= 1.5.3, < 4.0.0",
     "pyarrow >= 13.0.0",
 ]
 
diff --git a/packages/db-dtypes/tests/compliance/json/conftest.py b/packages/db-dtypes/tests/compliance/json/conftest.py
@@ -83,6 +83,13 @@ def data_missing_for_sorting():
     return JSONArray._from_sequence([json.dumps({"b": 1}), None, json.dumps({"a": 4})])
 
 
+@pytest.fixture
+def data_for_sorting():
+    return JSONArray._from_sequence(
+        [json.dumps({"b": 1}), json.dumps({"c": 4}), json.dumps({"a": 2})]
+    )
+
+
 @pytest.fixture
 def na_cmp():
     """
diff --git a/packages/db-dtypes/tests/compliance/json/test_json_compliance.py b/packages/db-dtypes/tests/compliance/json/test_json_compliance.py
@@ -232,6 +232,18 @@ def test_sort_values(self, data_for_sorting):
     def test_sort_values_frame(self, data_for_sorting):
         super().test_sort_values_frame(data_for_sorting)
 
+    @pytest.mark.skip(reason="BigQuery does not allow sort by a JSON-type column.")
+    def test_rank(self, data_for_sorting, ascending):
+        super().test_rank(data_for_sorting, ascending)
+
+    @pytest.mark.skip(reason="BigQuery does not allow sort by a JSON-type column.")
+    def test_rank_method(self, data_for_sorting, method):
+        super().test_rank_method(data_for_sorting, method)
+
+    @pytest.mark.skip(reason="BigQuery does not allow sort by a JSON-type column.")
+    def test_rank_missing(self, data_missing_for_sorting, na_option):
+        super().test_rank_missing(data_missing_for_sorting, na_option)
+
     def test_argmax_argmin_no_skipna_notimplemented(self, data_missing_for_sorting):
         # This test was failing compliance checks because it attempted to match
         # a pytest regex match using an empty string (""), which pytest version

Original file line number	Diff line number	Diff line change
`@@ -33,7 +33,7 @@`
`33`	`33`	`"numpy >= 1.24.0, <= 2.2.6 ; python_version == '3.10'",`
`34`	`34`	`"numpy >= 1.24.0 ; python_version != '3.10'",`
`35`	`35`	`"packaging >= 24.2.0",`
`36`		`- "pandas >= 1.5.3, < 3.0.0",`
	`36`	`+ "pandas >= 1.5.3, < 4.0.0",`
`37`	`37`	`"pyarrow >= 13.0.0",`
`38`	`38`	`]`
`39`	`39`