raise ValueError instead of AssertionError

bigframes/series.py

@@ -1237,9 +1237,10 @@ def apply(
             # It is not a remote function
             # Then it must be a vectorized function that applies to the Series
             # as a whole
-            assert (
-                not by_row
-            ), "A vectorized non-remote function can be provided only with by_row=False"
+            if by_row:
+                raise ValueError(
+                    "A vectorized non-remote function can be provided only with by_row=False"
+                )
 
             try:
                 return func(self)

tests/system/small/test_series.py

@@ -3111,7 +3111,7 @@ def test_apply_lambda(scalars_dfs, col, lambda_):
     bf_col = scalars_df[col]
 
     # Can't be applied to BigFrames Series without by_row=False
-    with pytest.raises(AssertionError, match="by_row=False"):
+    with pytest.raises(ValueError, match="by_row=False"):
         bf_col.apply(lambda_)
 
     bf_result = bf_col.apply(lambda_, by_row=False).to_pandas()
@@ -3142,7 +3142,7 @@ def test_apply_numpy_ufunc(scalars_dfs, ufunc):
     bf_col = scalars_df["int64_col"]
 
     # Can't be applied to BigFrames Series without by_row=False
-    with pytest.raises(AssertionError, match="by_row=False"):
+    with pytest.raises(ValueError, match="by_row=False"):
         bf_col.apply(ufunc)
 
     bf_result = bf_col.apply(ufunc, by_row=False).to_pandas()
@@ -3162,7 +3162,7 @@ def foo(x):
     bf_col = scalars_df["int64_col"]
 
     # Can't be applied to BigFrames Series without by_row=False
-    with pytest.raises(AssertionError, match="by_row=False"):
+    with pytest.raises(ValueError, match="by_row=False"):
         bf_col.apply(foo)
 
     bf_result = bf_col.apply(foo, by_row=False).to_pandas()