diff --git a/skrub/_reporting/tests/test_table_report.py b/skrub/_reporting/tests/test_table_report.py index f57b9335a..6268c7e71 100644 --- a/skrub/_reporting/tests/test_table_report.py +++ b/skrub/_reporting/tests/test_table_report.py @@ -7,13 +7,29 @@ import numpy as np import pytest +from sklearn.utils import Bunch -from skrub import TableReport, ToDatetime, datasets +from skrub import TableReport, ToDatetime from skrub import _dataframe as sbd from skrub._reporting._sample_table import make_table from skrub.conftest import polars_installed_without_pyarrow +@pytest.fixture +def simple_df(df_module): + return df_module.make_dataframe( + { + "A": [1, 2, 3, 4, 5], + "B": ["a", "b", "a", "b", "c"], + } + ) + + +@pytest.fixture +def simple_series(df_module): + return df_module.make_column(name="A", values=[1, 2, 3, 4, 5]) + + def get_report_id(html): return re.search(r'