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_arraymethod.py

"""
This file tests the generic aspects of ArrayMethod.  At the time of writing
this is private API, but when added, public API may be added here.
"""

from __future__ import annotations

import sys
import types
from typing import Any

import pytest

import numpy as np
from numpy.core._multiarray_umath import _get_castingimpl as get_castingimpl

class TestResolveDescriptors:
    # Test mainly error paths of the resolve_descriptors function,
    # note that the `casting_unittests` tests exercise this non-error paths.

# Casting implementations are the main/only current user:
    method = get_castingimpl(type(np.dtype("d")), type(np.dtype("f")))

@pytest.mark.parametrize("args", [
        (True,),  # Not a tuple.
        ((None,)),  # Too few elements
        ((None, None, None),),  # Too many
        ((None, None),),  # Input dtype is None, which is invalid.
        ((np.dtype("d"), True),),  # Output dtype is not a dtype
        ((np.dtype("f"), None),),  # Input dtype does not match method
    ])
    def test_invalid_arguments(self, args):
        with pytest.raises(TypeError):
            self.method._resolve_descriptors(*args)

class TestSimpleStridedCall:
    # Test mainly error paths of the resolve_descriptors function,
    # note that the `casting_unittests` tests exercise this non-error paths.

# Casting implementations are the main/only current user:
    method = get_castingimpl(type(np.dtype("d")), type(np.dtype("f")))

@pytest.mark.parametrize(["args", "error"], [
        ((True,), TypeError),  # Not a tuple
        (((None,),), TypeError),  # Too few elements
        ((None, None), TypeError),  # Inputs are not arrays.
        (((None, None, None),), TypeError),  # Too many
        (((np.arange(3), np.arange(3)),), TypeError),  # Incorrect dtypes
        (((np.ones(3, dtype=">d"), np.ones(3, dtype="<f")),),
         TypeError),  # Does not support byte-swapping
        (((np.ones((2, 2), dtype="d"), np.ones((2, 2), dtype="f")),),
         ValueError),  # not 1-D
        (((np.ones(3, dtype="d"), np.ones(4, dtype="f")),),
          ValueError),  # different length
        (((np.frombuffer(b"\0x00"*3*2, dtype="d"),
           np.frombuffer(b"\0x00"*3, dtype="f")),),
         ValueError),  # output not writeable
    ])
    def test_invalid_arguments(self, args, error):
        # This is private API, which may be modified freely
        with pytest.raises(error):
            self.method._simple_strided_call(*args)

@pytest.mark.parametrize(
    "cls", [np.ndarray, np.recarray, np.chararray, np.matrix, np.memmap]
)
class TestClassGetItem:
    def test_class_getitem(self, cls: type[np.ndarray]) -> None:
        """Test `ndarray.__class_getitem__`."""
        alias = cls[Any, Any]
        assert isinstance(alias, types.GenericAlias)
        assert alias.__origin__ is cls

@pytest.mark.parametrize("arg_len", range(4))
    def test_subscript_tup(self, cls: type[np.ndarray], arg_len: int) -> None:
        arg_tup = (Any,) * arg_len
        if arg_len in (1, 2):
            assert cls[arg_tup]
        else:
            match = f"Too {'few' if arg_len == 0 else 'many'} arguments"
            with pytest.raises(TypeError, match=match):
                cls[arg_tup]