Skip to content

Commit 806c9cb

Browse files
ganevgvproost
authored andcommitted
add test for pd.melt dtypes preservation (pandas-dev#29510)
1 parent c1f1c24 commit 806c9cb

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

pandas/tests/reshape/test_melt.py

+10
Original file line numberDiff line numberDiff line change
@@ -270,6 +270,16 @@ def test_pandas_dtypes(self, col):
270270
expected.columns = ["klass", "col", "attribute", "value"]
271271
tm.assert_frame_equal(result, expected)
272272

273+
def test_preserve_category(self):
274+
# GH 15853
275+
data = DataFrame({"A": [1, 2], "B": pd.Categorical(["X", "Y"])})
276+
result = pd.melt(data, ["B"], ["A"])
277+
expected = DataFrame(
278+
{"B": pd.Categorical(["X", "Y"]), "variable": ["A", "A"], "value": [1, 2]}
279+
)
280+
281+
tm.assert_frame_equal(result, expected)
282+
273283
def test_melt_missing_columns_raises(self):
274284
# GH-23575
275285
# This test is to ensure that pandas raises an error if melting is

0 commit comments

Comments
 (0)