test: enhance DataFrame HTML representation tests for structure and values

Author: kosiew

python/tests/test_dataframe.py

@@ -1493,8 +1493,8 @@ def add_with_parameter(df_internal, value: Any) -> DataFrame:
     assert result["new_col"] == [3 for _i in range(3)]
 
 
-def test_dataframe_repr_html(df) -> None:
-    """Test that DataFrame._repr_html_ produces expected HTML output."""
+def test_dataframe_repr_html_structure(df) -> None:
+    """Test that DataFrame._repr_html_ produces expected HTML output structure."""
 
     output = df._repr_html_()
 
@@ -1514,9 +1514,32 @@ def test_dataframe_repr_html(df) -> None:
     assert len(re.findall(body_pattern, output, re.DOTALL)) == 1
 
 
-def test_dataframe_repr_html(df):
-    """Test that DataFrame._repr_html_ produces expected HTML output."""
-    import re
+def test_dataframe_repr_html_values(df):
+    """Test that DataFrame._repr_html_ contains the expected data values."""
+    html = df._repr_html_()
+    assert html is not None
+
+    # Create a more flexible pattern that handles values being wrapped in spans
+    # This pattern will match the sequence of values 1,4,8,2,5,5,3,6,8 regardless of formatting
+    pattern = re.compile(
+        r"<td[^>]*?>(?:<span[^>]*?>)?1(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?4(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?8(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?2(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?5(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?5(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?3(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?6(?:</span>)?</td>.*?"
+        + r"<td[^>]*?>(?:<span[^>]*?>)?8(?:</span>)?</td>",
+        re.DOTALL,
+    )
+
+    # Print debug info if the test fails
+    matches = re.findall(pattern, html)
+    if not matches:
+        print(f"HTML output snippet: {html[:500]}...")
+
+    assert len(matches) > 0, "Expected pattern of values not found in HTML output"
 
     html = df._repr_html_()
     assert html is not None