Na}{

File Content: 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]

Name	Size	Permission
__pycache__	---	0755
data	---	0755
examples	---	0755
__init__.py	0 bytes	0644
_locales.py	2206 bytes	0644
test__exceptions.py	2846 bytes	0644
test_abc.py	2220 bytes	0644
test_api.py	22995 bytes	0644
test_argparse.py	1969 bytes	0644
test_array_coercion.py	34379 bytes	0644
test_array_interface.py	7596 bytes	0644
test_arraymethod.py	3244 bytes	0644
test_arrayprint.py	40462 bytes	0644
test_casting_floatingpoint_errors.py	5063 bytes	0644
test_casting_unittests.py	34298 bytes	0644
test_conversion_utils.py	6559 bytes	0644
test_cpu_dispatcher.py	1521 bytes	0644
test_cpu_features.py	14858 bytes	0644
test_custom_dtypes.py	9401 bytes	0644
test_cython.py	3623 bytes	0644
test_datetime.py	116211 bytes	0644
test_defchararray.py	24997 bytes	0644
test_deprecations.py	31076 bytes	0644
test_dlpack.py	3522 bytes	0644
test_dtype.py	75288 bytes	0644
test_einsum.py	52960 bytes	0644
test_errstate.py	2219 bytes	0644
test_extint128.py	5643 bytes	0644
test_function_base.py	15595 bytes	0644
test_getlimits.py	6718 bytes	0644
test_half.py	24226 bytes	0644
test_hashtable.py	1011 bytes	0644
test_indexerrors.py	5130 bytes	0644
test_indexing.py	54314 bytes	0644
test_item_selection.py	6458 bytes	0644
test_limited_api.py	1172 bytes	0644
test_longdouble.py	13905 bytes	0644
test_machar.py	1067 bytes	0644
test_mem_overlap.py	29086 bytes	0644
test_mem_policy.py	16004 bytes	0644
test_memmap.py	7477 bytes	0644
test_multiarray.py	379320 bytes	0644
test_nditer.py	130818 bytes	0644
test_nep50_promotions.py	8840 bytes	0644
test_numeric.py	136543 bytes	0644
test_numerictypes.py	21687 bytes	0644
test_overrides.py	26080 bytes	0644
test_print.py	6837 bytes	0644
test_protocols.py	1168 bytes	0644
test_records.py	20269 bytes	0644
test_regression.py	91447 bytes	0644
test_scalar_ctors.py	6115 bytes	0644
test_scalar_methods.py	7541 bytes	0644
test_scalarbuffer.py	5580 bytes	0644
test_scalarinherit.py	2368 bytes	0644
test_scalarmath.py	42885 bytes	0644
test_scalarprint.py	18771 bytes	0644
test_shape_base.py	29723 bytes	0644
test_simd.py	48696 bytes	0644
test_simd_module.py	3805 bytes	0644
test_strings.py	3835 bytes	0644
test_ufunc.py	124145 bytes	0644
test_umath.py	185129 bytes	0644
test_umath_accuracy.py	3897 bytes	0644
test_umath_complex.py	23243 bytes	0644
test_unicode.py	12775 bytes	0644