Fix whatsnew and add tests

Author: reidy-p

doc/source/whatsnew/v0.24.0.txt

@@ -225,7 +225,7 @@ Plotting
 Groupby/Resample/Rolling
 ^^^^^^^^^^^^^^^^^^^^^^^^
 
-- Bug in :func:`pandas.core.groupby.first` and :func:`pandas.core.groupby.last` with ``as_index=False`` leading to the loss of timezone information (:issue:`15884`)
+- Bug in :func:`pandas.core.groupby.GroupBy.first` and :func:`pandas.core.groupby.GroupBy.last` with ``as_index=False`` leading to the loss of timezone information (:issue:`15884`)
 -
 -
 

pandas/tests/groupby/test_nth.py

@@ -233,17 +233,20 @@ def test_nth_multi_index(three_group):
                Timestamp('2012-02-01 14:00:00',
                          tz='US/Central'),
                Timestamp('2012-03-01 12:00:00',
-                         tz='Europe/London')]},
+                         tz='Europe/London')],
+      'foo': [1, 2, 3]},
      {'id': ['A', 'B'],
       'time': [Timestamp('2012-01-01 13:00:00',
                          tz='America/New_York'),
                Timestamp('2012-02-01 14:00:00',
-                         tz='US/Central')]},
+                         tz='US/Central')],
+      'foo': [1, 2]},
      {'id': ['A', 'B'],
       'time': [Timestamp('2012-03-01 12:00:00',
                          tz='Europe/London'),
                Timestamp('2012-02-01 14:00:00',
-                         tz='US/Central')]})
+                         tz='US/Central')],
+      'foo': [3, 2]})
 ])
 def test_first_last_tz(data, expected_first, expected_last):
     # GH15884
@@ -256,10 +259,15 @@ def test_first_last_tz(data, expected_first, expected_last):
     expected = DataFrame(expected_first)
     assert_frame_equal(result, expected)
 
+    result = df.groupby('id', as_index=False)['time'].first()
+    assert_frame_equal(result, expected[['id', 'time']])
+
     result = df.groupby('id', as_index=False).last()
     expected = DataFrame(expected_last)
     assert_frame_equal(result, expected)
 
+    result = df.groupby('id', as_index=False)['time'].last()
+    assert_frame_equal(result, expected[['id', 'time']])
 
 def test_nth_multi_index_as_expected():
     # PR 9090, related to issue 8979