FM / tests

/usr/local/lib64/python3.6/site-packages/numpy/core/tests/

/usr/local/lib64/python3.6/site-packages/numpy/core/tests

Name	Size	Mode	Actions
data/	-	0755	rm
__pycache__/	-	0755	rm
test_abc.py	2328	0644	edit dl rm
test_api.py	20806	0644	edit dl rm
test_arrayprint.py	34493	0644	edit dl rm
test_conversion_utils.py	4981	0644	edit dl rm
test_cpu_features.py	6785	0644	edit dl rm
test_datetime.py	108192	0644	edit dl rm
test_defchararray.py	24381	0644	edit dl rm
test_deprecations.py	24187	0644	edit dl rm
test_dtype.py	49291	0644	edit dl rm
test_einsum.py	45197	0644	edit dl rm
test_errstate.py	2066	0644	edit dl rm
test_extint128.py	5643	0644	edit dl rm
test_function_base.py	13148	0644	edit dl rm
test_getlimits.py	4297	0644	edit dl rm
test_half.py	23102	0644	edit dl rm
test_indexerrors.py	5130	0644	edit dl rm
test_indexing.py	49260	0644	edit dl rm
test_item_selection.py	3579	0644	edit dl rm
test_longdouble.py	13041	0644	edit dl rm
test_machar.py	1066	0644	edit dl rm
test_memmap.py	7464	0644	edit dl rm
test_mem_overlap.py	28855	0644	edit dl rm
test_multiarray.py	319797	0644	edit dl rm
test_nditer.py	112298	0644	edit dl rm
test_numeric.py	124118	0644	edit dl rm
test_numerictypes.py	20846	0644	edit dl rm
test_overrides.py	14433	0644	edit dl rm
test_print.py	6737	0644	edit dl rm
test_protocols.py	1168	0644	edit dl rm
test_records.py	19694	0644	edit dl rm
test_regression.py	88840	0644	edit dl rm
test_scalarbuffer.py	4065	0644	edit dl rm
test_scalarinherit.py	2127	0644	edit dl rm
test_scalarmath.py	28565	0644	edit dl rm
test_scalarprint.py	15158	0644	edit dl rm
test_scalar_ctors.py	2580	0644	edit dl rm
test_scalar_methods.py	4063	0644	edit dl rm
test_shape_base.py	24679	0644	edit dl rm
test_ufunc.py	83978	0644	edit dl rm
test_umath.py	121442	0644	edit dl rm
test_umath_accuracy.py	3044	0644	edit dl rm
test_umath_complex.py	23145	0644	edit dl rm
test_unicode.py	12553	0644	edit dl rm
test__exceptions.py	2005	0644	edit dl rm
_locales.py	2192	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/numpy/core/tests/test_print.py (6737B)

import sys

import pytest

import numpy as np
from numpy.testing import assert_, assert_equal
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)))

def test_locale_longdouble(self):
        assert_equal(str(np.longdouble('1.2')), str(float(1.2)))

cmd: