chore: Fix incorrect debug assertion in `ChunkedArray::from_chunks_an…

…d_dtype`
pola-rs · Jun 4, 2024 · ac3ca3f · ac3ca3f
1 parent ea086d6
commit ac3ca3f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/crates/polars-core/src/chunked_array/from.rs b/crates/polars-core/src/chunked_array/from.rs
@@ -207,7 +207,10 @@ where
         #[cfg(debug_assertions)]
         {
             if !chunks.is_empty() && dtype.is_primitive() {
-                assert_eq!(chunks[0].data_type(), &dtype.to_physical().to_arrow(true))
+                assert_eq!(
+                    chunks[0].data_type().underlying_physical_type(),
+                    dtype.to_arrow(true)
+                )
             }
         }
         let field = Arc::new(Field::new(name, dtype));

diff --git a/py-polars/tests/unit/series/test_to_list.py b/py-polars/tests/unit/series/test_to_list.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from hypothesis import given
+from hypothesis import example, given
 
 import polars as pl
 from polars.testing import assert_series_equal
@@ -14,6 +14,7 @@
         allow_time_zones=False,
     )
 )
+@example(s=pl.Series(dtype=pl.Array(pl.Date, 1)))
 def test_to_list(s: pl.Series) -> None:
     values = s.to_list()
     result = pl.Series(values, dtype=s.dtype)