TST: GH26055 fixed docs description and added axis_frame to test

Author: yhaque1213

doc/source/whatsnew/v0.25.0.rst

@@ -389,7 +389,8 @@ Groupby/Resample/Rolling
 - Ensured that result group order is correct when grouping on an ordered ``Categorical`` and specifying ``observed=True`` (:issue:`25871`, :issue:`25167`)
 - Bug in :meth:`pandas.core.window.Rolling.min` and :meth:`pandas.core.window.Rolling.max` that caused a memory leak (:issue:`25893`)
 - Bug in :func:`idxmax` and :func:`idxmin` on :meth:`DataFrame.groupby` with datetime column would return incorrect dtype (:issue:`25444`, :issue:`15306`)
-- Bug in :meth:`pandas.core.window._Rolling_and_Expanding.count` that did not include axis parameter in constructor call (:issue:`13503`)
+- Bug in :meth:`pandas.core.window.Rolling.count` was previously ignoring the axis keyword (:issue:`13503`)
+- Bug in :meth:`pandas.core.window.Expanding.count` was previously ignoring the axis keyword (:issue:`13503`)
 
 Reshaping
 ^^^^^^^^^

pandas/tests/test_window.py

@@ -667,21 +667,19 @@ def test_rolling_axis_sum(self, axis_frame):
         result = df.rolling(3, axis=axis_frame).sum()
         tm.assert_frame_equal(result, expected)
 
-    def test_rolling_axis_count(self):
+    def test_rolling_axis_count(self, axis_frame):
         # see gh-26055
         df = DataFrame({'x': range(3), 'y': range(3)})
 
-        result = df.rolling(2).count()
-        expected = DataFrame({'x': [1.0, 2.0, 2.0], 'y': [1.0, 2.0, 2.0]})
-        tm.assert_frame_equal(result, expected)
+        axis = df._get_axis_number(axis_frame)
 
-        # not specifiying axis and making axis=rows
-        # are expected to yield the same result
-        result = df.rolling(2, axis='rows').count()
-        tm.assert_frame_equal(result, expected)
+        if axis == 0:
+            expected = DataFrame({'x': [1.0, 2.0, 2.0], 'y': [1.0, 2.0, 2.0]})
+        else:
+            # axis == 1
+            expected = DataFrame({'x': [1.0, 1.0, 1.0], 'y': [2.0, 2.0, 2.0]})
 
-        result = df.rolling(2, axis='columns').count()
-        expected = DataFrame({'x': [1.0, 1.0, 1.0], 'y': [2.0, 2.0, 2.0]})
+        result = df.rolling(2, axis=axis_frame).count()
         tm.assert_frame_equal(result, expected)