round

Author: AmirPupko

pandas_to_sql/engine/columns/numeric_columns.py

@@ -64,6 +64,10 @@ def __neg__(self):
     return type(self)(sql_string=f'(-({value_to_sql_string(self)}))')
 
 
+def round_(self):
+    return FloatColumn(sql_string=f'(ROUND({value_to_sql_string(self)}))')
+
+
 
 add_comparison_operators_to_class(FloatColumn)
 FloatColumn.__add__ = __add__
@@ -78,6 +82,7 @@ def __neg__(self):
 FloatColumn.__rtruediv__ = __rtruediv__
 FloatColumn.__abs__ = __abs__
 FloatColumn.__neg__ = __neg__
+FloatColumn.round = round_
 
 
 add_comparison_operators_to_class(IntColumn)
@@ -93,4 +98,4 @@ def __neg__(self):
 IntColumn.__rtruediv__ = __rtruediv__
 IntColumn.__abs__ = __abs__
 IntColumn.__neg__ = __neg__
-
+IntColumn.round = round_

pandas_to_sql/testing/tests/test_operations_numeric.py

@@ -72,4 +72,9 @@ def test_floordiv():
 def test_rfloordiv():
     df = pytest.df1
     df['new_value'] = 1 // df.random_float
-    assert_(df)
+    assert_(df)
+
+def test_round():
+    df = pytest.df1
+    df['new_value'] = df.random_float.round()
+    assert_(df)