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

from os import path
import numpy as np
from numpy.testing import *
from numpy.compat import asbytes, asunicode

import warnings

class TestFromrecords(TestCase):
    def test_fromrecords(self):
        r = np.rec.fromrecords([[456, 'dbe', 1.2], [2, 'de', 1.3]],
                            names='col1,col2,col3')
        assert_equal(r[0].item(), (456, 'dbe', 1.2))

def test_method_array(self):
        r = np.rec.array(asbytes('abcdefg') * 100, formats='i2,a3,i4', shape=3, byteorder='big')
        assert_equal(r[1].item(), (25444, asbytes('efg'), 1633837924))

def test_method_array2(self):
        r = np.rec.array([(1, 11, 'a'), (2, 22, 'b'), (3, 33, 'c'), (4, 44, 'd'), (5, 55, 'ex'),
                     (6, 66, 'f'), (7, 77, 'g')], formats='u1,f4,a1')
        assert_equal(r[1].item(), (2, 22.0, asbytes('b')))

def test_recarray_slices(self):
        r = np.rec.array([(1, 11, 'a'), (2, 22, 'b'), (3, 33, 'c'), (4, 44, 'd'), (5, 55, 'ex'),
                     (6, 66, 'f'), (7, 77, 'g')], formats='u1,f4,a1')
        assert_equal(r[1::2][1].item(), (4, 44.0, asbytes('d')))

def test_recarray_fromarrays(self):
        x1 = np.array([1, 2, 3, 4])
        x2 = np.array(['a', 'dd', 'xyz', '12'])
        x3 = np.array([1.1, 2, 3, 4])
        r = np.rec.fromarrays([x1, x2, x3], names='a,b,c')
        assert_equal(r[1].item(), (2, 'dd', 2.0))
        x1[1] = 34
        assert_equal(r.a, np.array([1, 2, 3, 4]))

def test_recarray_fromfile(self):
        data_dir = path.join(path.dirname(__file__), 'data')
        filename = path.join(data_dir, 'recarray_from_file.fits')
        fd = open(filename, 'rb')
        fd.seek(2880 * 2)
        r = np.rec.fromfile(fd, formats='f8,i4,a5', shape=3, byteorder='big')
        fd.seek(2880 * 2)
        r = np.rec.array(fd, formats='f8,i4,a5', shape=3, byteorder='big')
        fd.close()

def test_recarray_from_obj(self):
        count = 10
        a = np.zeros(count, dtype='O')
        b = np.zeros(count, dtype='f8')
        c = np.zeros(count, dtype='f8')
        for i in range(len(a)):
            a[i] = range(1, 10)

mine = np.rec.fromarrays([a, b, c], names='date,data1,data2')
        for i in range(len(a)):
            assert_((mine.date[i] == range(1, 10)))
            assert_((mine.data1[i] == 0.0))
            assert_((mine.data2[i] == 0.0))

def test_recarray_from_repr(self):
        x = np.rec.array([ (1, 2)], dtype=[('a', np.int8), ('b', np.int8)])
        y = eval("np." + repr(x))
        assert_(isinstance(y, np.recarray))
        assert_equal(y, x)

def test_recarray_from_names(self):
        ra = np.rec.array([
            (1, 'abc', 3.7000002861022949, 0),
            (2, 'xy', 6.6999998092651367, 1),
            (0, ' ', 0.40000000596046448, 0)],
                       names='c1, c2, c3, c4')
        pa = np.rec.fromrecords([
            (1, 'abc', 3.7000002861022949, 0),
            (2, 'xy', 6.6999998092651367, 1),
            (0, ' ', 0.40000000596046448, 0)],
                       names='c1, c2, c3, c4')
        assert_(ra.dtype == pa.dtype)
        assert_(ra.shape == pa.shape)
        for k in xrange(len(ra)):
            assert_(ra[k].item() == pa[k].item())

def test_recarray_conflict_fields(self):
        ra = np.rec.array([(1, 'abc', 2.3), (2, 'xyz', 4.2),
                        (3, 'wrs', 1.3)],
                       names='field, shape, mean')
        ra.mean = [1.1, 2.2, 3.3]
        assert_array_almost_equal(ra['mean'], [1.1, 2.2, 3.3])
        assert_(type(ra.mean) is type(ra.var))
        ra.shape = (1, 3)
        assert_(ra.shape == (1, 3))
        ra.shape = ['A', 'B', 'C']
        assert_array_equal(ra['shape'], [['A', 'B', 'C']])
        ra.field = 5
        assert_array_equal(ra['field'], [[5, 5, 5]])
        assert_(callable(ra.field))

def test_fromrecords_with_explicit_dtype(self):
        a = np.rec.fromrecords([(1, 'a'), (2, 'bbb')],
                                dtype=[('a', int), ('b', np.object)])
        assert_equal(a.a, [1, 2])
        assert_equal(a[0].a, 1)
        assert_equal(a.b, ['a', 'bbb'])
        assert_equal(a[-1].b, 'bbb')
        #
        ndtype = np.dtype([('a', int), ('b', np.object)])
        a = np.rec.fromrecords([(1, 'a'), (2, 'bbb')], dtype=ndtype)
        assert_equal(a.a, [1, 2])
        assert_equal(a[0].a, 1)
        assert_equal(a.b, ['a', 'bbb'])
        assert_equal(a[-1].b, 'bbb')

class TestRecord(TestCase):
    def setUp(self):
        self.data = np.rec.fromrecords([(1, 2, 3), (4, 5, 6)],
                            dtype=[("col1", "<i4"),
                                   ("col2", "<i4"),
                                   ("col3", "<i4")])

def test_assignment1(self):
        a = self.data
        assert_equal(a.col1[0], 1)
        a[0].col1 = 0
        assert_equal(a.col1[0], 0)

def test_assignment2(self):
        a = self.data
        assert_equal(a.col1[0], 1)
        a.col1[0] = 0
        assert_equal(a.col1[0], 0)

def test_invalid_assignment(self):
        a = self.data
        def assign_invalid_column(x):
            x[0].col5 = 1
        self.assertRaises(AttributeError, assign_invalid_column, a)

def test_out_of_order_fields(self):
        """Ticket #1431."""
        x = self.data[['col1', 'col2']]
        y = self.data[['col2', 'col1']]
        assert_equal(x[0][0], y[0][1])

def test_find_duplicate():
    l1 = [1, 2, 3, 4, 5, 6]
    assert_(np.rec.find_duplicate(l1) == [])

l2 = [1, 2, 1, 4, 5, 6]
    assert_(np.rec.find_duplicate(l2) == [1])

l3 = [1, 2, 1, 4, 1, 6, 2, 3]
    assert_(np.rec.find_duplicate(l3) == [1, 2])

l3 = [2, 2, 1, 4, 1, 6, 2, 3]
    assert_(np.rec.find_duplicate(l3) == [2, 1])

if __name__ == "__main__":
    run_module_suite()