fix named_expr with schema in struct and added corresponding test

pola-rs · Jul 21, 2024 · 50f7f0e · 50f7f0e
1 parent 7888d3b
commit 50f7f0e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/py-polars/polars/functions/as_datatype.py b/py-polars/polars/functions/as_datatype.py
@@ -598,7 +598,7 @@ def struct(
     pyexprs = parse_into_list_of_expressions(*exprs, **named_exprs)
 
     if schema:
-        if not exprs:
+        if not exprs and not named_exprs:
             # no columns or expressions provided; create one from schema keys
             expr = wrap_expr(
                 plr.as_struct(parse_into_list_of_expressions(list(schema.keys())))

diff --git a/py-polars/tests/unit/datatypes/test_struct.py b/py-polars/tests/unit/datatypes/test_struct.py
@@ -970,3 +970,11 @@ def test_struct_out_nullability_from_arrow() -> None:
 def test_empty_struct_raise() -> None:
     with pytest.raises(ValueError):
         pl.struct()
+
+
+def test_named_exprs() -> None:
+    df = pl.DataFrame({"a": 1})
+    schema = {"b": pl.Int64}
+    res = df.select(pl.struct(schema=schema, b=pl.col("a")))
+    expected = df.select(pl.struct(pl.col("a").alias("b"), schema=schema))
+    assert_frame_equal(res, expected)