GH1045 PR Feedback

Author: loicdiridollou

tests/test_frame.py

@@ -1064,21 +1064,35 @@ def test_types_groupby_as_index() -> None:
         ),
         pd.Series,
     )
+    check(
+        assert_type(
+            df.groupby("a").size(),
+            "pd.Series[int]",
+        ),
+        pd.Series,
+    )
 
 
 def test_types_groupby_as_index_list() -> None:
     """Test type of groupby.size method depending on list of grouper GH1045."""
     df = pd.DataFrame({"a": [1, 1, 2], "b": [2, 3, 2]})
     check(
         assert_type(
-            df.groupby("a", as_index=False).size(),
+            df.groupby(["a", "b"], as_index=False).size(),
             pd.DataFrame,
         ),
         pd.DataFrame,
     )
     check(
         assert_type(
-            df.groupby("a", as_index=True).size(),
+            df.groupby(["a", "b"], as_index=True).size(),
+            "pd.Series[int]",
+        ),
+        pd.Series,
+    )
+    check(
+        assert_type(
+            df.groupby(["a", "b"]).size(),
             "pd.Series[int]",
         ),
         pd.Series,