REDROOM

Navigation

Breadcrumbs

Create

Edit File Size: 1.42 KB

/opt/alt/python37/lib64/python3.7/site-packages/numpy/lib/tests/test_arrayterator.py

Text Base64

from __future__ import division, absolute_import, print_function

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))

if __name__ == '__main__':
    from numpy.testing import run_module_suite
    run_module_suite()

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
data DIR	-	drwxr-xr-x	2023-05-13 02:18:14	Edit Download recursive recursive
__pycache__ DIR	-	drwxr-xr-x	2023-05-13 02:18:14	Edit Download recursive recursive
test_arraypad.py	42.68 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_arraysetops.py	15.49 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_arrayterator.py	1.42 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_financial.py	6.62 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_format.py	33.49 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_function_base.py	126.81 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_index_tricks.py	13.10 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_io.py	75.59 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_mixins.py	6.70 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_nanfunctions.py	34.07 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_packbits.py	12.63 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_polynomial.py	7.03 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_recfunctions.py	30.54 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_regression.py	8.62 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_shape_base.py	19.02 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_stride_tricks.py	14.69 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_twodim_base.py	16.72 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_type_check.py	12.62 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_ufunclike.py	2.96 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test_utils.py	1.62 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test__datasource.py	10.16 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test__iotools.py	13.15 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive
test__version.py	2.08 KB	lrw-r--r--	2023-04-24 19:30:08	Edit Download recursive recursive

Directory Contents