Skip to content

Commit de804b0

Browse files
committed
fix test_read_gbq_w_json_in_array
1 parent 294f4c3 commit de804b0

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

tests/system/small/test_dataframe_io.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -418,15 +418,15 @@ def test_read_gbq_w_json_in_array(session):
418418
# TODO(b/401630655): JSON is not compatible with allow_large_results=False
419419
pd_data = data.to_pandas(allow_large_results=True)
420420

421-
assert pd_data.list[0][0] == '{"boolean":true}'
422-
assert pd_data.list[1][0] == '{"int":100}'
423-
assert pd_data.list[2][0] == '{"float":0.98}'
424-
assert pd_data.list[3][0] == '{"string":"hello world"}'
425-
assert pd_data.list[4][0] == '{"array":[8,9,10]}'
426-
assert pd_data.list[5][0] == '{"null":null}'
427-
assert (
428-
pd_data.list[6][0] == '{"dict":{"array":[{"bar":"hello"},{"foo":1}],"int":1}}'
429-
)
421+
assert pd_data[0] == [
422+
'{"boolean":true}',
423+
'{"int":100}',
424+
'{"float":0.98}',
425+
'{"string":"hello world"}',
426+
'{"array":[8,9,10]}',
427+
'{"null":null}',
428+
'{"dict":{"array":[{"bar":"hello"},{"foo":1}],"int":1}}',
429+
]
430430

431431

432432
def test_to_pandas_batches_w_correct_dtypes(scalars_df_default_index):

0 commit comments

Comments
 (0)