bug id fix

Author: shuoweil

bigframes/display/anywidget.py

@@ -297,7 +297,7 @@ def _set_table_html(self) -> None:
         page_data = cached_data.iloc[start:end].copy()
 
         # Handle index display
-        # TODO(b/332316283): Add tests for custom multiindex
+        # TODO(b/438181139): Add tests for custom multiindex
         if self._dataframe._block.has_index:
             index_name = page_data.index.name
             page_data.insert(

noxfile.py

@@ -66,6 +66,8 @@
 
 UNIT_TEST_PYTHON_VERSIONS = ["3.9", "3.10", "3.11", "3.12", "3.13"]
 UNIT_TEST_STANDARD_DEPENDENCIES = [
+    "mock",
+    "asyncmock",
     PYTEST_VERSION,
     "pytest-asyncio",
     "pytest-cov",
@@ -90,6 +92,7 @@
 SYSTEM_TEST_PYTHON_VERSIONS = ["3.9", "3.10", "3.11", "3.12", "3.13"]
 SYSTEM_TEST_STANDARD_DEPENDENCIES = [
     "jinja2",
+    "mock",
     "openpyxl",
     PYTEST_VERSION,
     "pytest-cov",
@@ -204,10 +207,7 @@ def lint_setup_py(session):
 
 def install_unittest_dependencies(session, install_test_extra, *constraints):
     standard_deps = UNIT_TEST_STANDARD_DEPENDENCIES + UNIT_TEST_DEPENDENCIES
-    session.install("setuptools", "wheel")
-    session.install(
-        "--extra-index-url", "https://pypi.org/simple/", *standard_deps, *constraints
-    )
+    session.install(*standard_deps, *constraints)
 
     if UNIT_TEST_LOCAL_DEPENDENCIES:
         session.install(*UNIT_TEST_LOCAL_DEPENDENCIES, *constraints)
@@ -224,7 +224,6 @@ def install_unittest_dependencies(session, install_test_extra, *constraints):
 
 def run_unit(session, install_test_extra):
     """Run the unit test suite."""
-    session.env["PIP_EXTRA_INDEX_URL"] = "https://pypi.org/simple/"
     constraints_path = str(
         CURRENT_DIRECTORY / "testing" / f"constraints-{session.python}.txt"
     )
@@ -314,13 +313,7 @@ def install_systemtest_dependencies(session, install_test_extra, *constraints):
     # See https://github.com/grpc/grpc/pull/30642
     session.install("--pre", "grpcio!=1.49.0rc1")
 
-    session.install("setuptools", "wheel")
-    session.install(
-        "--extra-index-url",
-        "https://pypi.org/simple/",
-        *SYSTEM_TEST_STANDARD_DEPENDENCIES,
-        *constraints,
-    )
+    session.install(*SYSTEM_TEST_STANDARD_DEPENDENCIES, *constraints)
 
     if SYSTEM_TEST_EXTERNAL_DEPENDENCIES:
         session.install(*SYSTEM_TEST_EXTERNAL_DEPENDENCIES, *constraints)
@@ -356,7 +349,7 @@ def run_system(
     timeout_seconds=900,
     num_workers=20,
 ):
-    session.env["PIP_EXTRA_INDEX_URL"] = "https://pypi.org/simple/"
+    """Run the system test suite."""
     constraints_path = str(
         CURRENT_DIRECTORY / "testing" / f"constraints-{session.python}.txt"
     )

tests/system/small/test_anywidget.py

@@ -1132,6 +1132,6 @@ def test_widget_with_custom_index_matches_pandas_output(
     assert "row_4" not in html  # Verify it respects max_rows
 
 
-# TODO(b/332316283): Add tests for custom multiindex
+# TODO(b/438181139): Add tests for custom multiindex
 # This may not be necessary for the SQL Cell use case but should be
 # considered for completeness.