Add nunique to an additional set of pivot tests

Author: sfc-gh-jkew

tests/integ/modin/pivot/test_pivot_table_dropna.py

@@ -107,7 +107,7 @@ def test_pivot_table_single_all_aggfuncs_dropna_and_null_data(
     df_data_with_nulls_2,
     values,
 ):
-    expected_join_count = 10 if len(values) > 1 else 5
+    expected_join_count = 12 if len(values) > 1 else 6
     with SqlCounter(query_count=1, join_count=expected_join_count):
         pivot_table_test_helper(
             df_data_with_nulls_2,
@@ -116,7 +116,7 @@ def test_pivot_table_single_all_aggfuncs_dropna_and_null_data(
                 "columns": ["C"],
                 "values": values,
                 "dropna": False,
-                "aggfunc": ["count", "sum", "min", "max", "mean"],
+                "aggfunc": ["count", "sum", "min", "max", "mean", "nunique"],
             },
         )
 

tests/integ/modin/pivot/test_pivot_table_margins.py

@@ -134,7 +134,7 @@ def test_pivot_table_multiple_columns_values_with_margins(
         ),
     ],
 )
-@sql_count_checker(query_count=1, join_count=5, union_count=1)
+@sql_count_checker(query_count=1, join_count=6, union_count=1)
 def test_pivot_table_multiple_pivot_values_null_data_with_margins(
     df_data_with_nulls, index, fill_value
 ):
@@ -144,7 +144,7 @@ def test_pivot_table_multiple_pivot_values_null_data_with_margins(
             "index": index,
             "columns": "C",
             "values": "F",
-            "aggfunc": ["count", "sum", "mean"],
+            "aggfunc": ["count", "sum", "mean", "nunique"],
             "dropna": False,
             "fill_value": fill_value,
             "margins": True,

tests/integ/modin/pivot/test_pivot_table_multiple.py

@@ -38,14 +38,14 @@ def test_pivot_table_no_index_single_column_multiple_values(df_data):
     )
 
 
-@sql_count_checker(query_count=1, union_count=1, join_count=2)
+@sql_count_checker(query_count=1, union_count=1, join_count=4)
 def test_pivot_table_no_index_single_column_multiple_values_multiple_aggr_func(df_data):
     pivot_table_test_helper(
         df_data,
         {
             "columns": "B",
             "values": ["D", "E"],
-            "aggfunc": ["mean", "max"],
+            "aggfunc": ["mean", "max", "nunique"],
         },
     )
 
@@ -119,7 +119,7 @@ def test_pivot_table_single_index_multiple_column_single_value(
     )
 
 
-@pytest.mark.parametrize("aggfunc", ["count", "sum", "min", "max", "mean"])
+@pytest.mark.parametrize("aggfunc", ["count", "sum", "min", "max", "mean", "nunique"])
 @pytest.mark.parametrize("values", ["D", ["D"]])
 @sql_count_checker(query_count=1)
 def test_pivot_table_no_index_multiple_column_single_value(df_data, aggfunc, values):

tests/integ/modin/pivot/test_pivot_table_single.py

@@ -54,6 +54,7 @@ def test_pivot_table_no_index_single_column_single_value(df_data):
         np.std,
         "var",
         np.var,
+        "nunique",
     ),
 )
 @sql_count_checker(query_count=1)
@@ -77,6 +78,7 @@ def test_pivot_table_single_index_single_column_single_value(df_data, aggfunc):
         "min",
         "max",
         "mean",
+        "nunique",
     ],
 )
 @sql_count_checker(query_count=1)
@@ -95,6 +97,7 @@ def test_pivot_table_multi_index_single_column_single_value(df_data, aggfunc):
         "min",
         "max",
         "mean",
+        "nunique",
     ],
 )
 @sql_count_checker(query_count=1)
@@ -201,6 +204,7 @@ def test_pivot_table_with_duplicate_values(
     [
         "count",
         "sum",
+        "nunique",
     ],
 )
 @pytest.mark.parametrize(
@@ -228,6 +232,7 @@ def test_pivot_table_with_sum_and_count_null_and_empty_values_matching_behavior(
     [
         "count",
         "sum",
+        "nunique",
     ],
 )
 def test_pivot_table_with_sum_and_count_null_and_empty_values_matching_behavior_skipped(