TST: tests for categorical apply (pandas-dev#25095)

Pingviinituutti · Feb 28, 2019 · 42a0d0b · 42a0d0b
1 parent 4ef86d1
commit 42a0d0b
Showing 1 changed file with 12 additions and 0 deletions.
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():