Gecko [ goons4good.org ]

Name	Size	Permission	Date
data	[ DIR ]	drwxr-xr-x	2021-09-16 10:54
test_api.py	6.56 KB	-rw-r--r--	2013-04-07 01:04
test_api.pyc	5.67 KB	-rw-r--r--	2018-04-10 19:40
test_api.pyo	5.67 KB	-rw-r--r--	2018-04-10 19:40
test_arrayprint.py	6.6 KB	-rw-r--r--	2013-04-07 01:04
test_arrayprint.pyc	9.75 KB	-rw-r--r--	2018-04-10 19:40
test_arrayprint.pyo	9.75 KB	-rw-r--r--	2018-04-10 19:40
test_blasdot.py	2.54 KB	-rw-r--r--	2013-04-07 01:04
test_blasdot.pyc	3.89 KB	-rw-r--r--	2018-04-10 19:40
test_blasdot.pyo	3.89 KB	-rw-r--r--	2018-04-10 19:40
test_datetime.py	82 KB	-rw-r--r--	2013-04-07 01:04
test_datetime.pyc	54.52 KB	-rw-r--r--	2018-04-10 19:40
test_datetime.pyo	54.39 KB	-rw-r--r--	2018-04-10 19:40
test_defchararray.py	25.06 KB	-rw-r--r--	2013-04-07 01:04
test_defchararray.pyc	36.88 KB	-rw-r--r--	2018-04-10 19:40
test_defchararray.pyo	36.88 KB	-rw-r--r--	2018-04-10 19:40
test_dtype.py	19.82 KB	-rw-r--r--	2013-04-07 01:04
test_dtype.pyc	20.52 KB	-rw-r--r--	2018-04-10 19:40
test_dtype.pyo	20.52 KB	-rw-r--r--	2018-04-10 19:40
test_einsum.py	19.46 KB	-rw-r--r--	2013-04-07 01:04
test_einsum.pyc	16.12 KB	-rw-r--r--	2018-04-10 19:40
test_einsum.pyo	16.12 KB	-rw-r--r--	2018-04-10 19:40
test_errstate.py	1.83 KB	-rw-r--r--	2013-04-07 01:04
test_errstate.pyc	1.76 KB	-rw-r--r--	2018-04-10 19:40
test_errstate.pyo	1.76 KB	-rw-r--r--	2018-04-10 19:40
test_function_base.py	994 B	-rw-r--r--	2013-04-07 01:04
test_function_base.pyc	2.23 KB	-rw-r--r--	2018-04-10 19:40
test_function_base.pyo	2.23 KB	-rw-r--r--	2018-04-10 19:40
test_getlimits.py	2.64 KB	-rw-r--r--	2013-04-07 01:04
test_getlimits.pyc	5.19 KB	-rw-r--r--	2018-04-10 19:40
test_getlimits.pyo	5.19 KB	-rw-r--r--	2018-04-10 19:40
test_half.py	17.95 KB	-rw-r--r--	2013-04-07 01:04
test_half.pyc	17.75 KB	-rw-r--r--	2018-04-10 19:40
test_half.pyo	17.75 KB	-rw-r--r--	2018-04-10 19:40
test_indexerrors.py	4.75 KB	-rw-r--r--	2013-04-07 01:04
test_indexerrors.pyc	11.02 KB	-rw-r--r--	2018-04-10 19:40
test_indexerrors.pyo	11.02 KB	-rw-r--r--	2018-04-10 19:40
test_indexing.py	800 B	-rw-r--r--	2013-04-07 01:04
test_indexing.pyc	786 B	-rw-r--r--	2018-04-10 19:40
test_indexing.pyo	786 B	-rw-r--r--	2018-04-10 19:40
test_item_selection.py	2.46 KB	-rw-r--r--	2013-04-07 01:04
test_item_selection.pyc	2.57 KB	-rw-r--r--	2018-04-10 19:40
test_item_selection.pyo	2.57 KB	-rw-r--r--	2018-04-10 19:40
test_machar.py	1000 B	-rw-r--r--	2013-04-07 01:04
test_machar.pyc	1.66 KB	-rw-r--r--	2018-04-10 19:40
test_machar.pyo	1.66 KB	-rw-r--r--	2018-04-10 19:40
test_memmap.py	3.81 KB	-rw-r--r--	2018-04-10 19:39
test_memmap.pyc	5.5 KB	-rw-r--r--	2018-04-10 19:40
test_memmap.pyo	5.28 KB	-rw-r--r--	2018-04-10 19:40
test_multiarray.py	107.04 KB	-rw-r--r--	2018-04-10 19:39
test_multiarray.pyc	125.9 KB	-rw-r--r--	2018-04-10 19:40
test_multiarray.pyo	125.53 KB	-rw-r--r--	2018-04-10 19:40
test_multiarray_assignment.py	1.81 KB	-rw-r--r--	2013-04-07 01:04
test_multiarray_assignment.pyc	2.65 KB	-rw-r--r--	2018-04-10 19:40
test_multiarray_assignment.pyo	2.55 KB	-rw-r--r--	2018-04-10 19:40
test_nditer.py	93.58 KB	-rw-r--r--	2013-04-07 01:04
test_nditer.pyc	79.07 KB	-rw-r--r--	2018-04-10 19:40
test_nditer.pyo	79.07 KB	-rw-r--r--	2018-04-10 19:40
test_numeric.py	52.63 KB	-rw-r--r--	2013-04-07 01:04
test_numeric.pyc	66.87 KB	-rw-r--r--	2018-04-10 19:40
test_numeric.pyo	66.87 KB	-rw-r--r--	2018-04-10 19:40
test_numerictypes.py	14.2 KB	-rw-r--r--	2013-04-07 01:04
test_numerictypes.pyc	17.63 KB	-rw-r--r--	2018-04-10 19:40
test_numerictypes.pyo	17.63 KB	-rw-r--r--	2018-04-10 19:40
test_print.py	9.3 KB	-rw-r--r--	2013-04-07 01:04
test_print.pyc	10.51 KB	-rw-r--r--	2018-04-10 19:40
test_print.pyo	10.51 KB	-rw-r--r--	2018-04-10 19:40
test_records.py	5.63 KB	-rw-r--r--	2013-04-07 01:04
test_records.pyc	9.4 KB	-rw-r--r--	2018-04-10 19:40
test_records.pyo	9.4 KB	-rw-r--r--	2018-04-10 19:40
test_regression.py	62.21 KB	-rw-r--r--	2013-04-07 01:04
test_regression.pyc	90.43 KB	-rw-r--r--	2018-04-10 19:40
test_regression.pyo	90.43 KB	-rw-r--r--	2018-04-10 19:40
test_scalarmath.py	5.53 KB	-rw-r--r--	2013-04-07 01:04
test_scalarmath.pyc	6.78 KB	-rw-r--r--	2018-04-10 19:40
test_scalarmath.pyo	6.78 KB	-rw-r--r--	2018-04-10 19:40
test_shape_base.py	7.07 KB	-rw-r--r--	2013-04-07 01:04
test_shape_base.pyc	10.66 KB	-rw-r--r--	2018-04-10 19:40
test_shape_base.pyo	10.66 KB	-rw-r--r--	2018-04-10 19:40
test_ufunc.py	30.64 KB	-rw-r--r--	2013-04-07 01:04
test_ufunc.pyc	31.48 KB	-rw-r--r--	2018-04-10 19:40
test_ufunc.pyo	31.34 KB	-rw-r--r--	2018-04-10 19:40
test_umath.py	44.02 KB	-rw-r--r--	2013-04-07 01:04
test_umath.pyc	62.8 KB	-rw-r--r--	2018-04-10 19:40
test_umath.pyo	62.73 KB	-rw-r--r--	2018-04-10 19:40
test_umath_complex.py	20.05 KB	-rw-r--r--	2013-04-07 01:04
test_umath_complex.pyc	18.83 KB	-rw-r--r--	2018-04-10 19:40
test_umath_complex.pyo	18.83 KB	-rw-r--r--	2018-04-10 19:40
test_unicode.py	12.18 KB	-rw-r--r--	2013-04-07 01:04
test_unicode.pyc	15.84 KB	-rw-r--r--	2018-04-10 19:40
test_unicode.pyo	15.84 KB	-rw-r--r--	2018-04-10 19:40

Rename

import numpy as np
from numpy.testing import *
import sys, warnings

class TestTake(TestCase):
    def test_simple(self):
        a = np.array([['1', '2'],['3', '4']])
        modes = ['raise', 'wrap', 'clip']
        indices = [-1, 4]
        index_arrays = [np.empty(0, dtype=np.intp),
                        np.empty((1,1), dtype=np.intp)]
        real_indices = {}
        real_indices['raise'] = {-1:1, 4:IndexError}
        real_indices['wrap'] = {-1:1, 4:0}
        real_indices['clip'] = {-1:0, 4:1}
        # Currently all types but object, use the same function generation.
        # So it should not be necessary to test all. However test also a non
        # refcounted struct on top of object.
        types = np.int, np.object, np.dtype([('', 'i', 2)])
        for t in types:
            ta = np.array(a, dtype=t)
            tresult = list(ta.T.copy())
            for index_array in index_arrays:
                if index_array.size != 0:
                    tresult[0].shape = (2,) + index_array.shape
                    tresult[1].shape = (2,) + index_array.shape
                for mode in modes:
                    for index in indices:
                        real_index = real_indices[mode][index]
                        if real_index is IndexError and index_array.size != 0:
                            index_array.put(0, index)
                            assert_raises(IndexError, ta.take, index_array,
                                          mode=mode, axis=1)
                        elif index_array.size != 0:
                            index_array.put(0, index)
                            res = ta.take(index_array, mode=mode, axis=1)
                            assert_array_equal(res, tresult[real_index])
                        else:
                            res = ta.take(index_array, mode=mode, axis=1)
                            assert_(res.shape == (2,) + index_array.shape)

def test_refcounting(self):
        objects = [object() for i in xrange(10)]
        for mode in ('raise', 'clip', 'wrap'):
            a = np.array(objects)
            b = np.array([2, 2, 4, 5, 3, 5])
            a.take(b, out=a[:6])
            del a
            assert_(np.all([sys.getrefcount(o) == 3 for o in objects]))
            # not contiguous, example:
            a = np.array(objects * 2)[::2]
            a.take(b, out=a[:6])
            del a
            assert_(np.all([sys.getrefcount(o) == 3 for o in objects]))

if __name__ == "__main__":
    run_module_suite()