From 2dd0ac3e956a7ba1bc4806bef6241bb56c535875 Mon Sep 17 00:00:00 2001 From: mliu08 Date: Thu, 26 Oct 2023 23:08:22 -0700 Subject: [PATCH 1/2] add test for mixed dtypes of df col index #47382 --- pandas/tests/frame/test_arithmetic.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pandas/tests/frame/test_arithmetic.py b/pandas/tests/frame/test_arithmetic.py index c59edbae05739..553060be2ddd6 100644 --- a/pandas/tests/frame/test_arithmetic.py +++ b/pandas/tests/frame/test_arithmetic.py @@ -2132,3 +2132,12 @@ def test_enum_column_equality(): expected = Series([True, True, True], name=Cols.col1) tm.assert_series_equal(result, expected) + + +def test_mixed_col_index_dtype(): + # GH 47382 + df1 = DataFrame(columns=list("abc"), data=0.0, index=[0]) + df2 = DataFrame(columns=list("abc"), data=0.0, index=[0]) + df1.columns = df2.columns.astype("string") + df3 = df1 + df2 + assert df3.columns.dtype == "object" From 8b240f59f08bd1a02014cdaa5b727dabd21e8e8c Mon Sep 17 00:00:00 2001 From: mliu08 Date: Fri, 27 Oct 2023 18:29:07 -0700 Subject: [PATCH 2/2] modified test to use result and expected dfs --- pandas/tests/frame/test_arithmetic.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pandas/tests/frame/test_arithmetic.py b/pandas/tests/frame/test_arithmetic.py index 553060be2ddd6..d73bced941d0d 100644 --- a/pandas/tests/frame/test_arithmetic.py +++ b/pandas/tests/frame/test_arithmetic.py @@ -2136,8 +2136,9 @@ def test_enum_column_equality(): def test_mixed_col_index_dtype(): # GH 47382 - df1 = DataFrame(columns=list("abc"), data=0.0, index=[0]) + df1 = DataFrame(columns=list("abc"), data=1.0, index=[0]) df2 = DataFrame(columns=list("abc"), data=0.0, index=[0]) df1.columns = df2.columns.astype("string") - df3 = df1 + df2 - assert df3.columns.dtype == "object" + result = df1 + df2 + expected = DataFrame(columns=list("abc"), data=1.0, index=[0]) + tm.assert_frame_equal(result, expected)