GH1411 Add __and__ for BooleanArray and tests

Author: loicdiridollou

pandas-stubs/_typing.pyi

@@ -21,7 +21,6 @@ from typing import (
     SupportsIndex,
     TypeAlias,
     TypedDict,
-    Union,
     overload,
 )
 
@@ -554,7 +553,7 @@ IndexKeyFunc: TypeAlias = Callable[[Index], Index | AnyArrayLike] | None
 # types of `func` kwarg for DataFrame.aggregate and Series.aggregate
 # More specific than what is in pandas
 # following Union is here to make it ty compliant https://github.com/astral-sh/ty/issues/591
-AggFuncTypeBase: TypeAlias = Union[Callable, str, np.ufunc]  # noqa: UP007
+AggFuncTypeBase: TypeAlias = Callable | str | np.ufunc
 AggFuncTypeDictSeries: TypeAlias = Mapping[HashableT, AggFuncTypeBase]
 AggFuncTypeDictFrame: TypeAlias = Mapping[
     HashableT, AggFuncTypeBase | list[AggFuncTypeBase]

pandas-stubs/core/arrays/boolean.pyi

@@ -1,6 +1,7 @@
 from typing import Any
 
 from pandas.core.arrays.masked import BaseMaskedArray as BaseMaskedArray
+from typing_extensions import Self
 
 from pandas._libs.missing import NAType
 from pandas._typing import (
@@ -25,3 +26,4 @@ class BooleanArray(BaseMaskedArray):
     def __setitem__(self, key, value) -> None: ...
     def any(self, *, skipna: bool = ..., **kwargs: Any): ...
     def all(self, *, skipna: bool = ..., **kwargs: Any): ...
+    def __and__(self, other: BooleanArray) -> Self: ...

tests/test_extension.py

@@ -3,6 +3,7 @@
 import numpy as np
 import pandas as pd
 from pandas.arrays import IntegerArray
+from pandas.core.arrays.boolean import BooleanArray
 from pandas.core.indexers import check_array_indexer
 from typing_extensions import assert_type
 
@@ -56,3 +57,11 @@ def test_array_indexer() -> None:
     check(assert_type(check_array_indexer(arr, 1), int), int)
 
     check(assert_type(check_array_indexer(arr, slice(0, 1, 1)), slice), slice)
+
+
+def test_boolean_array() -> None:
+    """Test creation of and operations on BooleanArray GH1411."""
+    arr = pd.array([True], dtype="boolean")
+    check(assert_type(arr, BooleanArray), BooleanArray)
+    arr_and = arr.__and__(arr)
+    check(assert_type(arr_and, BooleanArray), BooleanArray)