test_abs.py

"""
Special cases tests for abs.

These tests are generated from the special cases listed in the spec.

NOTE: This file is generated automatically by the generate_stubs.py script. Do
not modify it directly.
"""

from ..array_helpers import NaN, assert_exactly_equal, exactly_equal, infinity, zero
from ..hypothesis_helpers import numeric_arrays
from .._array_module import abs

from hypothesis import given


@given(numeric_arrays)
def test_abs_special_cases_one_arg_equal_1(arg1):
    """
    Special case test for `abs(x, /)`:

        -   If `x_i` is `NaN`, the result is `NaN`.

    """
    res = abs(arg1)
    mask = exactly_equal(arg1, NaN(arg1.shape, arg1.dtype))
    assert_exactly_equal(res[mask], (NaN(arg1.shape, arg1.dtype))[mask])


@given(numeric_arrays)
def test_abs_special_cases_one_arg_equal_2(arg1):
    """
    Special case test for `abs(x, /)`:

        -   If `x_i` is `-0`, the result is `+0`.

    """
    res = abs(arg1)
    mask = exactly_equal(arg1, -zero(arg1.shape, arg1.dtype))
    assert_exactly_equal(res[mask], (zero(arg1.shape, arg1.dtype))[mask])


@given(numeric_arrays)
def test_abs_special_cases_one_arg_equal_3(arg1):
    """
    Special case test for `abs(x, /)`:

        -   If `x_i` is `-infinity`, the result is `+infinity`.

    """
    res = abs(arg1)
    mask = exactly_equal(arg1, -infinity(arg1.shape, arg1.dtype))
    assert_exactly_equal(res[mask], (infinity(arg1.shape, arg1.dtype))[mask])