fix the tests

Author: jialuoo

bigframes/dataframe.py

@@ -4839,7 +4839,7 @@ def apply(self, func, *, axis=0, args: typing.Tuple = (), **kwargs):
                         f"Parameter count mismatch: BigFrames BigQuery function"
                         f" expected {len(udf_input_dtypes)} parameters but"
                         f" received {len(self.columns) + len(args)} values"
-                        f" ({len(self.columns)} DataFrame coulmns and"
+                        f" ({len(self.columns)} DataFrame columns and"
                         f" {len(args)} args)."
                     )
                 end_slice = -len(args) if args else None

tests/system/large/functions/test_managed_function.py

@@ -468,13 +468,13 @@ def foo(x, y, z):
         # Fails to apply on dataframe with incompatible number of columns.
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 2 DataFrame.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 2 DataFrame columns.",
         ):
             bf_df[["Id", "Age"]].apply(foo, axis=1)
 
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 4 DataFrame.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 4 DataFrame columns.",
         ):
             bf_df.assign(Country="lalaland").apply(foo, axis=1)
 
@@ -983,10 +983,10 @@ def the_sum(s1, s2, x):
 
         args1 = (1,)
 
-        # Fails to apply on dataframe with incompatible number of columns.
+        # Fails to apply on dataframe with incompatible number of columns and args.
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 4 values.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 4 values \\(3 DataFrame columns and 1 args\\)",
         ):
             scalars_df[columns + ["float64_col"]].apply(the_sum_mf, axis=1, args=args1)
 

tests/system/large/functions/test_remote_function.py

@@ -1956,10 +1956,10 @@ def the_sum(s1, s2, x):
 
         args1 = (1,)
 
-        # Fails to apply on dataframe with incompatible number of columns.
+        # Fails to apply on dataframe with incompatible number of columns and args.
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 4 values.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 4 values \\(2 DataFrame columns and 2 args\\)",
         ):
             scalars_df[columns].apply(
                 the_sum_mf,
@@ -2308,12 +2308,12 @@ def foo(x, y, z):
         # Fails to apply on dataframe with incompatible number of columns
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 2 DataFrame.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 2 DataFrame columns.",
         ):
             bf_df[["Id", "Age"]].apply(foo, axis=1)
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 4 DataFrame.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 4 DataFrame columns.",
         ):
             bf_df.assign(Country="lalaland").apply(foo, axis=1)
 
@@ -2392,12 +2392,12 @@ def foo(x, y, z):
         # Fails to apply on dataframe with incompatible number of columns
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 2 DataFrame.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 2 DataFrame columns.",
         ):
             bf_df[["Id", "Age"]].apply(foo, axis=1)
         with pytest.raises(
             ValueError,
-            match="^Parameter count mismatch:.* expected 3 parameters but received 4 DataFrame.*",
+            match="^Parameter count mismatch:.* expected 3 parameters but received 4 DataFrame columns.",
         ):
             bf_df.assign(Country="lalaland").apply(foo, axis=1)