X7ROOT File Manager
Current Path:
/opt/cloudlinux/venv/lib/python3.11/site-packages/numpy/core/tests
opt
/
cloudlinux
/
venv
/
lib
/
python3.11
/
site-packages
/
numpy
/
core
/
tests
/
📁
..
📄
__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)
Editing: test_strings.py
import pytest import operator import numpy as np from numpy.testing import assert_array_equal COMPARISONS = [ (operator.eq, np.equal, "=="), (operator.ne, np.not_equal, "!="), (operator.lt, np.less, "<"), (operator.le, np.less_equal, "<="), (operator.gt, np.greater, ">"), (operator.ge, np.greater_equal, ">="), ] @pytest.mark.parametrize(["op", "ufunc", "sym"], COMPARISONS) def test_mixed_string_comparison_ufuncs_fail(op, ufunc, sym): arr_string = np.array(["a", "b"], dtype="S") arr_unicode = np.array(["a", "c"], dtype="U") with pytest.raises(TypeError, match="did not contain a loop"): ufunc(arr_string, arr_unicode) with pytest.raises(TypeError, match="did not contain a loop"): ufunc(arr_unicode, arr_string) @pytest.mark.parametrize(["op", "ufunc", "sym"], COMPARISONS) def test_mixed_string_comparisons_ufuncs_with_cast(op, ufunc, sym): arr_string = np.array(["a", "b"], dtype="S") arr_unicode = np.array(["a", "c"], dtype="U") # While there is no loop, manual casting is acceptable: res1 = ufunc(arr_string, arr_unicode, signature="UU->?", casting="unsafe") res2 = ufunc(arr_string, arr_unicode, signature="SS->?", casting="unsafe") expected = op(arr_string.astype('U'), arr_unicode) assert_array_equal(res1, expected) assert_array_equal(res2, expected) @pytest.mark.parametrize(["op", "ufunc", "sym"], COMPARISONS) @pytest.mark.parametrize("dtypes", [ ("S2", "S2"), ("S2", "S10"), ("<U1", "<U1"), ("<U1", ">U1"), (">U1", ">U1"), ("<U1", "<U10"), ("<U1", ">U10")]) @pytest.mark.parametrize("aligned", [True, False]) def test_string_comparisons(op, ufunc, sym, dtypes, aligned): # ensure native byte-order for the first view to stay within unicode range native_dt = np.dtype(dtypes[0]).newbyteorder("=") arr = np.arange(2**15).view(native_dt).astype(dtypes[0]) if not aligned: # Make `arr` unaligned: new = np.zeros(arr.nbytes + 1, dtype=np.uint8)[1:].view(dtypes[0]) new[...] = arr arr = new arr2 = arr.astype(dtypes[1], copy=True) np.random.shuffle(arr2) arr[0] = arr2[0] # make sure one matches expected = [op(d1, d2) for d1, d2 in zip(arr.tolist(), arr2.tolist())] assert_array_equal(op(arr, arr2), expected) assert_array_equal(ufunc(arr, arr2), expected) assert_array_equal(np.compare_chararrays(arr, arr2, sym, False), expected) expected = [op(d2, d1) for d1, d2 in zip(arr.tolist(), arr2.tolist())] assert_array_equal(op(arr2, arr), expected) assert_array_equal(ufunc(arr2, arr), expected) assert_array_equal(np.compare_chararrays(arr2, arr, sym, False), expected) @pytest.mark.parametrize(["op", "ufunc", "sym"], COMPARISONS) @pytest.mark.parametrize("dtypes", [ ("S2", "S2"), ("S2", "S10"), ("<U1", "<U1"), ("<U1", ">U10")]) def test_string_comparisons_empty(op, ufunc, sym, dtypes): arr = np.empty((1, 0, 1, 5), dtype=dtypes[0]) arr2 = np.empty((100, 1, 0, 1), dtype=dtypes[1]) expected = np.empty(np.broadcast_shapes(arr.shape, arr2.shape), dtype=bool) assert_array_equal(op(arr, arr2), expected) assert_array_equal(ufunc(arr, arr2), expected) assert_array_equal(np.compare_chararrays(arr, arr2, sym, False), expected) @pytest.mark.parametrize("str_dt", ["S", "U"]) @pytest.mark.parametrize("float_dt", np.typecodes["AllFloat"]) def test_float_to_string_cast(str_dt, float_dt): float_dt = np.dtype(float_dt) fi = np.finfo(float_dt) arr = np.array([np.nan, np.inf, -np.inf, fi.max, fi.min], dtype=float_dt) expected = ["nan", "inf", "-inf", repr(fi.max), repr(fi.min)] if float_dt.kind == 'c': expected = [f"({r}+0j)" for r in expected] res = arr.astype(str_dt) assert_array_equal(res, np.array(expected, dtype=str_dt))
Upload File
Create Folder