From 722276dfaccb0913aac6b2f877afb75f5e2f54a0 Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Wed, 8 Feb 2023 07:38:46 -0500 Subject: [PATCH] TST: categorical alignment --- pandas/tests/series/test_arithmetic.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/pandas/tests/series/test_arithmetic.py b/pandas/tests/series/test_arithmetic.py index a7f73f2e22fca..08fdad5ff1edd 100644 --- a/pandas/tests/series/test_arithmetic.py +++ b/pandas/tests/series/test_arithmetic.py @@ -287,6 +287,20 @@ def test_alignment_doesnt_change_tz(self): assert ser.index is dti assert ser_utc.index is dti_utc + def test_alignment_categorical(self): + # GH13365 + cat = Categorical(["3z53", "3z53", "LoJG", "LoJG", "LoJG", "N503"]) + ser1 = Series(2, index=cat) + ser2 = Series(2, index=cat[:-1]) + result = ser1 * ser2 + + exp_index = ["3z53"] * 4 + ["LoJG"] * 9 + ["N503"] + exp_index = pd.CategoricalIndex(exp_index, categories=cat.categories) + exp_values = [4.0] * 13 + [np.nan] + expected = Series(exp_values, exp_index) + + tm.assert_series_equal(result, expected) + def test_arithmetic_with_duplicate_index(self): # GH#8363