fix series to arrow for dates

pola-rs · Jan 7, 2022 · 84c7a81 · 84c7a81
1 parent 941c49a
commit 84c7a81
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/polars/polars-core/src/series/into.rs b/polars/polars-core/src/series/into.rs
@@ -11,7 +11,7 @@ impl Series {
     /// Convert a chunk in the Series to the correct Arrow type.
     /// This conversion is needed because polars doesn't use a
     /// 1 on 1 mapping for logical/ categoricals, etc.
-    pub(crate) fn to_arrow(&self, chunk_idx: usize) -> ArrayRef {
+    pub fn to_arrow(&self, chunk_idx: usize) -> ArrayRef {
         match self.dtype() {
             #[cfg(feature = "dtype-categorical")]
             DataType::Categorical => {

diff --git a/py-polars/src/series.rs b/py-polars/src/series.rs
@@ -761,7 +761,8 @@ impl PySeries {
         let gil = Python::acquire_gil();
         let py = gil.python();
         let pyarrow = py.import("pyarrow")?;
-        arrow_interop::to_py::to_py_array(self.series.chunks()[0].clone(), py, pyarrow)
+
+        arrow_interop::to_py::to_py_array(self.series.to_arrow(0), py, pyarrow)
     }
 
     #[cfg(feature = "is_in")]

diff --git a/py-polars/tests/test_datelike.py b/py-polars/tests/test_datelike.py
@@ -335,3 +335,11 @@ def test_truncate_negative_offset() -> None:
         datetime(2021, 5, 1),
         datetime(2021, 4, 1),
     ]
+
+
+def test_to_arrow() -> None:
+    date_series = pl.Series("dates", ["2022-01-16", "2022-01-17"]).str.strptime(
+        pl.Date, "%Y-%m-%d"
+    )
+    arr = date_series.to_arrow()
+    assert arr.type == pa.date32()