python: boolean numpy array to boolean type

pola-rs · Aug 12, 2021 · b87be32 · b87be32
1 parent 0847ecd
commit b87be32
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/py-polars/polars/datatypes.py b/py-polars/polars/datatypes.py
@@ -308,7 +308,7 @@ def polars_type_to_constructor(
         np.uint32: PySeries.new_u32,
         np.uint64: PySeries.new_u64,
         np.str_: PySeries.new_str,
-        bool: PySeries.new_bool,
+        np.bool_: PySeries.new_bool,
     }
 
 

diff --git a/py-polars/tests/test_series.py b/py-polars/tests/test_series.py
@@ -19,10 +19,14 @@ def test_init_inputs():
     pl.Series(name="a", values=[1, 2])
     pl.Series(values=[1, 2], name="a")
 
-    pl.Series([1, 2])
-    pl.Series(values=[1, 2])
-    pl.Series()
-    pl.Series("a")
+    assert pl.Series([1, 2]).dtype == pl.Int64
+    assert pl.Series(values=[1, 2]).dtype == pl.Int64
+    assert pl.Series("a").dtype == pl.Float32  # f32 type used in case of no data
+    assert pl.Series().dtype == pl.Float32
+    assert pl.Series(values=[True, False]).dtype == pl.Boolean
+    assert pl.Series(values=np.array([True, False])).dtype == pl.Boolean
+    assert pl.Series(values=np.array(["foo", "bar"])).dtype == pl.Utf8
+    assert pl.Series(values=["foo", "bar"]).dtype == pl.Utf8
 
     # Bad inputs
     with pytest.raises(ValueError):