from typing import Any, Literal, NoReturn, assert_type

import numpy as np
from numpy._typing import NDArray, _AnyShape

type MaskedArray[ScalarT: np.generic] = np.ma.MaskedArray[_AnyShape, np.dtype[ScalarT]]
type _NoMaskType = np.bool[Literal[False]]
type _Array1D[ScalarT: np.generic] = np.ndarray[tuple[int], np.dtype[ScalarT]]

###

class MaskedArraySubclass[ScalarT: np.generic](np.ma.MaskedArray[_AnyShape, np.dtype[ScalarT]]): ...

class IntoMaskedArraySubClass[ScalarT: np.generic]:
    def __array__(self) -> MaskedArraySubclass[ScalarT]: ...

type MaskedArraySubclassC = MaskedArraySubclass[np.complex128]
type MaskedArraySubclassI = MaskedArraySubclass[np.intp]

AR_b: NDArray[np.bool]
AR_f4: NDArray[np.float32]
AR_i8: NDArray[np.int64]
AR_u4: NDArray[np.uint32]
AR_dt64: NDArray[np.datetime64]
AR_td64: NDArray[np.timedelta64]
AR_o: NDArray[np.timedelta64]

AR_LIKE_b: list[bool]
AR_LIKE_u: list[np.uint32]
AR_LIKE_i: list[int]
AR_LIKE_f: list[float]
AR_LIKE_c: list[complex]
AR_LIKE_td64: list[np.timedelta64]
AR_LIKE_dt64: list[np.datetime64]
AR_LIKE_o: list[np.object_]
AR_number: NDArray[np.number]

MAR_c8: MaskedArray[np.complex64]
MAR_c16: MaskedArray[np.complex128]
MAR_b: MaskedArray[np.bool]
MAR_f4: MaskedArray[np.float32]
MAR_f8: MaskedArray[np.float64]
MAR_i8: MaskedArray[np.int64]
MAR_u4: MaskedArray[np.uint32]
MAR_dt64: MaskedArray[np.datetime64]
MAR_td64: MaskedArray[np.timedelta64]
MAR_o: MaskedArray[np.object_]
MAR_s: MaskedArray[np.str_]
MAR_byte: MaskedArray[np.bytes_]
MAR_V: MaskedArray[np.void]
MAR_floating: MaskedArray[np.floating]
MAR_number: MaskedArray[np.number]

MAR_subclass: MaskedArraySubclassC
MAR_subclass_i: MaskedArraySubclassI
MAR_into_subclass: IntoMaskedArraySubClass[np.float32]

MAR_1d: np.ma.MaskedArray[tuple[int], np.dtype]
MAR_2d_f4: np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]]
MAR_2d_V: np.ma.MaskedArray[tuple[int, int], np.dtype[np.void]]

b: np.bool
f4: np.float32
f: float
i: int

assert_type(MAR_1d.shape, tuple[int])

assert_type(MAR_f4.dtype, np.dtype[np.float32])

assert_type(int(MAR_i8), int)
assert_type(float(MAR_f4), float)

assert_type(np.ma.min(MAR_b), np.bool)
assert_type(np.ma.min(MAR_f4), np.float32)
assert_type(np.ma.min(MAR_b, axis=0), Any)
assert_type(np.ma.min(MAR_f4, axis=0), Any)
assert_type(np.ma.min(MAR_b, keepdims=True), Any)
assert_type(np.ma.min(MAR_f4, keepdims=True), Any)
assert_type(np.ma.min(MAR_f4, out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.min(MAR_f4, 0, MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.min(MAR_f4, None, MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.min(), np.bool)
assert_type(MAR_f4.min(), np.float32)
assert_type(MAR_b.min(axis=0), Any)
assert_type(MAR_f4.min(axis=0), Any)
assert_type(MAR_b.min(keepdims=True), Any)
assert_type(MAR_f4.min(keepdims=True), Any)
assert_type(MAR_f4.min(out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.min(0, MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.min(None, MAR_subclass), MaskedArraySubclassC)

assert_type(np.ma.max(MAR_b), np.bool)
assert_type(np.ma.max(MAR_f4), np.float32)
assert_type(np.ma.max(MAR_b, axis=0), Any)
assert_type(np.ma.max(MAR_f4, axis=0), Any)
assert_type(np.ma.max(MAR_b, keepdims=True), Any)
assert_type(np.ma.max(MAR_f4, keepdims=True), Any)
assert_type(np.ma.max(MAR_f4, out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.max(MAR_f4, 0, MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.max(MAR_f4, None, MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.max(), np.bool)
assert_type(MAR_f4.max(), np.float32)
assert_type(MAR_b.max(axis=0), Any)
assert_type(MAR_f4.max(axis=0), Any)
assert_type(MAR_b.max(keepdims=True), Any)
assert_type(MAR_f4.max(keepdims=True), Any)
assert_type(MAR_f4.max(out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.max(0, MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.max(None, MAR_subclass), MaskedArraySubclassC)

assert_type(np.ma.ptp(MAR_b), np.bool)
assert_type(np.ma.ptp(MAR_f4), np.float32)
assert_type(np.ma.ptp(MAR_b, axis=0), Any)
assert_type(np.ma.ptp(MAR_f4, axis=0), Any)
assert_type(np.ma.ptp(MAR_b, keepdims=True), Any)
assert_type(np.ma.ptp(MAR_f4, keepdims=True), Any)
assert_type(np.ma.ptp(MAR_f4, out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.ptp(MAR_f4, 0, MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.ptp(MAR_f4, None, MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.ptp(), np.bool)
assert_type(MAR_f4.ptp(), np.float32)
assert_type(MAR_b.ptp(axis=0), Any)
assert_type(MAR_f4.ptp(axis=0), Any)
assert_type(MAR_b.ptp(keepdims=True), Any)
assert_type(MAR_f4.ptp(keepdims=True), Any)
assert_type(MAR_f4.ptp(out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.ptp(0, MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.ptp(None, MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.argmin(), np.intp)
assert_type(MAR_f4.argmin(), np.intp)
assert_type(MAR_f4.argmin(fill_value=6.28318, keepdims=False), np.intp)
assert_type(MAR_b.argmin(axis=0), MaskedArray[np.intp])
assert_type(MAR_f4.argmin(axis=0), MaskedArray[np.intp])
assert_type(MAR_b.argmin(keepdims=True), MaskedArray[np.intp])
assert_type(MAR_f4.argmin(out=MAR_subclass_i), MaskedArraySubclassI)
assert_type(MAR_f4.argmin(None, None, out=MAR_subclass_i), MaskedArraySubclassI)

assert_type(np.ma.argmin(MAR_b), np.intp)
assert_type(np.ma.argmin(MAR_f4), np.intp)
assert_type(np.ma.argmin(MAR_f4, fill_value=6.28318, keepdims=False), np.intp)
assert_type(np.ma.argmin(MAR_b, axis=0), Any)
assert_type(np.ma.argmin(MAR_f4, axis=0), Any)
assert_type(np.ma.argmin(MAR_b, keepdims=True), Any)
assert_type(np.ma.argmin(MAR_f4, out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.argmin(MAR_f4, None, None, out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.argmax(), np.intp)
assert_type(MAR_f4.argmax(), np.intp)
assert_type(MAR_f4.argmax(fill_value=6.28318, keepdims=False), np.intp)
assert_type(MAR_b.argmax(axis=0), MaskedArray[np.intp])
assert_type(MAR_f4.argmax(axis=0), MaskedArray[np.intp])
assert_type(MAR_b.argmax(keepdims=True), MaskedArray[np.intp])
assert_type(MAR_f4.argmax(out=MAR_subclass_i), MaskedArraySubclassI)
assert_type(MAR_f4.argmax(None, None, out=MAR_subclass_i), MaskedArraySubclassI)

assert_type(np.ma.argmax(MAR_b), np.intp)
assert_type(np.ma.argmax(MAR_f4), np.intp)
assert_type(np.ma.argmax(MAR_f4, fill_value=6.28318, keepdims=False), np.intp)
assert_type(np.ma.argmax(MAR_b, axis=0), Any)
assert_type(np.ma.argmax(MAR_f4, axis=0), Any)
assert_type(np.ma.argmax(MAR_b, keepdims=True), Any)
assert_type(np.ma.argmax(MAR_f4, out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.argmax(MAR_f4, None, None, out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.all(), np.bool)
assert_type(MAR_f4.all(), np.bool)
assert_type(MAR_f4.all(keepdims=False), np.bool)
assert_type(MAR_b.all(axis=0), np.bool | MaskedArray[np.bool])
assert_type(MAR_b.all(axis=0, keepdims=True), MaskedArray[np.bool])
assert_type(MAR_b.all(0, None, True), MaskedArray[np.bool])
assert_type(MAR_f4.all(axis=0), np.bool | MaskedArray[np.bool])
assert_type(MAR_b.all(keepdims=True), MaskedArray[np.bool])
assert_type(MAR_f4.all(out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.all(None, out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_b.any(), np.bool)
assert_type(MAR_f4.any(), np.bool)
assert_type(MAR_f4.any(keepdims=False), np.bool)
assert_type(MAR_b.any(axis=0), np.bool | MaskedArray[np.bool])
assert_type(MAR_b.any(axis=0, keepdims=True), MaskedArray[np.bool])
assert_type(MAR_b.any(0, None, True), MaskedArray[np.bool])
assert_type(MAR_f4.any(axis=0), np.bool | MaskedArray[np.bool])
assert_type(MAR_b.any(keepdims=True), MaskedArray[np.bool])
assert_type(MAR_f4.any(out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.any(None, out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f4.sort(), None)
assert_type(MAR_f4.sort(axis=0, kind="quicksort", order="K", endwith=False, fill_value=42., stable=False), None)

assert_type(np.ma.sort(MAR_f4), MaskedArray[np.float32])
assert_type(np.ma.sort(MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.sort([[0, 1], [2, 3]]), NDArray[Any])
assert_type(np.ma.sort(AR_f4), NDArray[np.float32])

assert_type(MAR_f8.take(0), np.float64)
assert_type(MAR_1d.take(0), Any)
assert_type(MAR_f8.take([0]), MaskedArray[np.float64])
assert_type(MAR_f8.take(0, out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f8.take([0], out=MAR_subclass), MaskedArraySubclassC)

assert_type(np.ma.take(f, 0), Any)
assert_type(np.ma.take(f4, 0), np.float32)
assert_type(np.ma.take(MAR_f8, 0), np.float64)
assert_type(np.ma.take(AR_f4, 0), np.float32)
assert_type(np.ma.take(MAR_1d, 0), Any)
assert_type(np.ma.take(MAR_f8, [0]), MaskedArray[np.float64])
assert_type(np.ma.take(AR_f4, [0]), MaskedArray[np.float32])
assert_type(np.ma.take(MAR_f8, 0, out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.take(MAR_f8, [0], out=MAR_subclass), MaskedArraySubclassC)
assert_type(np.ma.take([1], [0]), MaskedArray[Any])
assert_type(np.ma.take(np.eye(2), 1, axis=0), MaskedArray[np.float64])

assert_type(MAR_f4.partition(1), None)
assert_type(MAR_V.partition(1, axis=0, kind="introselect", order="K"), None)

assert_type(MAR_f4.argpartition(1), MaskedArray[np.intp])
assert_type(
    MAR_1d.argpartition(1, axis=0, kind="introselect", order="K"),
    np.ma.MaskedArray[tuple[int], np.dtype[np.intp]],
)

assert_type(np.ma.ndim(f4), int)
assert_type(np.ma.ndim(MAR_b), int)
assert_type(np.ma.ndim(AR_f4), int)

assert_type(np.ma.size(b), int)
assert_type(np.ma.size(MAR_f4, axis=0), int)
assert_type(np.ma.size(AR_f4), int)

assert_type(np.ma.is_masked(MAR_f4), bool)

assert_type(MAR_f4.ids(), tuple[int, int])

assert_type(MAR_f4.iscontiguous(), bool)

assert_type(MAR_f4 >= 3, MaskedArray[np.bool])
assert_type(MAR_i8 >= AR_td64, MaskedArray[np.bool])
assert_type(MAR_b >= AR_td64, MaskedArray[np.bool])
assert_type(MAR_td64 >= AR_td64, MaskedArray[np.bool])
assert_type(MAR_dt64 >= AR_dt64, MaskedArray[np.bool])
assert_type(MAR_o >= AR_o, MaskedArray[np.bool])
assert_type(MAR_1d >= 0, MaskedArray[np.bool])
assert_type(MAR_s >= MAR_s, MaskedArray[np.bool])
assert_type(MAR_byte >= MAR_byte, MaskedArray[np.bool])

assert_type(MAR_f4 > 3, MaskedArray[np.bool])
assert_type(MAR_i8 > AR_td64, MaskedArray[np.bool])
assert_type(MAR_b > AR_td64, MaskedArray[np.bool])
assert_type(MAR_td64 > AR_td64, MaskedArray[np.bool])
assert_type(MAR_dt64 > AR_dt64, MaskedArray[np.bool])
assert_type(MAR_o > AR_o, MaskedArray[np.bool])
assert_type(MAR_1d > 0, MaskedArray[np.bool])
assert_type(MAR_s > MAR_s, MaskedArray[np.bool])
assert_type(MAR_byte > MAR_byte, MaskedArray[np.bool])

assert_type(MAR_f4 <= 3, MaskedArray[np.bool])
assert_type(MAR_i8 <= AR_td64, MaskedArray[np.bool])
assert_type(MAR_b <= AR_td64, MaskedArray[np.bool])
assert_type(MAR_td64 <= AR_td64, MaskedArray[np.bool])
assert_type(MAR_dt64 <= AR_dt64, MaskedArray[np.bool])
assert_type(MAR_o <= AR_o, MaskedArray[np.bool])
assert_type(MAR_1d <= 0, MaskedArray[np.bool])
assert_type(MAR_s <= MAR_s, MaskedArray[np.bool])
assert_type(MAR_byte <= MAR_byte, MaskedArray[np.bool])

assert_type(MAR_f4 < 3, MaskedArray[np.bool])
assert_type(MAR_i8 < AR_td64, MaskedArray[np.bool])
assert_type(MAR_b < AR_td64, MaskedArray[np.bool])
assert_type(MAR_td64 < AR_td64, MaskedArray[np.bool])
assert_type(MAR_dt64 < AR_dt64, MaskedArray[np.bool])
assert_type(MAR_o < AR_o, MaskedArray[np.bool])
assert_type(MAR_1d < 0, MaskedArray[np.bool])
assert_type(MAR_s < MAR_s, MaskedArray[np.bool])
assert_type(MAR_byte < MAR_byte, MaskedArray[np.bool])

assert_type(MAR_f4 <= 3, MaskedArray[np.bool])
assert_type(MAR_i8 <= AR_td64, MaskedArray[np.bool])
assert_type(MAR_b <= AR_td64, MaskedArray[np.bool])
assert_type(MAR_td64 <= AR_td64, MaskedArray[np.bool])
assert_type(MAR_dt64 <= AR_dt64, MaskedArray[np.bool])
assert_type(MAR_o <= AR_o, MaskedArray[np.bool])
assert_type(MAR_1d <= 0, MaskedArray[np.bool])
assert_type(MAR_s <= MAR_s, MaskedArray[np.bool])
assert_type(MAR_byte <= MAR_byte, MaskedArray[np.bool])

assert_type(MAR_byte.count(), int)
assert_type(MAR_f4.count(axis=None), int)
assert_type(MAR_f4.count(axis=0), NDArray[np.int_])
assert_type(MAR_b.count(axis=(0, 1)), NDArray[np.int_])
assert_type(MAR_o.count(keepdims=True), NDArray[np.int_])
assert_type(MAR_o.count(axis=None, keepdims=True), NDArray[np.int_])
assert_type(MAR_o.count(None, True), NDArray[np.int_])

assert_type(np.ma.count(MAR_byte), int)
assert_type(np.ma.count(MAR_byte, axis=None), int)
assert_type(np.ma.count(MAR_f4, axis=0), NDArray[np.int_])
assert_type(np.ma.count(MAR_b, axis=(0, 1)), NDArray[np.int_])
assert_type(np.ma.count(MAR_o, keepdims=True), NDArray[np.int_])
assert_type(np.ma.count(MAR_o, axis=None, keepdims=True), NDArray[np.int_])
assert_type(np.ma.count(MAR_o, None, True), NDArray[np.int_])

assert_type(MAR_f4.compressed(), np.ndarray[tuple[int], np.dtype[np.float32]])

assert_type(MAR_f4.compress([True, False]), np.ma.MaskedArray[tuple[int], np.dtype[np.float32]])
assert_type(MAR_f4.compress([True, False], axis=0), MaskedArray[np.float32])
assert_type(MAR_f4.compress([True, False], axis=0, out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f4.compress([True, False], 0, MAR_subclass), MaskedArraySubclassC)

assert_type(np.ma.compressed(MAR_i8), np.ndarray[tuple[int], np.dtype[np.int64]])
assert_type(np.ma.compressed([[1, 2, 3]]), np.ndarray[tuple[int], np.dtype])

assert_type(MAR_f4.put([0, 4, 8], [10, 20, 30]), None)
assert_type(MAR_f4.put(4, 999), None)
assert_type(MAR_f4.put(4, 999, mode="clip"), None)

assert_type(MAR_c8.__array_wrap__(AR_b), MaskedArray[np.bool])

assert_type(np.ma.put(MAR_f4, [0, 4, 8], [10, 20, 30]), None)
assert_type(np.ma.put(MAR_f4, 4, 999), None)
assert_type(np.ma.put(MAR_f4, 4, 999, mode="clip"), None)

assert_type(np.ma.putmask(MAR_f4, [True, False], [0, 1]), None)
assert_type(np.ma.putmask(MAR_f4, np.False_, [0, 1]), None)

assert_type(MAR_f4.filled(float("nan")), NDArray[np.float32])
assert_type(MAR_i8.filled(), NDArray[np.int64])
assert_type(MAR_1d.filled(), np.ndarray[tuple[int], np.dtype])

assert_type(np.ma.filled(MAR_f4, float("nan")), NDArray[np.float32])
assert_type(np.ma.filled([[1, 2, 3]]), NDArray[Any])
# PyRight detects this one correctly, but mypy doesn't.
# https://github.com/numpy/numpy/pull/28742#discussion_r2048968375
assert_type(np.ma.filled(MAR_1d), np.ndarray[tuple[int], np.dtype])  # type: ignore[assert-type]

assert_type(MAR_b.repeat(3), np.ma.MaskedArray[tuple[int], np.dtype[np.bool]])
assert_type(MAR_2d_f4.repeat(MAR_i8), np.ma.MaskedArray[tuple[int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.repeat(MAR_i8, axis=None), np.ma.MaskedArray[tuple[int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.repeat(MAR_i8, axis=0), MaskedArray[np.float32])

assert_type(np.ma.allequal(AR_f4, MAR_f4), bool)
assert_type(np.ma.allequal(AR_f4, MAR_f4, fill_value=False), bool)

assert_type(np.ma.allclose(AR_f4, MAR_f4), bool)
assert_type(np.ma.allclose(AR_f4, MAR_f4, masked_equal=False), bool)
assert_type(np.ma.allclose(AR_f4, MAR_f4, rtol=.4, atol=.3), bool)

assert_type(MAR_2d_f4.ravel(), np.ma.MaskedArray[tuple[int], np.dtype[np.float32]])
assert_type(MAR_1d.ravel(order="A"), np.ma.MaskedArray[tuple[int], np.dtype[Any]])

assert_type(np.ma.getmask(MAR_f4), NDArray[np.bool] | _NoMaskType)
# PyRight detects this one correctly, but mypy doesn't:
# `Revealed type is "Union[numpy.ndarray[Any, Any], numpy.bool[Any]]"`
assert_type(np.ma.getmask(MAR_1d), np.ndarray[tuple[int], np.dtype[np.bool]] | np.bool)  # type: ignore[assert-type]
assert_type(np.ma.getmask(MAR_2d_f4), np.ndarray[tuple[int, int], np.dtype[np.bool]] | _NoMaskType)
assert_type(np.ma.getmask([1, 2]), NDArray[np.bool] | _NoMaskType)
assert_type(np.ma.getmask(np.int64(1)), _NoMaskType)

assert_type(np.ma.is_mask(MAR_1d), bool)
assert_type(np.ma.is_mask(AR_b), bool)

def func(x: object) -> None:
    if np.ma.is_mask(x):
        assert_type(x, NDArray[np.bool])
    else:
        assert_type(x, object)

assert_type(MAR_2d_f4.mT, np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])

assert_type(MAR_c16.real, MaskedArray[np.float64])
assert_type(MAR_c16.imag, MaskedArray[np.float64])

assert_type(MAR_2d_f4.baseclass, type[NDArray[Any]])

assert_type(MAR_b.swapaxes(0, 1), MaskedArray[np.bool])
assert_type(MAR_2d_f4.swapaxes(1, 0), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])

assert_type(MAR_2d_f4[AR_i8], MaskedArray[np.float32])
assert_type(MAR_2d_f4[[1, 2, 3]], MaskedArray[np.float32])
assert_type(MAR_2d_f4[1:], MaskedArray[np.float32])
assert_type(MAR_2d_f4[:], MaskedArray[np.float32])
assert_type(MAR_2d_f4[0, 0], Any)
assert_type(MAR_2d_f4[:, np.newaxis], MaskedArray[np.float32])
assert_type(MAR_2d_f4[..., -1], MaskedArray[np.float32])
assert_type(MAR_2d_V["field_0"], np.ma.MaskedArray[tuple[int, int], np.dtype])
assert_type(MAR_2d_V[["field_0", "field_1"]], np.ma.MaskedArray[tuple[int, int], np.dtype[np.void]])

assert_type(np.ma.nomask, np.bool[Literal[False]])
assert_type(np.ma.MaskType, type[np.bool])

assert_type(MAR_1d.__setmask__([True, False]), None)
assert_type(MAR_1d.__setmask__(np.False_), None)

assert_type(MAR_2d_f4.harden_mask(), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_i8.harden_mask(), MaskedArray[np.int64])
assert_type(MAR_2d_f4.soften_mask(), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_i8.soften_mask(), MaskedArray[np.int64])
assert_type(MAR_f4.unshare_mask(), MaskedArray[np.float32])
assert_type(MAR_b.shrink_mask(), MaskedArray[np.bool_])

assert_type(MAR_i8.hardmask, bool)
assert_type(MAR_i8.sharedmask, bool)

assert_type(MAR_i8.recordmask, np.ma.MaskType | NDArray[np.ma.MaskType])
assert_type(MAR_2d_f4.recordmask, np.ma.MaskType | np.ndarray[tuple[int, int], np.dtype[np.ma.MaskType]])

assert_type(MAR_2d_f4.anom(), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.anom(axis=0, dtype=np.float16), np.ma.MaskedArray[tuple[int, int], np.dtype])
assert_type(MAR_2d_f4.anom(0, np.float16), np.ma.MaskedArray[tuple[int, int], np.dtype])
assert_type(MAR_2d_f4.anom(0, "float16"), np.ma.MaskedArray[tuple[int, int], np.dtype])

assert_type(MAR_i8.fill_value, np.int64)

assert_type(MAR_b.transpose(), MaskedArray[np.bool])
assert_type(MAR_2d_f4.transpose(), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.transpose(1, 0), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.transpose((1, 0)), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_b.T, MaskedArray[np.bool])
assert_type(MAR_2d_f4.T, np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])

assert_type(MAR_2d_f4.dot(1), MaskedArray[Any])
assert_type(MAR_2d_f4.dot([1]), MaskedArray[Any])
assert_type(MAR_2d_f4.dot(1, out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_2d_f4.nonzero(), tuple[_Array1D[np.intp], _Array1D[np.intp]])

assert_type(MAR_f8.trace(), Any)
assert_type(MAR_f8.trace(out=MAR_subclass), MaskedArraySubclassC)
assert_type(MAR_f8.trace(out=MAR_subclass, dtype=None), MaskedArraySubclassC)

assert_type(MAR_f8.round(), MaskedArray[np.float64])
assert_type(MAR_f8.round(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_i8.reshape(None), MaskedArray[np.int64])
assert_type(MAR_f8.reshape(-1), np.ma.MaskedArray[tuple[int], np.dtype[np.float64]])
assert_type(MAR_c8.reshape(2, 3, 4, 5), np.ma.MaskedArray[tuple[int, int, int, int], np.dtype[np.complex64]])
assert_type(MAR_td64.reshape(()), np.ma.MaskedArray[tuple[()], np.dtype[np.timedelta64]])
assert_type(MAR_s.reshape([]), np.ma.MaskedArray[tuple[()], np.dtype[np.str_]])
assert_type(MAR_V.reshape((480, 720, 4)), np.ma.MaskedArray[tuple[int, int, int], np.dtype[np.void]])

assert_type(MAR_f8.cumprod(), np.ma.MaskedArray[tuple[int], np.dtype[np.float64]])
assert_type(MAR_f8.cumprod(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f8.cumsum(), np.ma.MaskedArray[tuple[int], np.dtype[np.float64]])
assert_type(MAR_f8.cumsum(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f8.view(), MaskedArray[np.float64])
assert_type(MAR_f8.view(dtype=np.float32), MaskedArray[np.float32])
assert_type(MAR_f8.view(dtype=np.dtype(np.float32)), MaskedArray[np.float32])
assert_type(MAR_f8.view(dtype=np.float32, fill_value=0), MaskedArray[np.float32])
assert_type(MAR_f8.view(type=np.ndarray), np.ndarray[Any, Any])
assert_type(MAR_f8.view(None, np.ndarray), np.ndarray[Any, Any])
assert_type(MAR_f8.view(dtype=np.ndarray), np.ndarray[Any, Any])
assert_type(MAR_f8.view(dtype="float32"), MaskedArray[Any])
assert_type(MAR_f8.view(dtype="float32", type=np.ndarray), np.ndarray[Any, Any])
assert_type(MAR_2d_f4.view(dtype=np.float16), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float16]])
assert_type(MAR_2d_f4.view(dtype=np.dtype(np.float16)), np.ma.MaskedArray[tuple[int, int], np.dtype[np.float16]])

assert_type(MAR_f8.__deepcopy__(), MaskedArray[np.float64])

assert_type(MAR_f8.argsort(), MaskedArray[np.intp])
assert_type(MAR_f8.argsort(axis=0, kind="heap", order=("x", "y")), MaskedArray[np.intp])
assert_type(MAR_f8.argsort(endwith=True, fill_value=1.5, stable=False), MaskedArray[np.intp])

assert_type(MAR_2d_f4.flat, np.ma.core.MaskedIterator[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.flat.ma, np.ma.MaskedArray[tuple[int, int], np.dtype[np.float32]])
assert_type(MAR_2d_f4.flat[AR_i8], MaskedArray[np.float32])
assert_type(MAR_2d_f4.flat[[1, 2, 3]], MaskedArray[np.float32])
assert_type(MAR_2d_f4.flat[1:], MaskedArray[np.float32])
assert_type(MAR_2d_f4.flat[:], MaskedArray[np.float32])
assert_type(MAR_2d_f4.flat[0, 0], Any)
assert_type(MAR_2d_f4.flat[:, np.newaxis], MaskedArray[np.float32])
assert_type(MAR_2d_f4.flat[..., -1], MaskedArray[np.float32])

def invalid_resize() -> None:
    assert_type(MAR_f8.resize((1, 1)), NoReturn)  # type: ignore[arg-type]

assert_type(np.ma.MaskedArray(AR_f4), MaskedArray[np.float32])
assert_type(np.ma.MaskedArray(np.array([1, 2, 3]), [True, True, False], np.float16), MaskedArray[np.float16])
assert_type(np.ma.MaskedArray(np.array([1, 2, 3]), dtype=np.float16), MaskedArray[np.float16])
assert_type(np.ma.MaskedArray(np.array([1, 2, 3]), copy=True), MaskedArray[Any])
# TODO: This one could be made more precise, the return type could be `MaskedArraySubclassC`
assert_type(np.ma.MaskedArray(MAR_subclass), MaskedArray[np.complex128])
# TODO: This one could be made more precise, the return type could be `MaskedArraySubclass[np.float32]`
assert_type(np.ma.MaskedArray(MAR_into_subclass), MaskedArray[np.float32])

# Masked Array addition

assert_type(MAR_b + AR_LIKE_u, MaskedArray[np.uint32])
assert_type(MAR_b + AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_b + AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_b + AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_b + AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_b + AR_LIKE_o, Any)

assert_type(AR_LIKE_u + MAR_b, MaskedArray[np.uint32])
assert_type(AR_LIKE_i + MAR_b, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f + MAR_b, MaskedArray[np.floating])
assert_type(AR_LIKE_c + MAR_b, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_td64 + MAR_b, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 + MAR_b, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o + MAR_b, Any)

assert_type(MAR_u4 + AR_LIKE_b, MaskedArray[np.uint32])
assert_type(MAR_u4 + AR_LIKE_u, MaskedArray[np.unsignedinteger])
assert_type(MAR_u4 + AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_u4 + AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_u4 + AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_u4 + AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_u4 + AR_LIKE_o, Any)

assert_type(AR_LIKE_b + MAR_u4, MaskedArray[np.uint32])
assert_type(AR_LIKE_u + MAR_u4, MaskedArray[np.unsignedinteger])
assert_type(AR_LIKE_i + MAR_u4, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f + MAR_u4, MaskedArray[np.floating])
assert_type(AR_LIKE_c + MAR_u4, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_td64 + MAR_u4, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 + MAR_u4, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o + MAR_u4, Any)

assert_type(MAR_i8 + AR_LIKE_b, MaskedArray[np.int64])
assert_type(MAR_i8 + AR_LIKE_u, MaskedArray[np.signedinteger])
assert_type(MAR_i8 + AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_i8 + AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_i8 + AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_i8 + AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_i8 + AR_LIKE_o, Any)

assert_type(AR_LIKE_b + MAR_i8, MaskedArray[np.int64])
assert_type(AR_LIKE_u + MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_i + MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f + MAR_i8, MaskedArray[np.floating])
assert_type(AR_LIKE_c + MAR_i8, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_td64 + MAR_i8, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 + MAR_i8, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o + MAR_i8, Any)

assert_type(MAR_f8 + AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_f8 + AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_f8 + AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_f8 + AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_f8 + AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_f8 + AR_LIKE_o, Any)

assert_type(AR_LIKE_b + MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_u + MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_i + MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_f + MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_c + MAR_f8, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_o + MAR_f8, Any)

assert_type(MAR_c16 + AR_LIKE_b, MaskedArray[np.complex128])
assert_type(MAR_c16 + AR_LIKE_u, MaskedArray[np.complex128])
assert_type(MAR_c16 + AR_LIKE_i, MaskedArray[np.complex128])
assert_type(MAR_c16 + AR_LIKE_f, MaskedArray[np.complex128])
assert_type(MAR_c16 + AR_LIKE_c, MaskedArray[np.complex128])
assert_type(MAR_c16 + AR_LIKE_o, Any)

assert_type(AR_LIKE_b + MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_u + MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_i + MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_f + MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_c + MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_o + MAR_c16, Any)

assert_type(MAR_td64 + AR_LIKE_b, MaskedArray[np.timedelta64])
assert_type(MAR_td64 + AR_LIKE_u, MaskedArray[np.timedelta64])
assert_type(MAR_td64 + AR_LIKE_i, MaskedArray[np.timedelta64])
assert_type(MAR_td64 + AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_td64 + AR_LIKE_o, Any)

assert_type(AR_LIKE_b + MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_u + MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_i + MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_td64 + MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 + MAR_td64, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o + MAR_td64, Any)

assert_type(MAR_dt64 + AR_LIKE_b, MaskedArray[np.datetime64])
assert_type(MAR_dt64 + AR_LIKE_u, MaskedArray[np.datetime64])
assert_type(MAR_dt64 + AR_LIKE_i, MaskedArray[np.datetime64])
assert_type(MAR_dt64 + AR_LIKE_td64, MaskedArray[np.datetime64])
assert_type(MAR_dt64 + AR_LIKE_o, Any)

assert_type(AR_LIKE_o + MAR_dt64, Any)

assert_type(MAR_o + AR_LIKE_b, Any)
assert_type(MAR_o + AR_LIKE_u, Any)
assert_type(MAR_o + AR_LIKE_i, Any)
assert_type(MAR_o + AR_LIKE_f, Any)
assert_type(MAR_o + AR_LIKE_c, Any)
assert_type(MAR_o + AR_LIKE_td64, Any)
assert_type(MAR_o + AR_LIKE_dt64, Any)
assert_type(MAR_o + AR_LIKE_o, Any)

assert_type(AR_LIKE_b + MAR_o, Any)
assert_type(AR_LIKE_u + MAR_o, Any)
assert_type(AR_LIKE_i + MAR_o, Any)
assert_type(AR_LIKE_f + MAR_o, Any)
assert_type(AR_LIKE_c + MAR_o, Any)
assert_type(AR_LIKE_td64 + MAR_o, Any)
assert_type(AR_LIKE_dt64 + MAR_o, Any)
assert_type(AR_LIKE_o + MAR_o, Any)

# Masked Array subtraction
# Keep in sync with numpy/typing/tests/data/reveal/arithmetic.pyi

assert_type(MAR_number - AR_number, MaskedArray[np.number])

assert_type(MAR_b - AR_LIKE_u, MaskedArray[np.uint32])
assert_type(MAR_b - AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_b - AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_b - AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_b - AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_b - AR_LIKE_o, Any)

assert_type(AR_LIKE_u - MAR_b, MaskedArray[np.uint32])
assert_type(AR_LIKE_i - MAR_b, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f - MAR_b, MaskedArray[np.floating])
assert_type(AR_LIKE_c - MAR_b, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_td64 - MAR_b, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 - MAR_b, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o - MAR_b, Any)

assert_type(MAR_u4 - AR_LIKE_b, MaskedArray[np.uint32])
assert_type(MAR_u4 - AR_LIKE_u, MaskedArray[np.unsignedinteger])
assert_type(MAR_u4 - AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_u4 - AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_u4 - AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_u4 - AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_u4 - AR_LIKE_o, Any)

assert_type(AR_LIKE_b - MAR_u4, MaskedArray[np.uint32])
assert_type(AR_LIKE_u - MAR_u4, MaskedArray[np.unsignedinteger])
assert_type(AR_LIKE_i - MAR_u4, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f - MAR_u4, MaskedArray[np.floating])
assert_type(AR_LIKE_c - MAR_u4, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_td64 - MAR_u4, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 - MAR_u4, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o - MAR_u4, Any)

assert_type(MAR_i8 - AR_LIKE_b, MaskedArray[np.int64])
assert_type(MAR_i8 - AR_LIKE_u, MaskedArray[np.signedinteger])
assert_type(MAR_i8 - AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_i8 - AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_i8 - AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_i8 - AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_i8 - AR_LIKE_o, Any)

assert_type(AR_LIKE_b - MAR_i8, MaskedArray[np.int64])
assert_type(AR_LIKE_u - MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_i - MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f - MAR_i8, MaskedArray[np.floating])
assert_type(AR_LIKE_c - MAR_i8, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_td64 - MAR_i8, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 - MAR_i8, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o - MAR_i8, Any)

assert_type(MAR_f8 - AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_f8 - AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_f8 - AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_f8 - AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_f8 - AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_f8 - AR_LIKE_o, Any)

assert_type(AR_LIKE_b - MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_u - MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_i - MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_f - MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_c - MAR_f8, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_o - MAR_f8, Any)

assert_type(MAR_c16 - AR_LIKE_b, MaskedArray[np.complex128])
assert_type(MAR_c16 - AR_LIKE_u, MaskedArray[np.complex128])
assert_type(MAR_c16 - AR_LIKE_i, MaskedArray[np.complex128])
assert_type(MAR_c16 - AR_LIKE_f, MaskedArray[np.complex128])
assert_type(MAR_c16 - AR_LIKE_c, MaskedArray[np.complex128])
assert_type(MAR_c16 - AR_LIKE_o, Any)

assert_type(AR_LIKE_b - MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_u - MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_i - MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_f - MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_c - MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_o - MAR_c16, Any)

assert_type(MAR_td64 - AR_LIKE_b, MaskedArray[np.timedelta64])
assert_type(MAR_td64 - AR_LIKE_u, MaskedArray[np.timedelta64])
assert_type(MAR_td64 - AR_LIKE_i, MaskedArray[np.timedelta64])
assert_type(MAR_td64 - AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_td64 - AR_LIKE_o, Any)

assert_type(AR_LIKE_b - MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_u - MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_i - MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_td64 - MAR_td64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_dt64 - MAR_td64, MaskedArray[np.datetime64])
assert_type(AR_LIKE_o - MAR_td64, Any)

assert_type(MAR_dt64 - AR_LIKE_b, MaskedArray[np.datetime64])
assert_type(MAR_dt64 - AR_LIKE_u, MaskedArray[np.datetime64])
assert_type(MAR_dt64 - AR_LIKE_i, MaskedArray[np.datetime64])
assert_type(MAR_dt64 - AR_LIKE_td64, MaskedArray[np.datetime64])
assert_type(MAR_dt64 - AR_LIKE_dt64, MaskedArray[np.timedelta64])
assert_type(MAR_dt64 - AR_LIKE_o, Any)

assert_type(AR_LIKE_dt64 - MAR_dt64, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o - MAR_dt64, Any)

assert_type(MAR_o - AR_LIKE_b, Any)
assert_type(MAR_o - AR_LIKE_u, Any)
assert_type(MAR_o - AR_LIKE_i, Any)
assert_type(MAR_o - AR_LIKE_f, Any)
assert_type(MAR_o - AR_LIKE_c, Any)
assert_type(MAR_o - AR_LIKE_td64, Any)
assert_type(MAR_o - AR_LIKE_dt64, Any)
assert_type(MAR_o - AR_LIKE_o, Any)

assert_type(AR_LIKE_b - MAR_o, Any)
assert_type(AR_LIKE_u - MAR_o, Any)
assert_type(AR_LIKE_i - MAR_o, Any)
assert_type(AR_LIKE_f - MAR_o, Any)
assert_type(AR_LIKE_c - MAR_o, Any)
assert_type(AR_LIKE_td64 - MAR_o, Any)
assert_type(AR_LIKE_dt64 - MAR_o, Any)
assert_type(AR_LIKE_o - MAR_o, Any)

# Masked Array multiplication

assert_type(MAR_b * AR_LIKE_u, MaskedArray[np.uint32])
assert_type(MAR_b * AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_b * AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_b * AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_b * AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_b * AR_LIKE_o, Any)

# Ignore due to https://github.com/python/mypy/issues/19341
assert_type(AR_LIKE_u * MAR_b, MaskedArray[np.uint32])  # type: ignore[assert-type]
assert_type(AR_LIKE_i * MAR_b, MaskedArray[np.signedinteger])  # type: ignore[assert-type]
assert_type(AR_LIKE_f * MAR_b, MaskedArray[np.floating])  # type: ignore[assert-type]
assert_type(AR_LIKE_c * MAR_b, MaskedArray[np.complexfloating])  # type: ignore[assert-type]
assert_type(AR_LIKE_td64 * MAR_b, MaskedArray[np.timedelta64])  # type: ignore[assert-type]
assert_type(AR_LIKE_o * MAR_b, Any)  # type: ignore[assert-type]

assert_type(MAR_u4 * AR_LIKE_b, MaskedArray[np.uint32])
assert_type(MAR_u4 * AR_LIKE_u, MaskedArray[np.unsignedinteger])
assert_type(MAR_u4 * AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_u4 * AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_u4 * AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_u4 * AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_u4 * AR_LIKE_o, Any)

assert_type(MAR_i8 * AR_LIKE_b, MaskedArray[np.int64])
assert_type(MAR_i8 * AR_LIKE_u, MaskedArray[np.signedinteger])
assert_type(MAR_i8 * AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_i8 * AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_i8 * AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_i8 * AR_LIKE_td64, MaskedArray[np.timedelta64])
assert_type(MAR_i8 * AR_LIKE_o, Any)

assert_type(MAR_f8 * AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_f8 * AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_f8 * AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_f8 * AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_f8 * AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_f8 * AR_LIKE_o, Any)

# Ignore due to https://github.com/python/mypy/issues/19341
assert_type(AR_LIKE_b * MAR_f8, MaskedArray[np.float64])  # type: ignore[assert-type]
assert_type(AR_LIKE_u * MAR_f8, MaskedArray[np.float64])  # type: ignore[assert-type]
assert_type(AR_LIKE_i * MAR_f8, MaskedArray[np.float64])  # type: ignore[assert-type]
assert_type(AR_LIKE_f * MAR_f8, MaskedArray[np.float64])  # type: ignore[assert-type]
assert_type(AR_LIKE_c * MAR_f8, MaskedArray[np.complexfloating])  # type: ignore[assert-type]
assert_type(AR_LIKE_o * MAR_f8, Any)  # type: ignore[assert-type]

assert_type(MAR_c16 * AR_LIKE_b, MaskedArray[np.complex128])
assert_type(MAR_c16 * AR_LIKE_u, MaskedArray[np.complex128])
assert_type(MAR_c16 * AR_LIKE_i, MaskedArray[np.complex128])
assert_type(MAR_c16 * AR_LIKE_f, MaskedArray[np.complex128])
assert_type(MAR_c16 * AR_LIKE_c, MaskedArray[np.complex128])
assert_type(MAR_c16 * AR_LIKE_o, Any)

# Ignore due to https://github.com/python/mypy/issues/19341
assert_type(AR_LIKE_b * MAR_c16, MaskedArray[np.complex128])  # type: ignore[assert-type]
assert_type(AR_LIKE_u * MAR_c16, MaskedArray[np.complex128])  # type: ignore[assert-type]
assert_type(AR_LIKE_i * MAR_c16, MaskedArray[np.complex128])  # type: ignore[assert-type]
assert_type(AR_LIKE_f * MAR_c16, MaskedArray[np.complex128])  # type: ignore[assert-type]
assert_type(AR_LIKE_c * MAR_c16, MaskedArray[np.complex128])  # type: ignore[assert-type]
assert_type(AR_LIKE_o * MAR_c16, Any)  # type: ignore[assert-type]

assert_type(MAR_td64 * AR_LIKE_b, MaskedArray[np.timedelta64])
assert_type(MAR_td64 * AR_LIKE_u, MaskedArray[np.timedelta64])
assert_type(MAR_td64 * AR_LIKE_i, MaskedArray[np.timedelta64])
assert_type(MAR_td64 * AR_LIKE_o, Any)

# Ignore due to https://github.com/python/mypy/issues/19341
assert_type(AR_LIKE_b * MAR_td64, MaskedArray[np.timedelta64])  # type: ignore[assert-type]
assert_type(AR_LIKE_u * MAR_td64, MaskedArray[np.timedelta64])  # type: ignore[assert-type]
assert_type(AR_LIKE_i * MAR_td64, MaskedArray[np.timedelta64])  # type: ignore[assert-type]
assert_type(AR_LIKE_td64 * MAR_td64, MaskedArray[np.timedelta64])  # type: ignore[assert-type]
assert_type(AR_LIKE_dt64 * MAR_td64, MaskedArray[np.datetime64])  # type: ignore[assert-type]
assert_type(AR_LIKE_o * MAR_td64, Any)  # type: ignore[assert-type]

assert_type(AR_LIKE_o * MAR_dt64, Any)  # type: ignore[assert-type]

assert_type(MAR_o * AR_LIKE_b, Any)
assert_type(MAR_o * AR_LIKE_u, Any)
assert_type(MAR_o * AR_LIKE_i, Any)
assert_type(MAR_o * AR_LIKE_f, Any)
assert_type(MAR_o * AR_LIKE_c, Any)
assert_type(MAR_o * AR_LIKE_td64, Any)
assert_type(MAR_o * AR_LIKE_dt64, Any)
assert_type(MAR_o * AR_LIKE_o, Any)

# Ignore due to https://github.com/python/mypy/issues/19341
assert_type(AR_LIKE_b * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_u * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_i * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_f * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_c * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_td64 * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_dt64 * MAR_o, Any)  # type: ignore[assert-type]
assert_type(AR_LIKE_o * MAR_o, Any)  # type: ignore[assert-type]

assert_type(MAR_f8.sum(), Any)
assert_type(MAR_f8.sum(axis=0), Any)
assert_type(MAR_f8.sum(keepdims=True), Any)
assert_type(MAR_f8.sum(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f8.std(), Any)
assert_type(MAR_f8.std(axis=0), Any)
assert_type(MAR_f8.std(keepdims=True, mean=0.), Any)
assert_type(MAR_f8.std(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f8.var(), Any)
assert_type(MAR_f8.var(axis=0), Any)
assert_type(MAR_f8.var(keepdims=True, mean=0.), Any)
assert_type(MAR_f8.var(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f8.mean(), Any)
assert_type(MAR_f8.mean(axis=0), Any)
assert_type(MAR_f8.mean(keepdims=True), Any)
assert_type(MAR_f8.mean(out=MAR_subclass), MaskedArraySubclassC)

assert_type(MAR_f8.prod(), Any)
assert_type(MAR_f8.prod(axis=0), Any)
assert_type(MAR_f8.prod(keepdims=True), Any)
assert_type(MAR_f8.prod(out=MAR_subclass), MaskedArraySubclassC)

# MaskedArray "true" division

assert_type(MAR_f8 / b, MaskedArray[np.float64])
assert_type(MAR_f8 / i, MaskedArray[np.float64])
assert_type(MAR_f8 / f, MaskedArray[np.float64])

assert_type(b / MAR_f8, MaskedArray[np.float64])
assert_type(i / MAR_f8, MaskedArray[np.float64])
assert_type(f / MAR_f8, MaskedArray[np.float64])

assert_type(MAR_b / AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_b / AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_b / AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_b / AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_b / AR_LIKE_o, Any)

assert_type(AR_LIKE_b / MAR_b, MaskedArray[np.float64])
assert_type(AR_LIKE_u / MAR_b, MaskedArray[np.float64])
assert_type(AR_LIKE_i / MAR_b, MaskedArray[np.float64])
assert_type(AR_LIKE_f / MAR_b, MaskedArray[np.float64])
assert_type(AR_LIKE_o / MAR_b, Any)

assert_type(MAR_u4 / AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_u4 / AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_u4 / AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_u4 / AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_u4 / AR_LIKE_o, Any)

assert_type(AR_LIKE_b / MAR_u4, MaskedArray[np.float64])
assert_type(AR_LIKE_u / MAR_u4, MaskedArray[np.float64])
assert_type(AR_LIKE_i / MAR_u4, MaskedArray[np.float64])
assert_type(AR_LIKE_f / MAR_u4, MaskedArray[np.float64])
assert_type(AR_LIKE_td64 / MAR_u4, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o / MAR_u4, Any)

assert_type(MAR_i8 / AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_i8 / AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_i8 / AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_i8 / AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_i8 / AR_LIKE_o, Any)

assert_type(AR_LIKE_b / MAR_i8, MaskedArray[np.float64])
assert_type(AR_LIKE_u / MAR_i8, MaskedArray[np.float64])
assert_type(AR_LIKE_i / MAR_i8, MaskedArray[np.float64])
assert_type(AR_LIKE_f / MAR_i8, MaskedArray[np.float64])
assert_type(AR_LIKE_td64 / MAR_i8, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o / MAR_i8, Any)

assert_type(MAR_f8 / AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_f8 / AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_f8 / AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_f8 / AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_f8 / AR_LIKE_o, Any)

assert_type(AR_LIKE_b / MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_u / MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_i / MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_f / MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_td64 / MAR_f8, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o / MAR_f8, Any)

assert_type(MAR_td64 / AR_LIKE_u, MaskedArray[np.timedelta64])
assert_type(MAR_td64 / AR_LIKE_i, MaskedArray[np.timedelta64])
assert_type(MAR_td64 / AR_LIKE_f, MaskedArray[np.timedelta64])
assert_type(MAR_td64 / AR_LIKE_td64, MaskedArray[np.float64])
assert_type(MAR_td64 / AR_LIKE_o, Any)

assert_type(AR_LIKE_td64 / MAR_td64, MaskedArray[np.float64])
assert_type(AR_LIKE_o / MAR_td64, Any)

assert_type(MAR_o / AR_LIKE_b, Any)
assert_type(MAR_o / AR_LIKE_u, Any)
assert_type(MAR_o / AR_LIKE_i, Any)
assert_type(MAR_o / AR_LIKE_f, Any)
assert_type(MAR_o / AR_LIKE_td64, Any)
assert_type(MAR_o / AR_LIKE_dt64, Any)
assert_type(MAR_o / AR_LIKE_o, Any)

assert_type(AR_LIKE_b / MAR_o, Any)
assert_type(AR_LIKE_u / MAR_o, Any)
assert_type(AR_LIKE_i / MAR_o, Any)
assert_type(AR_LIKE_f / MAR_o, Any)
assert_type(AR_LIKE_td64 / MAR_o, Any)
assert_type(AR_LIKE_dt64 / MAR_o, Any)
assert_type(AR_LIKE_o / MAR_o, Any)

# MaskedArray floor division

assert_type(MAR_b // AR_LIKE_b, MaskedArray[np.int8])
assert_type(MAR_b // AR_LIKE_u, MaskedArray[np.uint32])
assert_type(MAR_b // AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_b // AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_b // AR_LIKE_o, Any)

assert_type(AR_LIKE_b // MAR_b, MaskedArray[np.int8])
assert_type(AR_LIKE_u // MAR_b, MaskedArray[np.uint32])
assert_type(AR_LIKE_i // MAR_b, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f // MAR_b, MaskedArray[np.floating])
assert_type(AR_LIKE_o // MAR_b, Any)

assert_type(MAR_u4 // AR_LIKE_b, MaskedArray[np.uint32])
assert_type(MAR_u4 // AR_LIKE_u, MaskedArray[np.unsignedinteger])
assert_type(MAR_u4 // AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_u4 // AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_u4 // AR_LIKE_o, Any)

assert_type(AR_LIKE_b // MAR_u4, MaskedArray[np.uint32])
assert_type(AR_LIKE_u // MAR_u4, MaskedArray[np.unsignedinteger])
assert_type(AR_LIKE_i // MAR_u4, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f // MAR_u4, MaskedArray[np.floating])
assert_type(AR_LIKE_td64 // MAR_u4, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o // MAR_u4, Any)

assert_type(MAR_i8 // AR_LIKE_b, MaskedArray[np.int64])
assert_type(MAR_i8 // AR_LIKE_u, MaskedArray[np.signedinteger])
assert_type(MAR_i8 // AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_i8 // AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_i8 // AR_LIKE_o, Any)

assert_type(AR_LIKE_b // MAR_i8, MaskedArray[np.int64])
assert_type(AR_LIKE_u // MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_i // MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f // MAR_i8, MaskedArray[np.floating])
assert_type(AR_LIKE_td64 // MAR_i8, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o // MAR_i8, Any)

assert_type(MAR_f8 // AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_f8 // AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_f8 // AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_f8 // AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_f8 // AR_LIKE_o, Any)

assert_type(AR_LIKE_b // MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_u // MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_i // MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_f // MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_td64 // MAR_f8, MaskedArray[np.timedelta64])
assert_type(AR_LIKE_o // MAR_f8, Any)

assert_type(MAR_td64 // AR_LIKE_u, MaskedArray[np.timedelta64])
assert_type(MAR_td64 // AR_LIKE_i, MaskedArray[np.timedelta64])
assert_type(MAR_td64 // AR_LIKE_f, MaskedArray[np.timedelta64])
assert_type(MAR_td64 // AR_LIKE_td64, MaskedArray[np.int64])
assert_type(MAR_td64 // AR_LIKE_o, Any)

assert_type(AR_LIKE_td64 // MAR_td64, MaskedArray[np.int64])
assert_type(AR_LIKE_o // MAR_td64, Any)

assert_type(MAR_o // AR_LIKE_b, Any)
assert_type(MAR_o // AR_LIKE_u, Any)
assert_type(MAR_o // AR_LIKE_i, Any)
assert_type(MAR_o // AR_LIKE_f, Any)
assert_type(MAR_o // AR_LIKE_td64, Any)
assert_type(MAR_o // AR_LIKE_dt64, Any)
assert_type(MAR_o // AR_LIKE_o, Any)

assert_type(AR_LIKE_b // MAR_o, Any)
assert_type(AR_LIKE_u // MAR_o, Any)
assert_type(AR_LIKE_i // MAR_o, Any)
assert_type(AR_LIKE_f // MAR_o, Any)
assert_type(AR_LIKE_td64 // MAR_o, Any)
assert_type(AR_LIKE_dt64 // MAR_o, Any)
assert_type(AR_LIKE_o // MAR_o, Any)

# Masked Array power

assert_type(MAR_b ** AR_LIKE_u, MaskedArray[np.uint32])
assert_type(MAR_b ** AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_b ** AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_b ** AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_b ** AR_LIKE_o, Any)

assert_type(AR_LIKE_u ** MAR_b, MaskedArray[np.uint32])
assert_type(AR_LIKE_i ** MAR_b, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f ** MAR_b, MaskedArray[np.floating])
assert_type(AR_LIKE_c ** MAR_b, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_o ** MAR_b, Any)

assert_type(MAR_u4 ** AR_LIKE_b, MaskedArray[np.uint32])
assert_type(MAR_u4 ** AR_LIKE_u, MaskedArray[np.unsignedinteger])
assert_type(MAR_u4 ** AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_u4 ** AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_u4 ** AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_u4 ** AR_LIKE_o, Any)

assert_type(AR_LIKE_b ** MAR_u4, MaskedArray[np.uint32])
assert_type(AR_LIKE_u ** MAR_u4, MaskedArray[np.unsignedinteger])
assert_type(AR_LIKE_i ** MAR_u4, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f ** MAR_u4, MaskedArray[np.floating])
assert_type(AR_LIKE_c ** MAR_u4, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_o ** MAR_u4, Any)

assert_type(MAR_i8 ** AR_LIKE_b, MaskedArray[np.int64])
assert_type(MAR_i8 ** AR_LIKE_u, MaskedArray[np.signedinteger])
assert_type(MAR_i8 ** AR_LIKE_i, MaskedArray[np.signedinteger])
assert_type(MAR_i8 ** AR_LIKE_f, MaskedArray[np.floating])
assert_type(MAR_i8 ** AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_i8 ** AR_LIKE_o, Any)
assert_type(MAR_i8 ** AR_LIKE_b, MaskedArray[np.int64])

assert_type(AR_LIKE_u ** MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_i ** MAR_i8, MaskedArray[np.signedinteger])
assert_type(AR_LIKE_f ** MAR_i8, MaskedArray[np.floating])
assert_type(AR_LIKE_c ** MAR_i8, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_o ** MAR_i8, Any)

assert_type(MAR_f8 ** AR_LIKE_b, MaskedArray[np.float64])
assert_type(MAR_f8 ** AR_LIKE_u, MaskedArray[np.float64])
assert_type(MAR_f8 ** AR_LIKE_i, MaskedArray[np.float64])
assert_type(MAR_f8 ** AR_LIKE_f, MaskedArray[np.float64])
assert_type(MAR_f8 ** AR_LIKE_c, MaskedArray[np.complexfloating])
assert_type(MAR_f8 ** AR_LIKE_o, Any)

assert_type(AR_LIKE_b ** MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_u ** MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_i ** MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_f ** MAR_f8, MaskedArray[np.float64])
assert_type(AR_LIKE_c ** MAR_f8, MaskedArray[np.complexfloating])
assert_type(AR_LIKE_o ** MAR_f8, Any)

assert_type(MAR_c16 ** AR_LIKE_b, MaskedArray[np.complex128])
assert_type(MAR_c16 ** AR_LIKE_u, MaskedArray[np.complex128])
assert_type(MAR_c16 ** AR_LIKE_i, MaskedArray[np.complex128])
assert_type(MAR_c16 ** AR_LIKE_f, MaskedArray[np.complex128])
assert_type(MAR_c16 ** AR_LIKE_c, MaskedArray[np.complex128])
assert_type(MAR_c16 ** AR_LIKE_o, Any)

assert_type(AR_LIKE_b ** MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_u ** MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_i ** MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_f ** MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_c ** MAR_c16, MaskedArray[np.complex128])
assert_type(AR_LIKE_o ** MAR_c16, Any)

assert_type(MAR_o ** AR_LIKE_b, Any)
assert_type(MAR_o ** AR_LIKE_u, Any)
assert_type(MAR_o ** AR_LIKE_i, Any)
assert_type(MAR_o ** AR_LIKE_f, Any)
assert_type(MAR_o ** AR_LIKE_c, Any)
assert_type(MAR_o ** AR_LIKE_o, Any)

assert_type(AR_LIKE_b ** MAR_o, Any)
assert_type(AR_LIKE_u ** MAR_o, Any)
assert_type(AR_LIKE_i ** MAR_o, Any)
assert_type(AR_LIKE_f ** MAR_o, Any)
assert_type(AR_LIKE_c ** MAR_o, Any)
assert_type(AR_LIKE_o ** MAR_o, Any)
