Skip to content

Commit 4cb54c7

Browse files
ZanirPZanirP
ZanirP
authored and
ZanirP
committed
test_replace.py working
1 parent 4d73daa commit 4cb54c7

File tree

2 files changed

+19
-18
lines changed

2 files changed

+19
-18
lines changed

pandas/core/generic.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -6463,7 +6463,7 @@ def astype(
64636463
warnings.warn(
64646464
"The 'category' dtype is being set to ordered=False by default.",
64656465
DeprecationWarning,
6466-
stacklevel=3,
6466+
stacklevel=find_stack_level()
64676467
)
64686468

64696469
if isinstance(dtype, CategoricalDtype):

pandas/tests/frame/methods/test_replace.py

+18-17
Original file line numberDiff line numberDiff line change
@@ -1311,26 +1311,27 @@ def test_replace_value_category_type(self):
13111311
expected = DataFrame(data=expected_dict).astype(
13121312
{"col2": "category", "col4": "category"}
13131313
)
1314+
# GH#61074
13141315
expected["col2"] = expected["col2"].cat.reorder_categories(
1315-
["a", "b", "c", "z"], ordered=True
1316+
["a", "b", "c", "z"], ordered=False
13161317
)
13171318
expected["col4"] = expected["col4"].cat.reorder_categories(
1318-
["cat1", "catX", "cat3", "cat4"], ordered=True
1319-
)
1320-
1321-
# replace values in input dataframe
1322-
input_df = input_df.apply(
1323-
lambda x: x.astype("category").cat.rename_categories({"d": "z"})
1324-
)
1325-
input_df = input_df.apply(
1326-
lambda x: x.astype("category").cat.rename_categories({"obj1": "obj9"})
1327-
)
1328-
result = input_df.apply(
1329-
lambda x: x.astype("category").cat.rename_categories({"cat2": "catX"})
1330-
)
1331-
1332-
result = result.astype({"col1": "int64", "col3": "float64", "col5": "str"})
1333-
tm.assert_frame_equal(result, expected)
1319+
["cat1", "catX", "cat3", "cat4"], ordered=False
1320+
)
1321+
1322+
# replace values in input dataframe
1323+
# GH#61074
1324+
msg = "The 'category' dtype is being set to ordered=False by default."
1325+
for col in ["col2", "col4"]:
1326+
if input_df[col].dtype.ordered:
1327+
with tm.assert_produces_warning(DeprecationWarning, match=msg):
1328+
input_df[col] = input_df[col].astype("category")
1329+
1330+
input_df["col5"] = input_df["col5"].astype("category")
1331+
1332+
input_df["col2"] = input_df["col2"].cat.rename_categories({"d": "z"})
1333+
input_df["col4"] = input_df["col4"].cat.rename_categories({"cat2": "catX"})
1334+
input_df["col5"] = input_df["col5"].cat.rename_categories({"obj1": "obj9"})
13341335

13351336
def test_replace_dict_category_type(self):
13361337
"""

0 commit comments

Comments
 (0)