festival.edmaven.com:80

Linux b747.shapiroconsult.com 3.10.0-1160.119.1.vz7.224.4 #1 SMP Mon Sep 30 15:36:27 MSK 2024 x86_64
Apache
173.231.212.80

File Name : 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))

Name	Type	Size	Owner/Group	Permission	Last Modified	Actions
.	dir	-	root/root	0755	2025-01-07 04:51:45
..	dir	-	root/root	0755	2025-01-07 04:51:45
__pycache__	dir	-	root/root	0755	2025-01-07 04:51:45
data	dir	-	root/root	0755	2025-01-07 04:51:45
examples	dir	-	root/root	0755	2025-01-06 04:11:07
__init__.py	application/x-empty	0 B	root/root	0644	2025-01-06 04:11:07
_locales.py	text/plain	2.15 KB	root/root	0644	2025-01-06 04:11:07
test__exceptions.py	text/plain	2.78 KB	root/root	0644	2025-01-06 04:11:07
test_abc.py	text/x-python	2.17 KB	root/root	0644	2025-01-06 04:11:07
test_api.py	text/x-python	22.46 KB	root/root	0644	2025-01-06 04:11:07
test_argparse.py	text/plain	1.92 KB	root/root	0644	2025-01-06 04:11:07
test_array_coercion.py	text/plain	33.57 KB	root/root	0644	2025-01-06 04:11:07
test_array_interface.py	text/x-python	7.59 KB	root/root	0644	2025-01-06 04:11:07
test_arraymethod.py	text/plain	3.17 KB	root/root	0644	2025-01-06 04:11:07
test_arrayprint.py	text/x-python	39.51 KB	root/root	0644	2025-01-06 04:11:07
test_casting_floatingpoint_errors.py	text/x-python	4.94 KB	root/root	0644	2025-01-06 04:11:07
test_casting_unittests.py	text/plain	33.49 KB	root/root	0644	2025-01-06 04:11:07
test_conversion_utils.py	text/plain	6.41 KB	root/root	0644	2025-01-06 04:11:07
test_cpu_dispatcher.py	text/x-python	1.49 KB	root/root	0644	2025-01-06 04:11:07
test_cpu_features.py	text/x-python	14.51 KB	root/root	0644	2025-01-06 04:11:07
test_custom_dtypes.py	text/x-python	9.18 KB	root/root	0644	2025-01-06 04:11:07
test_cython.py	text/x-python	3.36 KB	root/root	0644	2025-01-06 04:11:07
test_datetime.py	text/x-python	113.49 KB	root/root	0644	2025-01-06 04:11:07
test_defchararray.py	text/x-python	24.41 KB	root/root	0644	2025-01-06 04:11:07
test_deprecations.py	text/plain	30.35 KB	root/root	0644	2025-01-06 04:11:07
test_dlpack.py	text/x-python	3.44 KB	root/root	0644	2025-01-06 04:11:07
test_dtype.py	text/x-python	73.68 KB	root/root	0644	2025-01-06 04:11:07
test_einsum.py	text/x-python	52.45 KB	root/root	0644	2025-01-06 04:11:07
test_errstate.py	text/x-python	2.17 KB	root/root	0644	2025-01-06 04:11:07
test_extint128.py	text/x-python	5.51 KB	root/root	0644	2025-01-06 04:11:07
test_function_base.py	text/x-python	15.23 KB	root/root	0644	2025-01-06 04:11:07
test_getlimits.py	text/plain	6.56 KB	root/root	0644	2025-01-06 04:11:07
test_half.py	text/x-python	23.66 KB	root/root	0644	2025-01-06 04:11:07
test_hashtable.py	text/x-python	1011 B	root/root	0644	2025-01-06 04:11:07
test_indexerrors.py	text/x-python	5.01 KB	root/root	0644	2025-01-06 04:11:07
test_indexing.py	text/x-python	53.04 KB	root/root	0644	2025-01-06 04:11:07
test_item_selection.py	text/x-python	6.31 KB	root/root	0644	2025-01-06 04:11:07
test_limited_api.py	text/x-python	1.14 KB	root/root	0644	2025-01-06 04:11:07
test_longdouble.py	text/x-python	13.58 KB	root/root	0644	2025-01-06 04:11:07
test_machar.py	text/plain	1.04 KB	root/root	0644	2025-01-06 04:11:07
test_mem_overlap.py	text/x-python	28.4 KB	root/root	0644	2025-01-06 04:11:07
test_mem_policy.py	text/x-python	16.39 KB	root/root	0644	2025-01-06 04:11:07
test_memmap.py	text/x-python	7.3 KB	root/root	0644	2025-01-06 04:11:07
test_multiarray.py	text/x-python	370.92 KB	root/root	0644	2025-01-06 04:11:07
test_nditer.py	text/x-python	127.75 KB	root/root	0644	2025-01-06 04:11:07
test_nep50_promotions.py	text/plain	8.63 KB	root/root	0644	2025-01-06 04:11:07
test_numeric.py	text/x-python	133.34 KB	root/root	0644	2025-01-06 04:11:07
test_numerictypes.py	text/x-python	21.18 KB	root/root	0644	2025-01-06 04:11:07
test_overrides.py	text/x-python	25.47 KB	root/root	0644	2025-01-06 04:11:07
test_print.py	text/x-python	6.68 KB	root/root	0644	2025-01-06 04:11:07
test_protocols.py	text/x-python	1.14 KB	root/root	0644	2025-01-06 04:11:07
test_records.py	text/x-python	19.79 KB	root/root	0644	2025-01-06 04:11:07
test_regression.py	text/x-python	89.45 KB	root/root	0644	2025-01-06 04:11:07
test_scalar_ctors.py	text/plain	5.97 KB	root/root	0644	2025-01-06 04:11:07
test_scalar_methods.py	text/plain	7.36 KB	root/root	0644	2025-01-06 04:11:07
test_scalarbuffer.py	text/plain	5.45 KB	root/root	0644	2025-01-06 04:11:07
test_scalarinherit.py	text/plain	2.31 KB	root/root	0644	2025-01-06 04:11:07
test_scalarmath.py	text/x-python	42.23 KB	root/root	0644	2025-01-06 04:11:07
test_scalarprint.py	text/plain	18.33 KB	root/root	0644	2025-01-06 04:11:07
test_shape_base.py	text/x-python	29.03 KB	root/root	0644	2025-01-06 04:11:07
test_simd.py	text/x-python	47.55 KB	root/root	0644	2025-01-06 04:11:07
test_simd_module.py	text/x-python	3.72 KB	root/root	0644	2025-01-06 04:11:07
test_strings.py	text/x-python	3.75 KB	root/root	0644	2025-01-06 04:11:07
test_ufunc.py	text/x-python	121.88 KB	root/root	0644	2025-01-06 04:11:07
test_umath.py	text/x-python	181.25 KB	root/root	0644	2025-01-06 04:11:07
test_umath_accuracy.py	text/x-python	3.81 KB	root/root	0644	2025-01-06 04:11:07
test_umath_complex.py	text/x-python	22.7 KB	root/root	0644	2025-01-06 04:11:07
test_unicode.py	text/x-python	12.48 KB	root/root	0644	2025-01-06 04:11:07