Test from_polars large_list

huggingface · Jul 3, 2024 · 9bc5182 · 9bc5182
1 parent df13687
commit 9bc5182
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/tests/test_arrow_dataset.py b/tests/test_arrow_dataset.py
@@ -4846,6 +4846,15 @@ def test_categorical_dataset(tmpdir):
     assert entry["animals"] == "Flamingo"
 
 
+@require_polars
+def test_from_polars_large_list():
+    import polars as pl
+
+    df = pl.from_dict({"col_1": [[1, 2], [3, 4]]})
+    ds = Dataset.from_polars(df)
+    assert isinstance(ds, Dataset)
+
+
 @require_polars
 def test_polars_round_trip():
     ds = Dataset.from_dict({"x": [[1, 2], [3, 4, 5]], "y": ["a", "b"]})