tests: add testing for Python 3.15

Author: parthea

.github/workflows/unittest.yml

@@ -83,7 +83,7 @@ jobs:
     strategy:
       fail-fast: true
       matrix:
-        python: ['3.10', "3.11", "3.12", "3.13", "3.14"]
+        python: ["3.10", "3.11", "3.12", "3.13", "3.14", "3.15"]
         package_shard: ${{ fromJson(needs.initialize.outputs.matrix) }}
     name: ${{ matrix.package_shard.is_sharded && format('unit ({0}, {1})', matrix.python, matrix.package_shard.name) || format('unit ({0})', matrix.python) }}
     steps:
@@ -100,10 +100,38 @@ jobs:
       with:
         python-version: ${{ matrix.python }}
         cache: 'pip'
+        allow-prereleases: true
+    - name: Setup uv
+      uses: astral-sh/setup-uv@d4b2f3b6ecc6e67c4457f6d3e41ec42d3d0fcb86 # v5
+      with:
+        enable-cache: true
+        cache-dependency-glob: 'packages/**/testing/constraints*.txt'
     - name: Install nox
       run: |
         python -m pip install --upgrade setuptools pip wheel
         python -m pip install nox uv
+    - name: Run unit tests
+        uv pip install --system nox nox-uv
+    # Caches compiled wheels locally to prevent building heavy libraries
+    # such as grpcio, which we build from scratch on every run for Python 3.15+.
+    # Follow https://github.com/grpc/grpc/issues/41010 for updates.
+    - name: Cache Built Python 3.15 Wheels
+      if: matrix.python == '3.15'
+      uses: actions/cache@0057852bfaa89a56745cba8c7296529d2fc39830 # v4
+      with:
+        path: .uv-wheel-cache
+        key: ${{ runner.os }}-uv-wheels-3.15-${{ hashFiles('packages/**/testing/constraints*.txt') }}
+        restore-keys: |
+          ${{ runner.os }}-uv-wheels-3.15-
+    # Pre-cache heavy dependencies sequentially before running parallel tests
+    - name: Pre-cache heavy dependencies (grpcio)
+      if: matrix.python == '3.15'
+      run: |
+        # Create the directory so uv doesn't crash if there is a `cache miss`
+        mkdir -p .uv-wheel-cache
+        # Download and compile grpcio sequentially so that parallel test workers
+        # immediately hit the warm cache instead of deadlocking on compilation.
+        uv pip install --system grpcio --find-links .uv-wheel-cache
     - name: Run unit tests for ${{ matrix.package_shard.description }}
       env:
         BUILD_TYPE: presubmit

ci/run_single_test.sh

@@ -90,6 +90,11 @@ case ${TEST_TYPE} in
             nox -s unit-3.14
             retval=$?
             ;;
+        "3.15")
+            # This is needed to speed up builds
+            nox --force-venv-backend uv -s unit-3.15
+            retval=$?
+            ;;
         *)
             echo "unsupported PY_VERSION"
             exit 1

packages/bigframes/mypy.ini

@@ -60,7 +60,7 @@
 
 DEFAULT_PYTHON_VERSION = "3.14"
 
-ALL_PYTHON = ["3.10", "3.11", "3.12", "3.13", "3.14"]
+ALL_PYTHON = ["3.10", "3.11", "3.12", "3.13", "3.14", "3.15"]
 UNIT_TEST_STANDARD_DEPENDENCIES = [
     "mock",
     PYTEST_VERSION,
@@ -283,6 +283,10 @@ def run_unit(session, install_test_extra):
 @nox.session(python=ALL_PYTHON)
 @nox.parametrize("test_extra", [True, False])
 def unit(session, test_extra):
+    if session.python == "3.15":
+        session.skip(
+            "Skipping 3.15 until wheels are available for pyarrow. Also pyproj wheels are needed for dependency geopandas."
+        )
     if test_extra:
         run_unit(session, install_test_extra=test_extra)
     else:

packages/bigframes/noxfile.py

@@ -41,6 +41,7 @@
     "3.12",
     "3.13",
     "3.14",
+    "3.15",
 ]
 
 UNIT_TEST_STANDARD_DEPENDENCIES = [
@@ -237,6 +238,10 @@ def install_unittest_dependencies(session, *constraints):
 @nox.session(python=UNIT_TEST_PYTHON_VERSIONS)
 def unit(session):
     # Install all test dependencies, then install this package in-place.
+    if session.python == "3.15":
+        session.skip(
+            "Skipping 3.15 until wheels are available for pyproj needed for dependency geopandas"
+        )
 
     constraints_path = str(
         CURRENT_DIRECTORY / "testing" / f"constraints-{session.python}.txt"

packages/bigquery-magics/noxfile.py

@@ -42,6 +42,7 @@
     "3.12",
     "3.13",
     "3.14",
+    "3.15",
 ]
 
 UNIT_TEST_STANDARD_DEPENDENCIES = [
@@ -302,6 +303,8 @@ def prerelease(session, tests_path):
 @nox.parametrize("test_type", ["unit", "compliance"])
 def unit(session, test_type):
     """Run the unit test suite."""
+    if session.python == "3.15":
+        session.skip("Skipping 3.15 until wheels are available for pyarrow.")
 
     # Compliance tests only run on the latest Python version
     if test_type == "compliance" and session.python != DEFAULT_PYTHON_VERSION:

packages/bigquery-magics/testing/constraints-3.15.txt

@@ -34,6 +34,7 @@
     "3.12",
     "3.13",
     "3.14",
+    "3.15",
 ]
 ALL_PYTHON = list(UNIT_TEST_PYTHON_VERSIONS)