Add all_binary_operators

Author: fbourgey

pandas/tests/frame/test_api.py

@@ -324,21 +324,14 @@ def test_attrs_deepcopy(self):
         assert result.attrs == df.attrs
         assert result.attrs["tags"] is not df.attrs["tags"]
 
-    def test_attrs_binary_operations(self):
+    def test_attrs_binary_operations(self, all_binary_operators):
         # GH 51607
-        df_1 = DataFrame({"A": [2, 3]})
-        df_2 = DataFrame({"A": [-3, 9]})
+        df_1 = DataFrame([1])
+        df_2 = DataFrame([2])
         attrs = {"info": "DataFrame"}
         df_1.attrs = attrs
-        assert (df_1 + df_2).attrs == attrs
-        assert (df_2 + df_1).attrs == attrs
-        assert (df_2 - df_1).attrs == attrs
-        assert (df_2 / df_1).attrs == attrs
-        assert (df_2 * df_1).attrs == attrs
-        assert (df_2.add(df_1)).attrs == attrs
-        assert (df_2.sub(df_1)).attrs == attrs
-        assert (df_2.div(df_1)).attrs == attrs
-        assert (df_2.mul(df_1)).attrs == attrs
+        assert all_binary_operators(df_1, df_2).attrs == attrs
+        assert all_binary_operators(df_2, df_1).attrs == attrs
 
     @pytest.mark.parametrize("allows_duplicate_labels", [True, False, None])
     def test_set_flags(

pandas/tests/series/test_api.py

@@ -164,21 +164,14 @@ def test_attrs(self):
         result = s + 1
         assert result.attrs == {"version": 1}
 
-    def test_attrs_binary_operations(self):
+    def test_attrs_binary_operations(self, all_binary_operators):
         # GH 51607
-        s1 = Series([2, 5])
-        s2 = Series([7, -1])
+        s1 = Series([1])
+        s2 = Series([2])
         attrs = {"info": "Series"}
         s1.attrs = attrs
-        assert (s1 + s2).attrs == attrs
-        assert (s2 + s1).attrs == attrs
-        assert (s2 - s1).attrs == attrs
-        assert (s2 / s1).attrs == attrs
-        assert (s2 * s1).attrs == attrs
-        assert (s2.add(s1)).attrs == attrs
-        assert (s2.sub(s1)).attrs == attrs
-        assert (s2.div(s1)).attrs == attrs
-        assert (s2.mul(s1)).attrs == attrs
+        assert all_binary_operators(s1, s2).attrs == attrs
+        assert all_binary_operators(s2, s1).attrs == attrs
 
     @pytest.mark.xfail(
         using_string_dtype() and not HAS_PYARROW, reason="TODO(infer_string)"