From 2bf57af787c3e577a0846bc35e77e0f463efb492 Mon Sep 17 00:00:00 2001 From: Dave hughes <2043653+dwhu@users.noreply.github.com> Date: Fri, 3 Jan 2020 10:17:23 -0800 Subject: [PATCH] TST: Adding Test for merge_asof groupby=multiple with categorical column. --- pandas/tests/reshape/merge/test_merge_asof.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/pandas/tests/reshape/merge/test_merge_asof.py b/pandas/tests/reshape/merge/test_merge_asof.py index b2e764c5463fa..b461d1ec69152 100644 --- a/pandas/tests/reshape/merge/test_merge_asof.py +++ b/pandas/tests/reshape/merge/test_merge_asof.py @@ -1185,6 +1185,13 @@ def test_merge_datatype_categorical_error_raises(self): with pytest.raises(MergeError, match=msg): merge_asof(left, right, on="a") + def test_merge_groupby_multiple_column_with_categorical_column(self): + # GH 16454 + df = pd.DataFrame({"x": [0], "y": [0], "z": pd.Categorical([0])}) + result = merge_asof(df, df, on="x", by=["y", "z"]) + expected = pd.DataFrame({"x": [0], "y": [0], "z": pd.Categorical([0])}) + tm.assert_frame_equal(result, expected) + @pytest.mark.parametrize( "func", [lambda x: x, lambda x: to_datetime(x)], ids=["numeric", "datetime"] )