chore: migrate code from googleapis/python-db-dtypes-pandas #13776

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

iennae merged 49 commits into main from python-db-dtypes-pandas-migration

Feb 3, 2026

+212 −0

bigquery/python-db-dtypes-pandas/__init__.py

-Original file line number
+Diff line change
@@ -0,0 +1,13 @@
+    # Copyright 2021 Google LLC
+    #
+    # Licensed under the Apache License, Version 2.0 (the "License");
+    # you may not use this file except in compliance with the License.
+    # You may obtain a copy of the License at
+    #
+    #     http://www.apache.org/licenses/LICENSE-2.0
+    #
+    # Unless required by applicable law or agreed to in writing, software
+    # distributed under the License is distributed on an "AS IS" BASIS,
+    # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    # See the License for the specific language governing permissions and
+    # limitations under the License.

bigquery/python-db-dtypes-pandas/pytest.ini

Empty file.

bigquery/python-db-dtypes-pandas/snippets/__init__.py

-Original file line number
+Diff line change
@@ -0,0 +1,13 @@
+    # Copyright 2021 Google LLC
+    #
+    # Licensed under the Apache License, Version 2.0 (the "License");
+    # you may not use this file except in compliance with the License.
+    # You may obtain a copy of the License at
+    #
+    #     http://www.apache.org/licenses/LICENSE-2.0
+    #
+    # Unless required by applicable law or agreed to in writing, software
+    # distributed under the License is distributed on an "AS IS" BASIS,
+    # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    # See the License for the specific language governing permissions and
+    # limitations under the License.

bigquery/python-db-dtypes-pandas/snippets/noxconfig.py

-Original file line number
+Diff line change
@@ -0,0 +1,42 @@
+    # Copyright 2021 Google LLC
+    #
+    # Licensed under the Apache License, Version 2.0 (the "License");
+    # you may not use this file except in compliance with the License.
+    # You may obtain a copy of the License at
+    #
+    #      http://www.apache.org/licenses/LICENSE-2.0
+    #
+    # Unless required by applicable law or agreed to in writing, software
+    # distributed under the License is distributed on an "AS IS" BASIS,
+    # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    # See the License for the specific language governing permissions and
+    # limitations under the License.
+    # Default TEST_CONFIG_OVERRIDE for python repos.
+    # You can copy this file into your directory, then it will be imported from
+    # the noxfile.py.
+    # The source of truth:
+    # https://github.com/GoogleCloudPlatform/python-docs-samples/blob/main/noxfile_config.py
+    TEST_CONFIG_OVERRIDE = {
+        # You can opt out from the test for specific Python versions.
+        "ignored_versions": ["2.7", "3.7", "3.8"],
+        # Old samples are opted out of enforcing Python type hints
+        # All new samples should feature them
+        "enforce_type_hints": True,
+        # An envvar key for determining the project id to use. Change it
+        # to 'BUILD_SPECIFIC_GCLOUD_PROJECT' if you want to opt in using a
+        # build specific Cloud project. You can also use your own string
+        # to use your own Cloud project.
+        "gcloud_project_env": "GOOGLE_CLOUD_PROJECT",
+        # 'gcloud_project_env': 'BUILD_SPECIFIC_GCLOUD_PROJECT',
+        # If you need to use a specific version of pip,
+        # change pip_version_override to the string representation
+        # of the version number, for example, "20.2.4"
+        "pip_version_override": None,
+        # A dictionary you want to inject into your test. Don't put any
+        # secrets here. These values will override predefined values.
+        "envs": {},
+    }

bigquery/python-db-dtypes-pandas/snippets/pandas_date_and_time.py

-Original file line number
+Diff line change
@@ -0,0 +1,79 @@
+    # Copyright 2021 Google LLC
+    #
+    # Licensed under the Apache License, Version 2.0 (the "License");
+    # you may not use this file except in compliance with the License.
+    # You may obtain a copy of the License at
+    #
+    #     http://www.apache.org/licenses/LICENSE-2.0
+    #
+    # Unless required by applicable law or agreed to in writing, software
+    # distributed under the License is distributed on an "AS IS" BASIS,
+    # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    # See the License for the specific language governing permissions and
+    # limitations under the License.
+    def pandas_date_and_time():
+        # [START bigquery_pandas_date_create]
+        import datetime
+        import pandas as pd
+        import db_dtypes  # noqa import to register dtypes
+        dates = pd.Series([datetime.date(2021, 9, 17), "2021-9-18"], dtype="dbdate")
+        # [END bigquery_pandas_date_create]
+        # [START bigquery_pandas_date_as_datetime]
+        datetimes = dates.astype("datetime64")
+        # [END bigquery_pandas_date_as_datetime]
+        # [START bigquery_pandas_date_sub]
+        dates2 = pd.Series(["2021-1-1", "2021-1-2"], dtype="dbdate")
+        diffs = dates - dates2
+        # [END bigquery_pandas_date_sub]
+        # [START bigquery_pandas_date_add_offset]
+        do = pd.DateOffset(days=1)
+        after = dates + do
+        before = dates - do
+        # [END bigquery_pandas_date_add_offset]
+        # [START bigquery_pandas_time_create]
+        times = pd.Series([datetime.time(1, 2, 3, 456789), "12:00:00.6"], dtype="dbtime")
+        # [END bigquery_pandas_time_create]
+        # [START bigquery_pandas_time_as_timedelta]
+        timedeltas = times.astype("timedelta64")
+        # [END bigquery_pandas_time_as_timedelta]
+        # Combine datetime64 and timedelta64 to confirm adding dates and times are
+        # equivalent.
+        combined0 = datetimes + timedeltas
+        # [START bigquery_pandas_combine_date_time]
+        combined = dates + times
+        # [END bigquery_pandas_combine_date_time]
+        return (
+            dates,
+            datetimes,
+            dates2,
+            diffs,
+            do,
+            after,
+            before,
+            times,
+            timedeltas,
+            combined,
+            combined0,
+        )

bigquery/python-db-dtypes-pandas/snippets/pandas_date_and_time_test.py

-Original file line number
+Diff line change
@@ -0,0 +1,60 @@
+    # Copyright 2021 Google LLC
+    #
+    # Licensed under the Apache License, Version 2.0 (the "License");
+    # you may not use this file except in compliance with the License.
+    # You may obtain a copy of the License at
+    #
+    #     http://www.apache.org/licenses/LICENSE-2.0
+    #
+    # Unless required by applicable law or agreed to in writing, software
+    # distributed under the License is distributed on an "AS IS" BASIS,
+    # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+    # See the License for the specific language governing permissions and
+    # limitations under the License.
+    import datetime
+    import numpy as np
+    from pandas import Timestamp
+    def test_pandas_date_and_time():
+        from .pandas_date_and_time import pandas_date_and_time
+        (
+            dates,
+            _,
+            dates2,
+            diffs,
+            do,
+            after,
+            before,
+            times,
+            _,
+            combined,
+            combined0,
+        ) = pandas_date_and_time()
+        assert str(dates.dtype) == "dbdate"
+        assert list(dates) == [datetime.date(2021, 9, 17), datetime.date(2021, 9, 18)]
+        assert np.array_equal(
+            diffs,
+            dates.astype("datetime64") - dates2.astype("datetime64"),
+        )
+        assert np.array_equal(after, dates.astype("object") + do)
+        assert np.array_equal(before, dates.astype("object") - do)
+        assert str(times.dtype) == "dbtime"
+        assert list(times) == [
+            datetime.time(1, 2, 3, 456789),
+            datetime.time(12, 0, 0, 600000),
+        ]
+        for c in combined0, combined:
+            assert str(c.dtype) == "datetime64[ns]"
+            assert list(c) == [
+                Timestamp("2021-09-17 01:02:03.456789"),
+                Timestamp("2021-09-18 12:00:00.600000"),
+            ]

bigquery/python-db-dtypes-pandas/snippets/requirements-test.txt

Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		pytest==8.4.2
Copy link Contributor gemini-code-assist bot Feb 3, 2026 Choose a reason for hiding this comment The reason will be displayed to describe this comment to others. Learn more. Pinning `pytest` to an exact patch version (`8.4.2`) can be overly restrictive. It's generally recommended to use a more flexible version specifier (e.g., `pytest>=8.4.2,<9` or `pytest~=8.4.2`) to allow for minor bug fixes and security updates without requiring manual intervention, unless there's a specific compatibility issue that necessitates this exact pin. `pytest~=8.4.2`

bigquery/python-db-dtypes-pandas/snippets/requirements.txt

-Original file line number
+Diff line change
@@ -0,0 +1,4 @@
+    db-dtypes
+    numpy
+    pandas
+    pyarrow

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

chore: migrate code from googleapis/python-db-dtypes-pandas #13776

Diff view

Diff view

There are no files selected for viewing

gemini-code-assist bot Feb 3, 2026

Uh oh!

chore: migrate code from googleapis/python-db-dtypes-pandas #13776

chore: migrate code from googleapis/python-db-dtypes-pandas #13776

Uh oh!

Uh oh!

Diff view

Diff view

There are no files selected for viewing

gemini-code-assist bot Feb 3, 2026

Choose a reason for hiding this comment

Uh oh!