Fix and add test numpy 2d array Series init (#1851)

pola-rs · Nov 21, 2021 · 0c11e73 · 0c11e73
1 parent 66dc057
commit 0c11e73
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/py-polars/polars/internals/construction.py b/py-polars/polars/internals/construction.py
@@ -80,7 +80,7 @@ def numpy_to_pyseries(
         else:
             return constructor(name, values, strict)
     else:
-        return PySeries.new_object(name, values)
+        return PySeries.new_object(name, values, strict)
 
 
 def _get_first_non_none(values: Sequence[Optional[Any]]) -> Any:

diff --git a/py-polars/tests/test_series.py b/py-polars/tests/test_series.py
@@ -37,6 +37,11 @@ def test_init_inputs() -> None:
     assert pl.Series(values=["foo", "bar"]).dtype == pl.Utf8
     assert pl.Series("a", [pl.Series([1, 2, 4]), pl.Series([3, 2, 1])]).dtype == pl.List
 
+    # 2d numpy array
+    res = pl.Series(name="a", values=np.array([[1, 2], [3, 4]]))
+    assert all(res[0] == np.array([1, 2]))
+    assert all(res[1] == np.array([3, 4]))
+
     # Bad inputs
     with pytest.raises(ValueError):
         pl.Series([1, 2, 3], [1, 2, 3])