TST: tests for categorical apply (#25095)

alimcmaster1 · jreback · commit a488ab8c2a63 · 2019-02-03T12:44:31.000-05:00
diff --git a/pandas/tests/series/test_apply.py b/pandas/tests/series/test_apply.py
@@ -163,6 +163,18 @@ def test_apply_dict_depr(self):
         with tm.assert_produces_warning(FutureWarning):
             tsdf.A.agg({'foo': ['sum', 'mean']})
 
+    @pytest.mark.parametrize('series', [
+        ['1-1', '1-1', np.NaN],
+        ['1-1', '1-2', np.NaN]])
+    def test_apply_categorical_with_nan_values(self, series):
+        # GH 20714 bug fixed in: GH 24275
+        s = pd.Series(series, dtype='category')
+        result = s.apply(lambda x: x.split('-')[0])
+        result = result.astype(object)
+        expected = pd.Series(['1', '1', np.NaN], dtype='category')
+        expected = expected.astype(object)
+        tm.assert_series_equal(result, expected)
+
 
 class TestSeriesAggregate():
 
