feat: Add __abs__ to dataframe (#2186)

Author: TrevorBergeron

bigframes/dataframe.py

@@ -1306,6 +1306,11 @@ def __pos__(self) -> DataFrame:
     def __neg__(self) -> DataFrame:
         return self._apply_unary_op(ops.neg_op)
 
+    def __abs__(self) -> DataFrame:
+        return self._apply_unary_op(ops.abs_op)
+
+    __abs__.__doc__ = abs.__doc__
+
     def align(
         self,
         other: typing.Union[DataFrame, bigframes.series.Series],

bigframes/series.py

@@ -1364,6 +1364,8 @@ def update(self, other: Union[Series, Sequence, Mapping]) -> None:
     def __abs__(self) -> Series:
         return self.abs()
 
+    __abs__.__doc__ = inspect.getdoc(vendored_pandas_series.Series.abs)
+
     def abs(self) -> Series:
         return self._apply_unary_op(ops.abs_op)
 

tests/system/small/test_dataframe.py

@@ -2460,6 +2460,16 @@ def test_df_neg(scalars_dfs):
     assert_pandas_df_equal(pd_result, bf_result)
 
 
+def test_df__abs__(scalars_dfs):
+    scalars_df, scalars_pandas_df = scalars_dfs
+    bf_result = (
+        abs(scalars_df[["int64_col", "numeric_col", "float64_col"]])
+    ).to_pandas()
+    pd_result = abs(scalars_pandas_df[["int64_col", "numeric_col", "float64_col"]])
+
+    assert_pandas_df_equal(pd_result, bf_result)
+
+
 def test_df_invert(scalars_dfs):
     scalars_df, scalars_pandas_df = scalars_dfs
     columns = ["int64_col", "bool_col"]