foo",
+ "<bar> | ",
+ "<baz | ",
+ "spam> | ",
+ "1 | ",
+ "2 | ",
+ "3 | ",
+ ):
+ assert match in html, f"Expected to find {match!r} in html repr"
+
+
+def test_html_tables() -> None:
+ df = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]})
+
+ # default: header contains names/dtypes
+ header = "a | b | c |
---|
i64 | i64 | i64 |
"
+ assert header in df._repr_html_()
+
+ # validate that relevant config options are respected
+ with pl.Config(tbl_hide_column_names=True):
+ header = "i64 | i64 | i64 |
"
+ assert header in df._repr_html_()
+
+ with pl.Config(tbl_hide_column_data_types=True):
+ header = "a | b | c |
"
+ assert header in df._repr_html_()
+
+ with pl.Config(
+ tbl_hide_column_data_types=True,
+ tbl_hide_column_names=True,
+ ):
+ header = ""
+ assert header in df._repr_html_()
+
+
+def test_df_repr_html_max_rows_default() -> None:
+ df = pl.DataFrame({"a": range(50)})
+
+ html = df._repr_html_()
+
+ expected_rows = 10
+ assert html.count("") - 2 == expected_rows
+
+
+def test_df_repr_html_max_rows_odd() -> None:
+ df = pl.DataFrame({"a": range(50)})
+
+ with pl.Config(tbl_rows=9):
+ html = df._repr_html_()
+
+ expected_rows = 9
+ assert html.count(" | ") - 2 == expected_rows
+
+
+def test_series_repr_html_max_rows_default() -> None:
+ s = pl.Series("a", range(50))
+
+ html = s._repr_html_()
+
+ expected_rows = 10
+ assert html.count(" | ") - 2 == expected_rows
diff --git a/py-polars/tests/unit/test_config.py b/py-polars/tests/unit/test_config.py
index 17b58c7201c9..cfe98e6ede0c 100644
--- a/py-polars/tests/unit/test_config.py
+++ b/py-polars/tests/unit/test_config.py
@@ -87,30 +87,6 @@ def test_hide_header_elements() -> None:
)
-def test_html_tables() -> None:
- df = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]})
-
- # default: header contains names/dtypes
- header = "a | b | c |
---|
i64 | i64 | i64 | "
- assert header in df._repr_html_()
-
- # validate that relevant config options are respected
- with pl.Config(tbl_hide_column_names=True):
- header = "i64 | i64 | i64 | "
- assert header in df._repr_html_()
-
- with pl.Config(tbl_hide_column_data_types=True):
- header = "a | b | c | "
- assert header in df._repr_html_()
-
- with pl.Config(
- tbl_hide_column_data_types=True,
- tbl_hide_column_names=True,
- ):
- header = ""
- assert header in df._repr_html_()
-
-
def test_set_tbl_cols() -> None:
df = pl.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6], "c": [7, 8, 9]})
|