Add regression test and fix PEP

pandas-dev · jreback · May 19, 2019 · Apr 2, 2019 · Apr 21, 2019 · Apr 29, 2019
commit 36429a04c240577f17fc8eefdaa1fa50a19125c3
diff --git a/pandas/core/series.py b/pandas/core/series.py
@@ -3705,7 +3705,8 @@ def f(x):
         if len(mapped) and isinstance(mapped[0], Series):
             # GH 25959 use pd.array instead of tolist
             # so extension arrays can be used
-            return self._constructor_expanddim(pd.array(mapped), index=self.index)
+            return self._constructor_expanddim(pd.array(mapped),
+                                               index=self.index)
         else:
             return self._constructor(mapped,
                                      index=self.index).__finalize__(self)

diff --git a/pandas/tests/series/test_apply.py b/pandas/tests/series/test_apply.py
@@ -678,9 +678,16 @@ def test_map_missing_mixed(self, vals, mapping, exp):
         (tm.makeTimeSeries(nper=30),
             DataFrame(np.repeat([[1, 2]], 30, axis=0), dtype='int64'))
     ])
-    def test_apply_on_date_time_index_aware_series(self, dti, exp):
+    def test_apply_series_on_date_time_index_aware_series(self, dti, exp):
         # GH 25959
         # Calling apply on a localized time series should not cause an error
         index = dti.tz_localize('UTC').index
         result = pd.Series(index).apply(lambda x: pd.Series([1, 2]))
         assert_frame_equal(result, exp)
+
+    def test_apply_scaler_on_date_time_index_aware_series(self):
+        # GH 25959
+        # Calling apply on a localized time series should not cause an error
+        series = tm.makeTimeSeries(nper=30).tz_localize('UTC')
+        result = pd.Series(series.index).apply(lambda x: 1)
+        assert_series_equal(result, pd.Series(np.ones(30), dtype='int64'))