add more tests

Author: sfc-gh-nkrishna

src/snowflake/snowpark/modin/plugin/_internal/aggregation_utils.py

@@ -899,7 +899,9 @@ def _is_supported_snowflake_agg_func(
     if isinstance(agg_func, tuple) and len(agg_func) == 2:
         # For named aggregations, like `df.agg(new_col=("old_col", "sum"))`,
         # take the second part of the named aggregation.
-        agg_func = agg_func[1]
+        agg_func = (
+            agg_func.func if isinstance(agg_func, AggFuncWithLabel) else agg_func[1]
+        )
 
     if get_snowflake_agg_func(agg_func, agg_kwargs, axis, _is_df_agg) is None:
         return AggregationSupportResult(

tests/integ/modin/groupby/test_groupby_basic_agg.py

@@ -1323,7 +1323,7 @@ def test_groupby_timedelta_var(self):
         )
 
     @sql_count_checker(query_count=1)
-    def test_groupby_agg_named_aggregation_codepath(self):
+    def test_groupby_agg_named_aggregation_implicit(self):
         eval_snowpark_pandas_result(
             *create_test_dfs(
                 {
@@ -1333,3 +1333,11 @@ def test_groupby_agg_named_aggregation_codepath(self):
             ),
             lambda df: df.groupby("team").agg({"score": [("total_score", "sum")]}),
         )
+
+    @sql_count_checker(query_count=1)
+    def test_groupby_agg_named_aggregation_explicit(self):
+        agg1 = pd.NamedAgg(column="score", aggfunc="sum")
+        eval_snowpark_pandas_result(
+            *create_test_dfs({"team": ["A", "B"], "score": [10, 15]}),
+            lambda df: df.groupby("team").agg(total_score=agg1),
+        )