round tests

Author: andrewgsavage

pandas/tests/extension/base/__init__.py

@@ -54,6 +54,7 @@ class TestMyDtype(BaseDtypeTests):
     BaseComparisonOpsTests,
     BaseOpsUtil,
     BaseUnaryOpsTests,
+    BaseRoundingTests,
 )
 from pandas.tests.extension.base.printing import BasePrintingTests
 from pandas.tests.extension.base.reduce import (  # noqa: F401
@@ -86,6 +87,7 @@ class ExtensionTests(
     BaseArithmeticOpsTests,
     BaseComparisonOpsTests,
     BaseUnaryOpsTests,
+    BaseRoundingTests,
     BasePrintingTests,
     BaseReduceTests,
     BaseReshapingTests,

pandas/tests/extension/base/ops.py

@@ -267,5 +267,5 @@ def test_unary_ufunc_dunder_equivalence(self, data, ufunc):
 class BaseRoundingTests:
     def test_round(self, data):
         result = pd.Series(data).round()
-        expected = pd.Series([np.round(item) for item in data], dtype = self.dtype)
+        expected = pd.Series(np.round(data), dtype = data.dtype)
         tm.assert_series_equal(result, expected)

pandas/tests/extension/test_masked.py

@@ -433,3 +433,12 @@ class TestParsing(base.BaseParsingTests):
 
 class Test2DCompat(base.Dim2CompatTests):
     pass
+
+class TestRounding(base.BaseRoundingTests):
+    def test_round(self, data, request):
+        if data.dtype == "boolean":
+            mark = pytest.mark.xfail(
+                reason="Cannot round boolean dtype"
+            )
+            request.node.add_marker(mark)
+        super().test_round(data)