We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b2b81d8 commit 1d0ffc0Copy full SHA for 1d0ffc0
pandas/tests/frame/test_constructors.py
@@ -1004,6 +1004,20 @@ def test_constructor_dtype(self, data, index, columns, dtype, expected):
1004
df = DataFrame(data, index, columns, dtype)
1005
assert df.values.dtype == expected
1006
1007
+ @pytest.mark.parametrize(
1008
+ "data,input_dtype,expected_dtype",
1009
+ (
1010
+ ([True, False, None], "boolean", pd.BooleanDtype),
1011
+ ([1.0, 2.0, None], "Float64", pd.Float64Dtype),
1012
+ ([1, 2, None], "Int64", pd.Int64Dtype),
1013
+ ),
1014
+ )
1015
+ def test_constructor_dtype_nullable_extension_arrays(
1016
+ self, data, input_dtype, expected_dtype
1017
+ ):
1018
+ df = DataFrame({"a": data}, dtype=input_dtype)
1019
+ assert df["a"].dtype == expected_dtype()
1020
+
1021
def test_constructor_scalar_inference(self):
1022
data = {"int": 1, "bool": True, "float": 3.0, "complex": 4j, "object": "foo"}
1023
df = DataFrame(data, index=np.arange(10))
0 commit comments