From 8f5c81d7b91b3702cb96110f562a1c563fd0e7d6 Mon Sep 17 00:00:00 2001 From: Corrie Bartelheimer Date: Tue, 18 Apr 2023 16:18:52 +0200 Subject: [PATCH] adding test --- pandas/tests/frame/indexing/test_setitem.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/pandas/tests/frame/indexing/test_setitem.py b/pandas/tests/frame/indexing/test_setitem.py index 049874a4dd46f..672633cee2992 100644 --- a/pandas/tests/frame/indexing/test_setitem.py +++ b/pandas/tests/frame/indexing/test_setitem.py @@ -1251,3 +1251,14 @@ def test_setitem_column_update_inplace(self, using_copy_on_write): df[label][label] = 1 # original dataframe not updated assert np.all(values[np.arange(10), np.arange(10)] == 0) + + def test_setitem_column_frame_as_category(self): + # GH31581 + df = DataFrame([1, 2, 3]) + df["col1"] = DataFrame([1, 2, 3], dtype="category") + df["col2"] = Series([1, 2, 3], dtype="category") + + expected_types = Series( + ["int64", "category", "category"], index=[0, "col1", "col2"] + ) + tm.assert_series_equal(df.dtypes, expected_types)