From 42a0d0b6f74d1743a7810c33e229a3f94006f07d Mon Sep 17 00:00:00 2001
From: alimcmaster1 <alimcmaster1@gmail.com>
Date: Sun, 3 Feb 2019 17:44:31 +0000
Subject: [PATCH] TST: tests for categorical apply (#25095)

---
 pandas/tests/series/test_apply.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/pandas/tests/series/test_apply.py b/pandas/tests/series/test_apply.py
index 90cf6916df0d1d..162a27db34cb13 100644
--- 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():