Fix Series.to_physical (#2123)

pola-rs · Dec 22, 2021 · 29350bb · 29350bb
1 parent 17c5568
commit 29350bb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/py-polars/polars/internals/series.py b/py-polars/polars/internals/series.py
@@ -1791,7 +1791,7 @@ def to_physical(self) -> "Series":
         Time -> Int64
         other -> other
         """
-        return wrap_s(self._s.to_physical)
+        return wrap_s(self._s.to_physical())
 
     def to_list(self, use_pyarrow: bool = False) -> List[Optional[Any]]:
         """

diff --git a/py-polars/tests/test_series.py b/py-polars/tests/test_series.py
@@ -1112,3 +1112,14 @@ def test_shuffle() -> None:
 
     out = pl.select(pl.lit(a).shuffle(2)).to_series()
     testing.assert_series_equal(out, expected)
+
+
+def test_to_physical() -> None:
+    # casting an int result in an int
+    a = pl.Series("a", [1, 2, 3])
+    testing.assert_series_equal(a.to_physical(), a)
+
+    # casting a date results in an Int32
+    a = pl.Series("a", [date(2020, 1, 1)] * 3)
+    expected = pl.Series("a", [18262] * 3, dtype=Int32)
+    testing.assert_series_equal(a.to_physical(), expected)