Move and parameterize test

Author: ThomasKluiters

pandas/tests/resample/test_datetime_index.py

@@ -1489,12 +1489,3 @@ def test_get_timestamp_range_edges(first, last, offset,
     expected = (exp_first, exp_last)
     assert result == expected
 
-
-def test_apply_on_series():
-    # GH 25959
-    # Calling apply on a localized time series should not cause an error
-    index = pd.DatetimeIndex([1478064900001000000, 1480037118776792000])\
-        .tz_localize('UTC')
-    result = pd.Series(index).apply(lambda x: pd.Series([1, 2]))
-    expected = pd.DataFrame([[1, 2], [1, 2]])
-    assert_frame_equal(result, expected)

pandas/tests/series/test_apply.py

@@ -677,3 +677,14 @@ def test_map_missing_mixed(self, vals, mapping, exp):
         result = s.map(mapping)
 
         tm.assert_series_equal(result, pd.Series(exp))
+
+    @pytest.mark.parametrize("dti,exp", [
+        (Series([1, 2], index=pd.DatetimeIndex([0, 31536000000])), DataFrame(np.repeat([[1, 2]], 2, axis=0))),
+        (tm.makeTimeSeries(nper=30), DataFrame(np.repeat([[1, 2]], 30, axis=0)))
+    ])
+    def test_apply_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)