fix(rust): fix iso_year for Date dtype (#5074)

pola-rs · Oct 2, 2022 · d192e3f · d192e3f
1 parent be5c600
commit d192e3f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/polars/polars-time/src/chunkedarray/kernels.rs b/polars/polars-time/src/chunkedarray/kernels.rs
@@ -68,7 +68,7 @@ to_temporal_unit!(
 #[cfg(feature = "dtype-date")]
 to_temporal_unit!(
     date_to_iso_year,
-    week,
+    iso_year,
     date32_to_datetime,
     i32,
     ArrowDataType::Int32

diff --git a/py-polars/tests/unit/test_datelike.py b/py-polars/tests/unit/test_datelike.py
@@ -1516,8 +1516,8 @@ def test_short_formats() -> None:
 
 
 def test_iso_year() -> None:
-    dt = datetime(2022, 1, 1, 7, 8, 40)
-    assert pl.Series([dt]).dt.iso_year()[0] == 2021
+    assert pl.Series([datetime(2022, 1, 1, 7, 8, 40)]).dt.iso_year()[0] == 2021
+    assert pl.Series([date(2022, 1, 1)]).dt.iso_year()[0] == 2021
 
 
 def test_invalid_date_parsing_4898() -> None: