Cover to_list

posit-dev · Oct 17, 2024 · 8b9b8de · 8b9b8de
1 parent 492ee51
commit 8b9b8de
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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)