SNOW-1869802: Fix local testing pivot returning None (#2824)

Author: sfc-gh-jrose

CHANGELOG.md

@@ -22,6 +22,7 @@
 
 #### Bug Fixes
 
+- Fixed a bug in local testing mode that caused a column to contain None when it should contain 0
 - Fixed a bug in StructField.from_json that prevented TimestampTypes with tzinfo from being parsed correctly.
 
 ### Snowpark pandas API Updates

src/snowflake/snowpark/mock/_plan.py

@@ -1881,9 +1881,9 @@ def flatten_object_cell_func(cell):
         # This requires us to wrap the aggregation function with extract logic to handle this special case.
         def agg_function(column):
             return (
-                agg_functions[agg_function_name](column.dropna())
-                if column.any()
-                else sentinel
+                sentinel
+                if column.isnull().all()
+                else agg_functions[agg_function_name](column.dropna())
             )
 
         default = (

tests/integ/scala/test_dataframe_aggregate_suite.py

@@ -65,6 +65,17 @@ def test_pivot(session):
     )
 
 
+def test_pivot_snow_1869802_repro(session):
+    df = session.create_dataframe(
+        [[1, "A", 10], [1, "B", 0], [2, "A", 11], [2, "B", 12]]
+    )
+    Utils.check_answer(
+        df.pivot(pivot_col="_2", values=["A", "B"]).function("min")("_3"),
+        [Row(1, 10, 0), Row(2, 11, 12)],
+        False,
+    )
+
+
 @pytest.mark.parametrize(
     "func,expected",
     [