call finalize in DataFrame.count

Author: arw2019

pandas/core/frame.py

@@ -8571,7 +8571,8 @@ def count(self, axis=0, level=None, numeric_only=False):
         """
         axis = self._get_axis_number(axis)
         if level is not None:
-            return self._count_level(level, axis=axis, numeric_only=numeric_only)
+            result = self._count_level(level, axis=axis, numeric_only=numeric_only)
+            return result.__finalize__(self, method="count")
 
         if numeric_only:
             frame = self._get_numeric_data()
@@ -8594,7 +8595,7 @@ def count(self, axis=0, level=None, numeric_only=False):
                     counts, index=frame._get_agg_axis(axis)
                 )
 
-        return result.astype("int64")
+        return result.astype("int64").__finalize__(self, method="count")
 
     def _count_level(self, level, axis=0, numeric_only=False):
         if numeric_only:

pandas/tests/generic/test_finalize.py

@@ -238,14 +238,8 @@
         ),
         marks=not_implemented_mark,
     ),
-    pytest.param(
-        (pd.DataFrame, frame_data, operator.methodcaller("count")),
-        marks=not_implemented_mark,
-    ),
-    pytest.param(
-        (pd.DataFrame, frame_mi_data, operator.methodcaller("count", level="A")),
-        marks=not_implemented_mark,
-    ),
+    (pd.DataFrame, frame_data, operator.methodcaller("count")),
+    (pd.DataFrame, frame_mi_data, operator.methodcaller("count", level="A")),
     pytest.param(
         (pd.DataFrame, frame_data, operator.methodcaller("nunique")),
         marks=not_implemented_mark,