test: fix tests

Author: shuoweil

tests/system/small/test_dataframe.py

@@ -969,6 +969,8 @@ def test_mimebundle_html_repr_w_all_rows(scalars_dfs, session):
     executions_pre = metrics.execution_count
     # When there are 10 or fewer rows, the outputs should be identical except for the extra note.
     bundle = scalars_df.head(10)._repr_mimebundle_()
+    if isinstance(bundle, tuple):
+        bundle = bundle[0]
     actual = bundle["text/html"]
     executions_post = metrics.execution_count
 

tests/unit/test_dataframe_polars.py

@@ -746,6 +746,8 @@ def test_mimebundle_html_repr_w_all_rows(scalars_dfs, session):
 
     # When there are 10 or fewer rows, the outputs should be identical except for the extra note.
     bundle = scalars_df.head(10)._repr_mimebundle_()
+    if isinstance(bundle, tuple):
+        bundle = bundle[0]
     actual = bundle["text/html"]
 
     with display_options.pandas_repr(bigframes.options.display):