FM / tests

/opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests/

/opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests

Name	Size	Mode	Actions
data/	-	0755	rm
__pycache__/	-	0755	rm
test_arraypad.py	54830	0644	edit dl rm
test_arraysetops.py	35912	0644	edit dl rm
test_arrayterator.py	1291	0644	edit dl rm
test_financial_expired.py	247	0644	edit dl rm
test_format.py	41028	0644	edit dl rm
test_function_base.py	157726	0644	edit dl rm
test_histograms.py	32734	0644	edit dl rm
test_index_tricks.py	20256	0644	edit dl rm
test_io.py	107891	0644	edit dl rm
test_loadtxt.py	38560	0644	edit dl rm
test_mixins.py	7030	0644	edit dl rm
test_nanfunctions.py	47609	0644	edit dl rm
test_packbits.py	17546	0644	edit dl rm
test_polynomial.py	11395	0644	edit dl rm
test_recfunctions.py	44001	0644	edit dl rm
test_regression.py	8257	0644	edit dl rm
test_shape_base.py	26817	0644	edit dl rm
test_stride_tricks.py	22849	0644	edit dl rm
test_twodim_base.py	18841	0644	edit dl rm
test_type_check.py	15114	0644	edit dl rm
test_ufunclike.py	2982	0644	edit dl rm
test_utils.py	6218	0644	edit dl rm
test__datasource.py	10571	0644	edit dl rm
test__iotools.py	13743	0644	edit dl rm
test__version.py	1999	0644	edit dl rm
__init__.py	0	0644	edit dl rm

Edit: /opt/imh-python/lib/python3.9/site-packages/numpy/lib/tests/test_arrayterator.py (1291B)

from operator import mul
from functools import reduce

import numpy as np
from numpy.random import randint
from numpy.lib import Arrayterator
from numpy.testing import assert_

def test():
    np.random.seed(np.arange(10))

# Create a random array
    ndims = randint(5)+1
    shape = tuple(randint(10)+1 for dim in range(ndims))
    els = reduce(mul, shape)
    a = np.arange(els)
    a.shape = shape

buf_size = randint(2*els)
    b = Arrayterator(a, buf_size)

# Check that each block has at most ``buf_size`` elements
    for block in b:
        assert_(len(block.flat) <= (buf_size or els))

# Check that all elements are iterated correctly
    assert_(list(b.flat) == list(a.flat))

# Slice arrayterator
    start = [randint(dim) for dim in shape]
    stop = [randint(dim)+1 for dim in shape]
    step = [randint(dim)+1 for dim in shape]
    slice_ = tuple(slice(*t) for t in zip(start, stop, step))
    c = b[slice_]
    d = a[slice_]

# Check that each block has at most ``buf_size`` elements
    for block in c:
        assert_(len(block.flat) <= (buf_size or els))

# Check that the arrayterator is sliced correctly
    assert_(np.all(c.__array__() == d))

# Check that all elements are iterated correctly
    assert_(list(c.flat) == list(d.flat))

from operator import mul from functools import reduce import numpy as np from numpy.random import randint from numpy.lib import Arrayterator from numpy.testing import assert_ def test(): np.random.seed(np.arange(10)) # Create a random array ndims = randint(5)+1 shape = tuple(randint(10)+1 for dim in range(ndims)) els = reduce(mul, shape) a = np.arange(els) a.shape = shape buf_size = randint(2*els) b = Arrayterator(a, buf_size) # Check that each block has at most ``buf_size`` elements for block in b: assert_(len(block.flat) <= (buf_size or els)) # Check that all elements are iterated correctly assert_(list(b.flat) == list(a.flat)) # Slice arrayterator start = [randint(dim) for dim in shape] stop = [randint(dim)+1 for dim in shape] step = [randint(dim)+1 for dim in shape] slice_ = tuple(slice(*t) for t in zip(start, stop, step)) c = b[slice_] d = a[slice_] # Check that each block has at most ``buf_size`` elements for block in c: assert_(len(block.flat) <= (buf_size or els)) # Check that the arrayterator is sliced correctly assert_(np.all(c.__array__() == d)) # Check that all elements are iterated correctly assert_(list(c.flat) == list(d.flat))