Add tests for Series and DataFrame for attrs binary operations

Author: fbourgey

pandas/tests/frame/test_api.py

@@ -324,6 +324,22 @@ 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):
+        # GH 51607
+        df_1 = DataFrame({"A": [2, 3]})
+        df_2 = DataFrame({"A": [-3, 9]})
+        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
+
     @pytest.mark.parametrize("allows_duplicate_labels", [True, False, None])
     def test_set_flags(
         self,

pandas/tests/series/test_api.py

@@ -164,6 +164,22 @@ def test_attrs(self):
         result = s + 1
         assert result.attrs == {"version": 1}
 
+    def test_attrs_binary_operations(self):
+        # GH 51607
+        s1 = Series([2, 5])
+        s2 = Series([7, -1])
+        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
+
     @pytest.mark.xfail(
         using_string_dtype() and not HAS_PYARROW, reason="TODO(infer_string)"
     )