BUG: Allow plain bools in ExtensionArray.equals (#34661)

Author: xhochy

pandas/core/arrays/base.py

@@ -738,7 +738,7 @@ def equals(self, other: "ExtensionArray") -> bool:
                 # boolean array with NA -> fill with False
                 equal_values = equal_values.fillna(False)
             equal_na = self.isna() & other.isna()
-            return (equal_values | equal_na).all().item()
+            return bool((equal_values | equal_na).all())
 
     def _values_for_factorize(self) -> Tuple[np.ndarray, Any]:
         """

pandas/tests/extension/arrow/arrays.py

@@ -8,6 +8,7 @@
 """
 import copy
 import itertools
+import operator
 from typing import Type
 
 import numpy as np
@@ -106,6 +107,27 @@ def astype(self, dtype, copy=True):
     def dtype(self):
         return self._dtype
 
+    def _boolean_op(self, other, op):
+        if not isinstance(other, type(self)):
+            raise NotImplementedError()
+
+        result = op(np.array(self._data), np.array(other._data))
+        return ArrowBoolArray(
+            pa.chunked_array([pa.array(result, mask=pd.isna(self._data.to_pandas()))])
+        )
+
+    def __eq__(self, other):
+        if not isinstance(other, type(self)):
+            return False
+
+        return self._boolean_op(other, operator.eq)
+
+    def __and__(self, other):
+        return self._boolean_op(other, operator.and_)
+
+    def __or__(self, other):
+        return self._boolean_op(other, operator.or_)
+
     @property
     def nbytes(self):
         return sum(
@@ -153,10 +175,12 @@ def _reduce(self, method, skipna=True, **kwargs):
         return op(**kwargs)
 
     def any(self, axis=0, out=None):
-        return self._data.to_pandas().any()
+        # Explicitly return a plain bool to reproduce GH-34660
+        return bool(self._data.to_pandas().any())
 
     def all(self, axis=0, out=None):
-        return self._data.to_pandas().all()
+        # Explicitly return a plain bool to reproduce GH-34660
+        return bool(self._data.to_pandas().all())
 
 
 class ArrowBoolArray(ArrowExtensionArray):

pandas/tests/extension/arrow/test_bool.py

@@ -29,6 +29,11 @@ def data_missing():
     return ArrowBoolArray.from_scalars([None, True])
 
 
+def test_basic_equals(data):
+    # https://github.com/pandas-dev/pandas/issues/34660
+    assert pd.Series(data).equals(pd.Series(data))
+
+
 class BaseArrowTests:
     pass