Merge branch 'main' into b479135143

Author: msampathkumar

bigquery-datatransfer/snippets/manage_transfer_configs.py

@@ -13,27 +13,6 @@
 # limitations under the License.
 
 
-def list_configs(override_values={}):
-    # [START bigquerydatatransfer_list_configs]
-    from google.cloud import bigquery_datatransfer
-
-    transfer_client = bigquery_datatransfer.DataTransferServiceClient()
-
-    project_id = "my-project"
-    # [END bigquerydatatransfer_list_configs]
-    # To facilitate testing, we replace values with alternatives
-    # provided by the testing harness.
-    project_id = override_values.get("project_id", project_id)
-    # [START bigquerydatatransfer_list_configs]
-    parent = transfer_client.common_project_path(project_id)
-
-    configs = transfer_client.list_transfer_configs(parent=parent)
-    print("Got the following configs:")
-    for config in configs:
-        print(f"\tID: {config.name}, Schedule: {config.schedule}")
-    # [END bigquerydatatransfer_list_configs]
-
-
 def update_config(override_values={}):
     # [START bigquerydatatransfer_update_config]
     from google.cloud import bigquery_datatransfer

bigquery-datatransfer/snippets/manage_transfer_configs_test.py

@@ -15,13 +15,6 @@
 from . import manage_transfer_configs
 
 
-def test_list_configs(capsys, project_id, transfer_config_name):
-    manage_transfer_configs.list_configs({"project_id": project_id})
-    out, _ = capsys.readouterr()
-    assert "Got the following configs:" in out
-    assert transfer_config_name in out
-
-
 def test_update_config(capsys, transfer_config_name):
     manage_transfer_configs.update_config(
         {

…mples/v1/language_sentiment_text_test.py …uery/python-db-dtypes-pandas/__init__.pylanguage/snippets/generated-samples/v1/language_sentiment_text_test.py renamed to bigquery/python-db-dtypes-pandas/__init__.py language/snippets/generated-samples/v1/language_sentiment_text_test.py renamed to bigquery/python-db-dtypes-pandas/__init__.py

@@ -1,27 +1,13 @@
-# Copyright 2018 Google, Inc.
+# 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
+#     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 language_sentiment_text
-
-
-def test_analyze_sentiment_text_positive(capsys):
-    language_sentiment_text.sample_analyze_sentiment("Happy Happy Joy Joy")
-    out, _ = capsys.readouterr()
-    assert "Score: 0." in out
-
-
-def test_analyze_sentiment_text_negative(capsys):
-    language_sentiment_text.sample_analyze_sentiment("Angry Angry Sad Sad")
-    out, _ = capsys.readouterr()
-    assert "Score: -0." in out

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

@@ -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,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/noxconfig.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.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/pandas_date_and_time_test.py

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

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

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