diff --git a/tests/_plugins/ui/_impl/tables/test_polars_table.py b/tests/_plugins/ui/_impl/tables/test_polars_table.py index 1ebeb1535c8..1f80452b337 100644 --- a/tests/_plugins/ui/_impl/tables/test_polars_table.py +++ b/tests/_plugins/ui/_impl/tables/test_polars_table.py @@ -983,8 +983,7 @@ def test_to_json_binary(self) -> None: data.write_json() - def test_to_json_enum_list_not_supported(self) -> None: - # When this is supported, we can remove the casting to string + def test_to_json_enum_list_supported(self) -> None: import polars as pl data = {"A": [["A", "B", "C"], ["A", "B", "C"], ["A", "B", "C"]]} @@ -992,9 +991,7 @@ def test_to_json_enum_list_not_supported(self) -> None: data_enum = pl.DataFrame( data, schema={"A": pl.List(pl.Enum(categories=["A", "B", "C"]))} ) - with pytest.raises(pl.exceptions.PanicException): - data_enum.write_json() + data_enum.write_json() data_list = pl.DataFrame(data, schema={"A": pl.List(pl.Categorical())}) - with pytest.raises(pl.exceptions.PanicException): - data_list.write_json() + data_list.write_json()