Skip to content

Commit 1e5cf02

Browse files
authored
chore: add experimental blob read_gbq_object_table test (#1425)
1 parent 82d4382 commit 1e5cf02

File tree

1 file changed

+28
-0
lines changed

1 file changed

+28
-0
lines changed

tests/system/small/blob/test_io.py

Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -66,3 +66,31 @@ def test_blob_create_from_glob_path(bq_connection: str, session: bigframes.Sessi
6666
pd.testing.assert_frame_equal(
6767
pd_blob_df, expected_df, check_dtype=False, check_index_type=False
6868
)
69+
70+
71+
def test_blob_create_read_gbq_object_table(
72+
bq_connection: str, session: bigframes.Session
73+
):
74+
bigframes.options.experiments.blob = True
75+
76+
obj_table = session._create_object_table(
77+
"gs://bigframes_blob_test/images/*", bq_connection
78+
)
79+
80+
blob_df = session.read_gbq_object_table(obj_table, name="blob_col")
81+
pd_blob_df = blob_df["blob_col"].struct.explode().to_pandas()
82+
expected_df = pd.DataFrame(
83+
{
84+
"uri": [
85+
"gs://bigframes_blob_test/images/img0.jpg",
86+
"gs://bigframes_blob_test/images/img1.jpg",
87+
],
88+
"version": [None, None],
89+
"authorizer": [bq_connection.casefold(), bq_connection.casefold()],
90+
"details": [None, None],
91+
}
92+
)
93+
94+
pd.testing.assert_frame_equal(
95+
pd_blob_df, expected_df, check_dtype=False, check_index_type=False
96+
)

0 commit comments

Comments
 (0)