fix(rust, python): fix coalesce expreession expansion (#5521)

pola-rs · Nov 16, 2022 · 35c53af · 35c53af
1 parent 49058bb
commit 35c53af
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/polars/polars-lazy/polars-plan/src/dsl/functions.rs b/polars/polars-lazy/polars-plan/src/dsl/functions.rs
@@ -1035,6 +1035,7 @@ pub fn coalesce(exprs: &[Expr]) -> Expr {
         options: FunctionOptions {
             collect_groups: ApplyOptions::ApplyGroups,
             cast_to_supertypes: true,
+            input_wildcard_expansion: true,
             ..Default::default()
         },
     }

diff --git a/py-polars/polars/internals/lazy_functions.py b/py-polars/polars/internals/lazy_functions.py
@@ -2093,7 +2093,8 @@ def arg_where(
 def coalesce(
     exprs: Sequence[
         pli.Expr | str | date | datetime | timedelta | int | float | bool | pli.Series
-    ],
+    ]
+    | pli.Expr,
 ) -> pli.Expr:
     """
     Folds the expressions from left to right, keeping the first non-null value.

diff --git a/py-polars/tests/unit/test_functions.py b/py-polars/tests/unit/test_functions.py
@@ -248,6 +248,9 @@ def test_coalesce() -> None:
     assert df.select(pl.coalesce(["a", "b", "c", 10])).to_dict(False) == {
         "a": [1.0, 2.0, 3.0, 10.0]
     }
+    assert df.select(pl.coalesce(pl.col(["a", "b", "c"]))).to_dict(False) == {
+        "a": [1.0, 2.0, 3.0, None]
+    }
 
 
 def test_ones_zeros() -> None: