undo test file deletion

Author: sycai

samples/snippets/load_data_from_bigquery_test.py

@@ -0,0 +1,24 @@
+# Copyright 2023 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 test_bigquery_dataframes_load_data_from_bigquery() -> None:
+    # [START bigquery_dataframes_load_data_from_bigquery]
+    # Create a DataFrame from a BigQuery table:
+    import bigframes.pandas as bpd
+
+    query_or_table = "bigquery-public-data.ml_datasets.penguins"
+    bq_df = bpd.read_gbq(query_or_table)
+    # [END bigquery_dataframes_load_data_from_bigquery]
+    assert bq_df is not None

samples/snippets/load_data_from_csv_test.py

@@ -0,0 +1,25 @@
+# Copyright 2023 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 test_bigquery_dataframes_load_data_from_csv() -> None:
+    # [START bigquery_dataframes_load_data_from_csv]
+    import bigframes.pandas as bpd
+
+    filepath_or_buffer = "gs://cloud-samples-data/bigquery/us-states/us-states.csv"
+    df_from_gcs = bpd.read_csv(filepath_or_buffer)
+    # Display the first few rows of the DataFrame:
+    df_from_gcs.head()
+    # [END bigquery_dataframes_load_data_from_csv]
+    assert df_from_gcs is not None

samples/snippets/sessions_and_io_test.py

@@ -130,19 +130,19 @@ def test_sessions_and_io(project_id: str, dataset_id: str) -> None:
     # [END bigquery_dataframes_to_pandas_dry_run]
     assert df.to_pandas(dry_run=True) is not None
 
-    # [START bigquery_dataframes_load_data_from_csv]
+    # [START bigquery_dataframes_read_data_from_csv]
     import bigframes.pandas as bpd
 
     # Read a CSV file from GCS
     df = bpd.read_csv("gs://cloud-samples-data/bigquery/us-states/us-states.csv")
-    # [END bigquery_dataframes_load_data_from_csv]
+    # [END bigquery_dataframes_read_data_from_csv]
     assert df is not None
 
-    # [START bigquery_dataframes_load_data_from_bigquery]
+    # [START bigquery_dataframes_read_data_from_bigquery_table]
     import bigframes.pandas as bpd
 
     df = bpd.read_gbq("bigquery-public-data.ml_datasets.penguins")
-    # [END bigquery_dataframes_load_data_from_bigquery]
+    # [END bigquery_dataframes_read_data_from_bigquery_table]
     assert df is not None
 
     # [START bigquery_dataframes_read_from_sql_query]