Skip to content

Commit 64a366d

Browse files
committed
Test assignment from a pyarrow array and a series
1 parent bf98984 commit 64a366d

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pandas/tests/extension/test_arrow.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1071,8 +1071,9 @@ def test_comp_masked_numpy(self, masked_dtype, comparison_op):
10711071
def test_assign_column_in_dataframe(self, data):
10721072
df = pd.DataFrame(data=data, columns=["A"], dtype=data.dtype)
10731073
df["B"] = pa.array(data, type=data.dtype.pyarrow_dtype)
1074+
df["C"] = pd.Series(data)
10741075
result = df.dtypes
1075-
expected = pd.Series({"A": data.dtype, "B": data.dtype})
1076+
expected = pd.Series({"A": data.dtype, "B": data.dtype, "C": data.dtype})
10761077

10771078
tm.assert_series_equal(result, expected)
10781079

0 commit comments

Comments
 (0)