From b43daf621a7b06a2684567146ca3b84a270520c6 Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Fri, 20 Aug 2021 14:30:44 -0500 Subject: [PATCH] Minor test tweaks --- dask_bigquery/tests/test_core.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/dask_bigquery/tests/test_core.py b/dask_bigquery/tests/test_core.py index cfce96a..c2ca052 100644 --- a/dask_bigquery/tests/test_core.py +++ b/dask_bigquery/tests/test_core.py @@ -31,17 +31,18 @@ def df(): @pytest.fixture def dataset(df): "Push some data to BigQuery using pandas gbq" - dataset_id = uuid.uuid4().hex project_id = "dask-bigquery" + dataset_id = uuid.uuid4().hex + table_id = "table_test" # push data to gbq pd.DataFrame.to_gbq( df, - destination_table=f"{dataset_id}.table_test", + destination_table=f"{dataset_id}.{table_id}", project_id=project_id, chunksize=5, if_exists="append", ) - yield f"{project_id}.{dataset_id}.table_test" + yield (project_id, dataset_id, table_id) with bigquery.Client() as bq_client: bq_client.delete_dataset( @@ -53,14 +54,11 @@ def dataset(df): # test simple read def test_read_gbq(df, dataset, client): """Test simple read of data pushed to BigQuery using pandas-gbq""" - project_id, dataset_id, table_id = dataset.split(".") - + project_id, dataset_id, table_id = dataset ddf = read_gbq(project_id=project_id, dataset_id=dataset_id, table_id=table_id) - assert ddf.columns.tolist() == ["name", "number", "idx"] - assert len(ddf) == 10 + assert list(ddf.columns) == ["name", "number", "idx"] assert ddf.npartitions == 2 - assert assert_eq(ddf.set_index("idx"), df.set_index("idx"))