ꦫꦣꦺꦤ꧀ꦄꦤ꧀ꦠꦱꦺꦤ

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

"""
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]

[-] test_custom_dtypes.py [open]
[-] test_array_interface.py [open]
[-] test_mem_policy.py [open]
[-] test_dlpack.py [open]
[-] test_argparse.py [open]
[-] test_getlimits.py [open]
[+] ..
[-] test_machar.py [open]
[-] _locales.py [open]
[-] test_arraymethod.py [open]
[-] test_scalarinherit.py [open]
[-] test_shape_base.py [open]
[-] test_scalar_ctors.py [open]
[-] test_half.py [open]
[-] test_conversion_utils.py [open]
[+] __pycache__
[+] examples
[-] test_numerictypes.py [open]
[-] test_datetime.py [open]
[-] test_einsum.py [open]
[-] test_abc.py [open]
[-] test_print.py [open]
[-] test_nep50_promotions.py [open]
[-] test_indexerrors.py [open]
[-] test_errstate.py [open]
[-] test_simd_module.py [open]
[-] test_umath_complex.py [open]
[-] test_mem_overlap.py [open]
[-] test_overrides.py [open]
[-] test_records.py [open]
[-] test_umath_accuracy.py [open]
[-] test_cpu_dispatcher.py [open]
[-] test_scalarbuffer.py [open]
[-] test_numeric.py [open]
[-] test_dtype.py [open]
[-] test_multiarray.py [open]
[-] test_extint128.py [open]
[-] test__exceptions.py [open]
[-] test_deprecations.py [open]
[-] test_hashtable.py [open]
[-] test_ufunc.py [open]
[-] test_protocols.py [open]
[-] test_regression.py [open]
[-] test_casting_floatingpoint_errors.py [open]
[-] test_item_selection.py [open]
[-] test_indexing.py [open]
[-] test_nditer.py [open]
[-] test_scalar_methods.py [open]
[+] data
[-] test_simd.py [open]
[-] test_arrayprint.py [open]
[-] test_strings.py [open]
[-] test_unicode.py [open]
[-] test_cpu_features.py [open]
[-] test_casting_unittests.py [open]
[-] test_cython.py [open]
[-] test_memmap.py [open]
[-] __init__.py [open]
[-] test_defchararray.py [open]
[-] test_longdouble.py [open]
[-] test_array_coercion.py [open]
[-] test_function_base.py [open]
[-] test_scalarmath.py [open]
[-] test_api.py [open]
[-] test_limited_api.py [open]
[-] test_scalarprint.py [open]
[-] test_umath.py [open]