test: update tests

Author: shuoweil

tests/system/small/test_dataframe.py

@@ -969,8 +969,6 @@ 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,8 +746,6 @@ 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):