golden hour

/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/core/tests

File/Folder	Size	Actions
__init__.py	0 B
__pycache__	-
_locales.py	2.15 KB
data	-
examples	-
test__exceptions.py	2.78 KB
test_abc.py	2.17 KB
test_api.py	22.46 KB
test_argparse.py	1.92 KB
test_array_coercion.py	33.57 KB
test_array_interface.py	7.42 KB
test_arraymethod.py	3.17 KB
test_arrayprint.py	39.51 KB
test_casting_floatingpoint_errors.py	4.94 KB
test_casting_unittests.py	33.49 KB
test_conversion_utils.py	6.41 KB
test_cpu_dispatcher.py	1.49 KB
test_cpu_features.py	14.51 KB
test_custom_dtypes.py	9.18 KB
test_cython.py	3.54 KB
test_datetime.py	113.49 KB
test_defchararray.py	24.41 KB
test_deprecations.py	30.35 KB
test_dlpack.py	3.44 KB
test_dtype.py	73.52 KB
test_einsum.py	51.72 KB
test_errstate.py	2.17 KB
test_extint128.py	5.51 KB
test_function_base.py	15.23 KB
test_getlimits.py	6.56 KB
test_half.py	23.66 KB
test_hashtable.py	1011 B
test_indexerrors.py	5.01 KB
test_indexing.py	53.04 KB
test_item_selection.py	6.31 KB
test_limited_api.py	1.14 KB
test_longdouble.py	13.58 KB
test_machar.py	1.04 KB
test_mem_overlap.py	28.4 KB
test_mem_policy.py	15.63 KB
test_memmap.py	7.3 KB
test_multiarray.py	370.43 KB
test_nditer.py	127.75 KB
test_nep50_promotions.py	8.63 KB
test_numeric.py	133.34 KB
test_numerictypes.py	21.18 KB
test_overrides.py	25.47 KB
test_print.py	6.68 KB
test_protocols.py	1.14 KB
test_records.py	19.79 KB
test_regression.py	89.3 KB
test_scalar_ctors.py	5.97 KB
test_scalar_methods.py	7.36 KB
test_scalarbuffer.py	5.45 KB
test_scalarinherit.py	2.31 KB
test_scalarmath.py	41.88 KB
test_scalarprint.py	18.33 KB
test_shape_base.py	29.03 KB
test_simd.py	47.55 KB
test_simd_module.py	3.72 KB
test_strings.py	3.75 KB
test_ufunc.py	121.24 KB
test_umath.py	180.79 KB
test_umath_accuracy.py	3.81 KB
test_umath_complex.py	22.7 KB
test_unicode.py	12.48 KB

Edit: test_casting_floatingpoint_errors.py

import pytest
from pytest import param
from numpy.testing import IS_WASM
import numpy as np

def values_and_dtypes():
    """
    Generate value+dtype pairs that generate floating point errors during
    casts.  The invalid casts to integers will generate "invalid" value
    warnings, the float casts all generate "overflow".

(The Python int/float paths don't need to get tested in all the same
    situations, but it does not hurt.)
    """
    # Casting to float16:
    yield param(70000, "float16", id="int-to-f2")
    yield param("70000", "float16", id="str-to-f2")
    yield param(70000.0, "float16", id="float-to-f2")
    yield param(np.longdouble(70000.), "float16", id="longdouble-to-f2")
    yield param(np.float64(70000.), "float16", id="double-to-f2")
    yield param(np.float32(70000.), "float16", id="float-to-f2")
    # Casting to float32:
    yield param(10**100, "float32", id="int-to-f4")
    yield param(1e100, "float32", id="float-to-f2")
    yield param(np.longdouble(1e300), "float32", id="longdouble-to-f2")
    yield param(np.float64(1e300), "float32", id="double-to-f2")
    # Casting to float64:
    # If longdouble is double-double, its max can be rounded down to the double
    # max.  So we correct the double spacing (a bit weird, admittedly):
    max_ld = np.finfo(np.longdouble).max
    spacing = np.spacing(np.nextafter(np.finfo("f8").max, 0))
    if max_ld - spacing > np.finfo("f8").max:
        yield param(np.finfo(np.longdouble).max, "float64",
                    id="longdouble-to-f8")

# Cast to complex32:
    yield param(2e300, "complex64", id="float-to-c8")
    yield param(2e300+0j, "complex64", id="complex-to-c8")
    yield param(2e300j, "complex64", id="complex-to-c8")
    yield param(np.longdouble(2e300), "complex64", id="longdouble-to-c8")

# Invalid float to integer casts:
    with np.errstate(over="ignore"):
        for to_dt in np.typecodes["AllInteger"]:
            for value in [np.inf, np.nan]:
                for from_dt in np.typecodes["AllFloat"]:
                    from_dt = np.dtype(from_dt)
                    from_val = from_dt.type(value)

yield param(from_val, to_dt, id=f"{from_val}-to-{to_dt}")

def check_operations(dtype, value):
    """
    There are many dedicated paths in NumPy which cast and should check for
    floating point errors which occurred during those casts.
    """
    if dtype.kind != 'i':
        # These assignments use the stricter setitem logic:
        def assignment():
            arr = np.empty(3, dtype=dtype)
            arr[0] = value

yield assignment

def fill():
            arr = np.empty(3, dtype=dtype)
            arr.fill(value)

yield fill

def copyto_scalar():
        arr = np.empty(3, dtype=dtype)
        np.copyto(arr, value, casting="unsafe")

yield copyto_scalar

def copyto():
        arr = np.empty(3, dtype=dtype)
        np.copyto(arr, np.array([value, value, value]), casting="unsafe")

yield copyto

def copyto_scalar_masked():
        arr = np.empty(3, dtype=dtype)
        np.copyto(arr, value, casting="unsafe",
                  where=[True, False, True])

yield copyto_scalar_masked

def copyto_masked():
        arr = np.empty(3, dtype=dtype)
        np.copyto(arr, np.array([value, value, value]), casting="unsafe",
                  where=[True, False, True])

yield copyto_masked

def direct_cast():
        np.array([value, value, value]).astype(dtype)

yield direct_cast

def direct_cast_nd_strided():
        arr = np.full((5, 5, 5), fill_value=value)[:, ::2, :]
        arr.astype(dtype)

yield direct_cast_nd_strided

def boolean_array_assignment():
        arr = np.empty(3, dtype=dtype)
        arr[[True, False, True]] = np.array([value, value])

yield boolean_array_assignment

def integer_array_assignment():
        arr = np.empty(3, dtype=dtype)
        values = np.array([value, value])

arr[[0, 1]] = values

yield integer_array_assignment

def integer_array_assignment_with_subspace():
        arr = np.empty((5, 3), dtype=dtype)
        values = np.array([value, value, value])

arr[[0, 2]] = values

yield integer_array_assignment_with_subspace

def flat_assignment():
        arr = np.empty((3,), dtype=dtype)
        values = np.array([value, value, value])
        arr.flat[:] = values

yield flat_assignment

@pytest.mark.skipif(IS_WASM, reason="no wasm fp exception support")
@pytest.mark.parametrize(["value", "dtype"], values_and_dtypes())
@pytest.mark.filterwarnings("ignore::numpy.ComplexWarning")
def test_floatingpoint_errors_casting(dtype, value):
    dtype = np.dtype(dtype)
    for operation in check_operations(dtype, value):
        dtype = np.dtype(dtype)

match = "invalid" if dtype.kind in 'iu' else "overflow"
        with pytest.warns(RuntimeWarning, match=match):
            operation()

with np.errstate(all="raise"):
            with pytest.raises(FloatingPointError, match=match):
                operation()