struct logical type arrow conversion (#3193)

pola-rs · Apr 20, 2022 · 0f00810 · 0f00810
1 parent aa03cac
commit 0f00810
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/polars/polars-core/src/chunked_array/logical/struct_/mod.rs b/polars/polars-core/src/chunked_array/logical/struct_/mod.rs
@@ -19,14 +19,14 @@ pub struct StructChunked {
 }
 
 /// Returns an ['ArrayRef'](arrow::array::ArrayRef) for a given
-/// [`Series`], handling nested Struct-type series seperately.
+/// [`Series`], handling nested Struct-type series separately.
 fn array_ref_for_series(series: &Series) -> ArrayRef {
     match series.dtype() {
         DataType::Struct(_) => {
             let s = series.struct_().unwrap();
             s.arrow_array.clone()
         }
-        _ => series.chunks()[0].clone() as ArrayRef,
+        _ => series.to_arrow(0),
     }
 }
 

diff --git a/py-polars/tests/test_struct.py b/py-polars/tests/test_struct.py
@@ -1,3 +1,5 @@
+from datetime import datetime
+
 import pandas as pd
 import pytest
 
@@ -172,3 +174,9 @@ def test_struct_to_pandas() -> None:
     assert isinstance(pl_df.dtypes[0], pl.datatypes.Struct)
 
     assert pl_df.to_pandas().equals(df)
+
+
+def test_struct_logical_types_to_pandas() -> None:
+    timestamp = datetime(2022, 1, 1)
+    df = pd.DataFrame([{"struct": {"timestamp": timestamp}}])
+    assert pl.from_pandas(df).dtypes == [pl.Struct]