golden hour

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

⬆️ Go Up

Upload

File/Folder	Size	Actions
__init__.py	0 B	Del OK
__pycache__	-	Del OK
_locales.py	2.15 KB	Del OK
data	-	Del OK
examples	-	Del OK
test__exceptions.py	2.78 KB	Del OK
test_abc.py	2.17 KB	Del OK
test_api.py	22.46 KB	Del OK
test_argparse.py	1.92 KB	Del OK
test_array_coercion.py	33.57 KB	Del OK
test_array_interface.py	7.42 KB	Del OK
test_arraymethod.py	3.17 KB	Del OK
test_arrayprint.py	39.51 KB	Del OK
test_casting_floatingpoint_errors.py	4.94 KB	Del OK
test_casting_unittests.py	33.49 KB	Del OK
test_conversion_utils.py	6.41 KB	Del OK
test_cpu_dispatcher.py	1.49 KB	Del OK
test_cpu_features.py	14.51 KB	Del OK
test_custom_dtypes.py	9.18 KB	Del OK
test_cython.py	3.54 KB	Del OK
test_datetime.py	113.49 KB	Del OK
test_defchararray.py	24.41 KB	Del OK
test_deprecations.py	30.35 KB	Del OK
test_dlpack.py	3.44 KB	Del OK
test_dtype.py	73.52 KB	Del OK
test_einsum.py	51.72 KB	Del OK
test_errstate.py	2.17 KB	Del OK
test_extint128.py	5.51 KB	Del OK
test_function_base.py	15.23 KB	Del OK
test_getlimits.py	6.56 KB	Del OK
test_half.py	23.66 KB	Del OK
test_hashtable.py	1011 B	Del OK
test_indexerrors.py	5.01 KB	Del OK
test_indexing.py	53.04 KB	Del OK
test_item_selection.py	6.31 KB	Del OK
test_limited_api.py	1.14 KB	Del OK
test_longdouble.py	13.58 KB	Del OK
test_machar.py	1.04 KB	Del OK
test_mem_overlap.py	28.4 KB	Del OK
test_mem_policy.py	15.63 KB	Del OK
test_memmap.py	7.3 KB	Del OK
test_multiarray.py	370.43 KB	Del OK
test_nditer.py	127.75 KB	Del OK
test_nep50_promotions.py	8.63 KB	Del OK
test_numeric.py	133.34 KB	Del OK
test_numerictypes.py	21.18 KB	Del OK
test_overrides.py	25.47 KB	Del OK
test_print.py	6.68 KB	Del OK
test_protocols.py	1.14 KB	Del OK
test_records.py	19.79 KB	Del OK
test_regression.py	89.3 KB	Del OK
test_scalar_ctors.py	5.97 KB	Del OK
test_scalar_methods.py	7.36 KB	Del OK
test_scalarbuffer.py	5.45 KB	Del OK
test_scalarinherit.py	2.31 KB	Del OK
test_scalarmath.py	41.88 KB	Del OK
test_scalarprint.py	18.33 KB	Del OK
test_shape_base.py	29.03 KB	Del OK
test_simd.py	47.55 KB	Del OK
test_simd_module.py	3.72 KB	Del OK
test_strings.py	3.75 KB	Del OK
test_ufunc.py	121.24 KB	Del OK
test_umath.py	180.79 KB	Del OK
test_umath_accuracy.py	3.81 KB	Del OK
test_umath_complex.py	22.7 KB	Del OK
test_unicode.py	12.48 KB	Del OK

Edit: test_print.py

import sys import pytest import numpy as np from numpy.testing import assert_, assert_equal, IS_MUSL from numpy.core.tests._locales import CommaDecimalPointLocale from io import StringIO _REF = {np.inf: 'inf', -np.inf: '-inf', np.nan: 'nan'} @pytest.mark.parametrize('tp', [np.float32, np.double, np.longdouble]) def test_float_types(tp): """ Check formatting. This is only for the str function, and only for simple types. The precision of np.float32 and np.longdouble aren't the same as the python float precision. """ for x in [0, 1, -1, 1e20]: assert_equal(str(tp(x)), str(float(x)), err_msg='Failed str formatting for type %s' % tp) if tp(1e16).itemsize > 4: assert_equal(str(tp(1e16)), str(float('1e16')), err_msg='Failed str formatting for type %s' % tp) else: ref = '1e+16' assert_equal(str(tp(1e16)), ref, err_msg='Failed str formatting for type %s' % tp) @pytest.mark.parametrize('tp', [np.float32, np.double, np.longdouble]) def test_nan_inf_float(tp): """ Check formatting of nan & inf. This is only for the str function, and only for simple types. The precision of np.float32 and np.longdouble aren't the same as the python float precision. """ for x in [np.inf, -np.inf, np.nan]: assert_equal(str(tp(x)), _REF[x], err_msg='Failed str formatting for type %s' % tp) @pytest.mark.parametrize('tp', [np.complex64, np.cdouble, np.clongdouble]) def test_complex_types(tp): """Check formatting of complex types. This is only for the str function, and only for simple types. The precision of np.float32 and np.longdouble aren't the same as the python float precision. """ for x in [0, 1, -1, 1e20]: assert_equal(str(tp(x)), str(complex(x)), err_msg='Failed str formatting for type %s' % tp) assert_equal(str(tp(x*1j)), str(complex(x*1j)), err_msg='Failed str formatting for type %s' % tp) assert_equal(str(tp(x + x*1j)), str(complex(x + x*1j)), err_msg='Failed str formatting for type %s' % tp) if tp(1e16).itemsize > 8: assert_equal(str(tp(1e16)), str(complex(1e16)), err_msg='Failed str formatting for type %s' % tp) else: ref = '(1e+16+0j)' assert_equal(str(tp(1e16)), ref, err_msg='Failed str formatting for type %s' % tp) @pytest.mark.parametrize('dtype', [np.complex64, np.cdouble, np.clongdouble]) def test_complex_inf_nan(dtype): """Check inf/nan formatting of complex types.""" TESTS = { complex(np.inf, 0): "(inf+0j)", complex(0, np.inf): "infj", complex(-np.inf, 0): "(-inf+0j)", complex(0, -np.inf): "-infj", complex(np.inf, 1): "(inf+1j)", complex(1, np.inf): "(1+infj)", complex(-np.inf, 1): "(-inf+1j)", complex(1, -np.inf): "(1-infj)", complex(np.nan, 0): "(nan+0j)", complex(0, np.nan): "nanj", complex(-np.nan, 0): "(nan+0j)", complex(0, -np.nan): "nanj", complex(np.nan, 1): "(nan+1j)", complex(1, np.nan): "(1+nanj)", complex(-np.nan, 1): "(nan+1j)", complex(1, -np.nan): "(1+nanj)", } for c, s in TESTS.items(): assert_equal(str(dtype(c)), s) # print tests def _test_redirected_print(x, tp, ref=None): file = StringIO() file_tp = StringIO() stdout = sys.stdout try: sys.stdout = file_tp print(tp(x)) sys.stdout = file if ref: print(ref) else: print(x) finally: sys.stdout = stdout assert_equal(file.getvalue(), file_tp.getvalue(), err_msg='print failed for type%s' % tp) @pytest.mark.parametrize('tp', [np.float32, np.double, np.longdouble]) def test_float_type_print(tp): """Check formatting when using print """ for x in [0, 1, -1, 1e20]: _test_redirected_print(float(x), tp) for x in [np.inf, -np.inf, np.nan]: _test_redirected_print(float(x), tp, _REF[x]) if tp(1e16).itemsize > 4: _test_redirected_print(float(1e16), tp) else: ref = '1e+16' _test_redirected_print(float(1e16), tp, ref) @pytest.mark.parametrize('tp', [np.complex64, np.cdouble, np.clongdouble]) def test_complex_type_print(tp): """Check formatting when using print """ # We do not create complex with inf/nan directly because the feature is # missing in python < 2.6 for x in [0, 1, -1, 1e20]: _test_redirected_print(complex(x), tp) if tp(1e16).itemsize > 8: _test_redirected_print(complex(1e16), tp) else: ref = '(1e+16+0j)' _test_redirected_print(complex(1e16), tp, ref) _test_redirected_print(complex(np.inf, 1), tp, '(inf+1j)') _test_redirected_print(complex(-np.inf, 1), tp, '(-inf+1j)') _test_redirected_print(complex(-np.nan, 1), tp, '(nan+1j)') def test_scalar_format(): """Test the str.format method with NumPy scalar types""" tests = [('{0}', True, np.bool_), ('{0}', False, np.bool_), ('{0:d}', 130, np.uint8), ('{0:d}', 50000, np.uint16), ('{0:d}', 3000000000, np.uint32), ('{0:d}', 15000000000000000000, np.uint64), ('{0:d}', -120, np.int8), ('{0:d}', -30000, np.int16), ('{0:d}', -2000000000, np.int32), ('{0:d}', -7000000000000000000, np.int64), ('{0:g}', 1.5, np.float16), ('{0:g}', 1.5, np.float32), ('{0:g}', 1.5, np.float64), ('{0:g}', 1.5, np.longdouble), ('{0:g}', 1.5+0.5j, np.complex64), ('{0:g}', 1.5+0.5j, np.complex128), ('{0:g}', 1.5+0.5j, np.clongdouble)] for (fmat, val, valtype) in tests: try: assert_equal(fmat.format(val), fmat.format(valtype(val)), "failed with val %s, type %s" % (val, valtype)) except ValueError as e: assert_(False, "format raised exception (fmt='%s', val=%s, type=%s, exc='%s')" % (fmat, repr(val), repr(valtype), str(e))) # # Locale tests: scalar types formatting should be independent of the locale # class TestCommaDecimalPointLocale(CommaDecimalPointLocale): def test_locale_single(self): assert_equal(str(np.float32(1.2)), str(float(1.2))) def test_locale_double(self): assert_equal(str(np.double(1.2)), str(float(1.2))) @pytest.mark.skipif(IS_MUSL, reason="test flaky on musllinux") def test_locale_longdouble(self): assert_equal(str(np.longdouble('1.2')), str(float(1.2)))
Save