diff --git a/tests/test_tbl_data.py b/tests/test_tbl_data.py index 2391ebae5..335c06fb3 100644 --- a/tests/test_tbl_data.py +++ b/tests/test_tbl_data.py @@ -21,6 +21,7 @@ is_series, reorder, to_frame, + to_list, validate_frame, copy_frame, ) @@ -298,6 +299,12 @@ def test_is_series_false(): assert not is_series(1) +def test_to_list(ser: SeriesLike): + pylist = to_list(ser) + assert len(pylist) == 3 + assert pylist[:2] == [1.0, 2.0] + + def test_cast_frame_to_string_polars_list_col(): df = pl.DataFrame({"x": [[1, 2], [3]], "y": [1, None], "z": [{"a": 1}, {"a": 2}]}) new_df = cast_frame_to_string(df)