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

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

@@ -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/__init__.py

@@ -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

@@ -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

@@ -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

@@ -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

@@ -0,0 +1 @@
+pytest==8.4.2

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

@@ -0,0 +1,4 @@
+db-dtypes
+numpy
+pandas
+pyarrow