add to at/iat as well

Author: jorisvandenbossche

pandas/core/indexing.py

@@ -2585,8 +2585,8 @@ def __getitem__(self, key):
         return super().__getitem__(key)
 
     def __setitem__(self, key, value) -> None:
-        if not PYPY:
-            if sys.getrefcount(self.obj) <= 2:
+        if not PYPY and not WARNING_CHECK_BROKEN:
+            if sys.getrefcount(self.obj) <= REF_COUNT:
                 warnings.warn(
                     _chained_assignment_msg, ChainedAssignmentError, stacklevel=2
                 )
@@ -2616,8 +2616,8 @@ def _convert_key(self, key):
         return key
 
     def __setitem__(self, key, value) -> None:
-        if not PYPY:
-            if sys.getrefcount(self.obj) <= 2:
+        if not PYPY and not WARNING_CHECK_BROKEN:
+            if sys.getrefcount(self.obj) <= REF_COUNT:
                 warnings.warn(
                     _chained_assignment_msg, ChainedAssignmentError, stacklevel=2
                 )