Update for Tuple type.

Author: Genesis929

bigframes/dataframe.py

@@ -2085,17 +2085,17 @@ def pivot_table(
         columns: typing.Union[blocks.Label, Sequence[blocks.Label]] = None,
         aggfunc: str = "mean",
     ) -> DataFrame:
-        if isinstance(index, blocks.Label):
+        if isinstance(index, blocks.Label) and index in self.columns:
             index = [index]
         else:
             index = list(index)
 
-        if isinstance(columns, blocks.Label):
+        if isinstance(columns, blocks.Label) and columns in self.columns:
             columns = [columns]
         else:
             columns = list(columns)
 
-        if isinstance(values, blocks.Label):
+        if isinstance(values, blocks.Label) and values in self.columns:
             values = [values]
         else:
             values = list(values)

tests/system/small/test_dataframe.py

@@ -2557,8 +2557,8 @@ def test_df_pivot_hockey(hockey_df, hockey_pandas_df, values, index, columns):
 @pytest.mark.parametrize(
     ("values", "index", "columns", "aggfunc"),
     [
-        (["culmen_length_mm", "body_mass_g"], "species", "sex", "std"),
-        (["body_mass_g", "culmen_length_mm"], ["species", "island"], "sex", "sum"),
+        (("culmen_length_mm", "body_mass_g"), "species", "sex", "std"),
+        (["body_mass_g", "culmen_length_mm"], ("species", "island"), "sex", "sum"),
         ("body_mass_g", "sex", ["island", "species"], "mean"),
         ("culmen_depth_mm", "island", "species", "max"),
     ],

third_party/bigframes_vendored/pandas/core/frame.py

@@ -4732,6 +4732,7 @@ def pivot_table(self, values=None, index=None, columns=None, aggfunc="mean"):
 
             columns (str or object or a list of str):
                 Column to use to make new frame's columns.
+
             aggfunc (str, default "mean"):
                 Aggregation function name to compute summary statistics (e.g., 'sum', 'mean').