Skip to content

Commit 0b1f07f

Browse files
authored
chore: add a test for external table reading. (#812)
1 parent 1b613e0 commit 0b1f07f

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/system/small/test_session.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -570,6 +570,18 @@ def test_read_gbq_with_custom_global_labels(
570570
assert len(bigframes.options.compute.extra_query_labels) == 0
571571

572572

573+
def test_read_gbq_external_table(session: bigframes.Session):
574+
# Verify the table is external to ensure it hasn't been altered
575+
external_table_id = "bigframes-dev.bigframes_tests_sys.parquet_external_table"
576+
external_table = session.bqclient.get_table(external_table_id)
577+
assert external_table.table_type == "EXTERNAL"
578+
579+
df = session.read_gbq(external_table_id)
580+
581+
assert list(df.columns) == ["idx", "s1", "s2", "s3", "s4", "i1", "f1", "i2", "f2"]
582+
assert df["i1"].max() == 99
583+
584+
573585
def test_read_gbq_model(session, penguins_linear_model_name):
574586
model = session.read_gbq_model(penguins_linear_model_name)
575587
assert isinstance(model, bigframes.ml.linear_model.LinearRegression)

0 commit comments

Comments
 (0)