Skip to content
This repository was archived by the owner on Mar 6, 2026. It is now read-only.

test: Add test of datetime and time pandas load #895

Merged
jimfulton merged 6 commits into from
Aug 24, 2021
Merged

test: Add test of datetime and time pandas load #895

Changes from 1 commit
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
f9bd8b6
test: Add test of datetime and time pandas load
Aug 23, 2021
842c806
blacken
Aug 23, 2021
129778c
Uncommented the DATETIME column in test_load_table_from_dataframe_w_e…
Aug 24, 2021
8147187
Merge branch 'master' into test_date_time_load
plamut Aug 24, 2021
8b92e60
Add missing closing bracket
plamut Aug 24, 2021
d2f99bb
Add missing blank line
plamut Aug 24, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
blacken
  • Loading branch information
Jim Fulton
Jim Fulton committed Aug 23, 2021
commit 842c806f69547ee32b9e611e8978007e8bdba321
46 changes: 25 additions & 21 deletions tests/system/test_pandas.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -802,39 +802,43 @@ def test_list_rows_max_results_w_bqstorage(bigquery_client):

def test_upload_time_and_datetime(bigquery_client, dataset_id):
df = pandas.DataFrame(
dict(dt=[datetime.datetime(2020, 1, 8, 8, 0, 0),
datetime.datetime(
2020, 1, 8, 8, 0, 0,
tzinfo=datetime.timezone(datetime.timedelta(hours=-7))),
],
t=[datetime.time(0, 0, 10, 100001), None],
))
dict(
dt=[
datetime.datetime(2020, 1, 8, 8, 0, 0),
datetime.datetime(
2020,
1,
8,
8,
0,
0,
tzinfo=datetime.timezone(datetime.timedelta(hours=-7)),
),
],
t=[datetime.time(0, 0, 10, 100001), None],
)
)
table = f"{dataset_id}.test_upload_time_and_datetime"
bigquery_client.load_table_from_dataframe(df, table).result()
data = list(map(list, bigquery_client.list_rows(table)))
assert data == [
[datetime.datetime(2020, 1, 8, 8, 0, tzinfo=datetime.timezone.utc),
datetime.time(0, 0, 10, 100001),
],
[datetime.datetime(2020, 1, 8, 15, 0, tzinfo=datetime.timezone.utc),
None,
],
[
datetime.datetime(2020, 1, 8, 8, 0, tzinfo=datetime.timezone.utc),
datetime.time(0, 0, 10, 100001),
],
[datetime.datetime(2020, 1, 8, 15, 0, tzinfo=datetime.timezone.utc), None],
]

from google.cloud.bigquery import job, schema

table = f"{dataset_id}.test_upload_time_and_datetime_dt"
config = job.LoadJobConfig(
schema=[schema.SchemaField('dt', 'DATETIME'),
schema.SchemaField('t', 'TIME')])
schema=[schema.SchemaField("dt", "DATETIME"), schema.SchemaField("t", "TIME")]
)

bigquery_client.load_table_from_dataframe(df, table, job_config=config).result()
data = list(map(list, bigquery_client.list_rows(table)))
assert data == [
[datetime.datetime(2020, 1, 8, 8, 0),
datetime.time(0, 0, 10, 100001),
],
[datetime.datetime(2020, 1, 8, 15, 0),
None,
],
[datetime.datetime(2020, 1, 8, 8, 0), datetime.time(0, 0, 10, 100001)],
[datetime.datetime(2020, 1, 8, 15, 0), None],
]