FM / tests

/usr/lib64/python2.6/site-packages/numpy/lib/tests/

/usr/lib64/python2.6/site-packages/numpy/lib/tests

Name	Size	Mode	Actions
test_arraysetops.py	6993	0644	edit dl rm
test_arraysetops.pyc	7760	0644	edit dl rm
test_arraysetops.pyo	7760	0644	edit dl rm
test_arrayterator.py	1315	0644	edit dl rm
test_arrayterator.pyc	1964	0644	edit dl rm
test_arrayterator.pyo	1710	0644	edit dl rm
test_financial.py	1911	0644	edit dl rm
test_financial.pyc	3683	0644	edit dl rm
test_financial.pyo	3683	0644	edit dl rm
test_format.py	25312	0644	edit dl rm
test_format.pyc	25261	0644	edit dl rm
test_format.pyo	25236	0644	edit dl rm
test_function_base.py	31443	0644	edit dl rm
test_function_base.pyc	43457	0644	edit dl rm
test_function_base.pyo	42029	0644	edit dl rm
test_index_tricks.py	3928	0644	edit dl rm
test_index_tricks.pyc	7125	0644	edit dl rm
test_index_tricks.pyo	6529	0644	edit dl rm
test_io.py	44958	0644	edit dl rm
test_io.pyc	58173	0644	edit dl rm
test_io.pyo	57879	0644	edit dl rm
test_polynomial.py	3566	0644	edit dl rm
test_polynomial.pyc	5303	0644	edit dl rm
test_polynomial.pyo	4728	0644	edit dl rm
test_recfunctions.py	24942	0644	edit dl rm
test_recfunctions.pyc	29895	0644	edit dl rm
test_recfunctions.pyo	29813	0644	edit dl rm
test_regression.py	6357	0644	edit dl rm
test_regression.pyc	11069	0644	edit dl rm
test_regression.pyo	10974	0644	edit dl rm
test_shape_base.py	9945	0644	edit dl rm
test_shape_base.pyc	16188	0644	edit dl rm
test_shape_base.pyo	16059	0644	edit dl rm
test_stride_tricks.py	6958	0644	edit dl rm
test_stride_tricks.pyc	9183	0644	edit dl rm
test_stride_tricks.pyo	9144	0644	edit dl rm
test_twodim_base.py	10185	0644	edit dl rm
test_twodim_base.pyc	14005	0644	edit dl rm
test_twodim_base.pyo	14005	0644	edit dl rm
test_type_check.py	9579	0644	edit dl rm
test_type_check.pyc	18846	0644	edit dl rm
test_type_check.pyo	18044	0644	edit dl rm
test_ufunclike.py	1776	0644	edit dl rm
test_ufunclike.pyc	2129	0644	edit dl rm
test_ufunclike.pyo	2129	0644	edit dl rm
test_utils.py	910	0644	edit dl rm
test_utils.pyc	2211	0644	edit dl rm
test_utils.pyo	1955	0644	edit dl rm
test__datasource.py	9921	0644	edit dl rm
test__datasource.pyc	16260	0644	edit dl rm
test__datasource.pyo	15403	0644	edit dl rm
test__iotools.py	10899	0644	edit dl rm
test__iotools.pyc	12397	0644	edit dl rm
test__iotools.pyo	12305	0644	edit dl rm

Edit: /usr/lib64/python2.6/site-packages/numpy/lib/tests/test_arraysetops.py (6993B)

""" Test functions for 1D array set operations.

"""

from numpy.testing import *
import numpy as np
from numpy.lib.arraysetops import *

import warnings

class TestAso(TestCase):
    def test_unique( self ):
        a = np.array( [5, 7, 1, 2, 1, 5, 7] )

ec = np.array( [1, 2, 5, 7] )
        c = unique( a )
        assert_array_equal( c, ec )

warnings.simplefilter('ignore', Warning)
        vals, indices = unique( a, return_index=True )
        warnings.resetwarnings()

ed = np.array( [2, 3, 0, 1] )
        assert_array_equal(vals, ec)
        assert_array_equal(indices, ed)

warnings.simplefilter('ignore', Warning)
        vals, ind0, ind1 = unique( a, return_index=True,
                                     return_inverse=True )
        warnings.resetwarnings()

ee = np.array( [2, 3, 0, 1, 0, 2, 3] )
        assert_array_equal(vals, ec)
        assert_array_equal(ind0, ed)
        assert_array_equal(ind1, ee)

assert_array_equal([], unique([]))

def test_intersect1d( self ):
        # unique inputs
        a = np.array( [5, 7, 1, 2] )
        b = np.array( [2, 4, 3, 1, 5] )

ec = np.array( [1, 2, 5] )
        c = intersect1d( a, b, assume_unique=True )
        assert_array_equal( c, ec )

# non-unique inputs
        a = np.array( [5, 5, 7, 1, 2] )
        b = np.array( [2, 1, 4, 3, 3, 1, 5] )

ed = np.array( [1, 2, 5] )
        c = intersect1d( a, b )
        assert_array_equal( c, ed )

assert_array_equal([], intersect1d([],[]))

def test_intersect1d_nu( self ):
        # This should be removed when intersect1d_nu is removed.
        a = np.array( [5, 5, 7, 1, 2] )
        b = np.array( [2, 1, 4, 3, 3, 1, 5] )

ec = np.array( [1, 2, 5] )
        warnings.simplefilter('ignore', Warning)
        c = intersect1d_nu( a, b )
        warnings.resetwarnings()
        assert_array_equal( c, ec )

assert_array_equal([], intersect1d_nu([],[]))

def test_setxor1d( self ):
        a = np.array( [5, 7, 1, 2] )
        b = np.array( [2, 4, 3, 1, 5] )

ec = np.array( [3, 4, 7] )
        c = setxor1d( a, b )
        assert_array_equal( c, ec )

a = np.array( [1, 2, 3] )
        b = np.array( [6, 5, 4] )

ec = np.array( [1, 2, 3, 4, 5, 6] )
        c = setxor1d( a, b )
        assert_array_equal( c, ec )

a = np.array( [1, 8, 2, 3] )
        b = np.array( [6, 5, 4, 8] )

ec = np.array( [1, 2, 3, 4, 5, 6] )
        c = setxor1d( a, b )
        assert_array_equal( c, ec )

assert_array_equal([], setxor1d([],[]))

def test_ediff1d(self):
        zero_elem = np.array([])
        one_elem = np.array([1])
        two_elem = np.array([1,2])

assert_array_equal([],ediff1d(zero_elem))
        assert_array_equal([0],ediff1d(zero_elem,to_begin=0))
        assert_array_equal([0],ediff1d(zero_elem,to_end=0))
        assert_array_equal([-1,0],ediff1d(zero_elem,to_begin=-1,to_end=0))
        assert_array_equal([],ediff1d(one_elem))
        assert_array_equal([1],ediff1d(two_elem))

def test_setmember1d( self ):
        # This should be removed when setmember1d is removed.
        a = np.array( [5, 7, 1, 2] )
        b = np.array( [2, 4, 3, 1, 5] )

ec = np.array( [True, False, True, True] )
        warnings.simplefilter('ignore', Warning)
        c = setmember1d( a, b )
        warnings.resetwarnings()
        assert_array_equal( c, ec )

a[0] = 8
        ec = np.array( [False, False, True, True] )
        c = setmember1d( a, b )
        assert_array_equal( c, ec )

a[0], a[3] = 4, 8
        ec = np.array( [True, False, True, False] )
        c = setmember1d( a, b )
        assert_array_equal( c, ec )

assert_array_equal([], setmember1d([],[]))

def test_in1d(self):
        a = np.array( [5, 7, 1, 2] )
        b = np.array( [2, 4, 3, 1, 5] )

ec = np.array( [True, False, True, True] )
        c = in1d( a, b, assume_unique=True )
        assert_array_equal( c, ec )

a[0] = 8
        ec = np.array( [False, False, True, True] )
        c = in1d( a, b, assume_unique=True )
        assert_array_equal( c, ec )

a[0], a[3] = 4, 8
        ec = np.array( [True, False, True, False] )
        c = in1d( a, b, assume_unique=True )
        assert_array_equal( c, ec )
        
        a = np.array([5,4,5,3,4,4,3,4,3,5,2,1,5,5])
        b = [2,3,4]

ec = [False, True, False, True, True, True, True, True, True, False,
              True, False, False, False]
        c = in1d(a, b)
        assert_array_equal(c, ec)

b = b + [5, 5, 4]

ec = [True, True, True, True, True, True, True, True, True, True,
              True, False, True, True]
        c = in1d(a, b)
        assert_array_equal(c, ec)

a = np.array([5, 7, 1, 2])
        b = np.array([2, 4, 3, 1, 5])

ec = np.array([True, False, True, True])
        c = in1d(a, b)
        assert_array_equal(c, ec)

a = np.array([5, 7, 1, 1, 2])
        b = np.array([2, 4, 3, 3, 1, 5])

ec = np.array([True, False, True, True, True])
        c = in1d(a, b)
        assert_array_equal(c, ec)

a = np.array([5])
        b = np.array([2])

ec = np.array([False])
        c = in1d(a, b)
        assert_array_equal(c, ec)

a = np.array([5, 5])
        b = np.array([2, 2])

ec = np.array([False, False])
        c = in1d(a, b)
        assert_array_equal(c, ec)

assert_array_equal(in1d([], []), [])

def test_in1d_char_array( self ):
        a = np.array(['a', 'b', 'c','d','e','c','e','b'])
        b = np.array(['a','c'])

ec = np.array([True, False, True, False, False, True, False, False])
        c = in1d(a, b)

assert_array_equal(c, ec)

def test_union1d( self ):
        a = np.array( [5, 4, 7, 1, 2] )
        b = np.array( [2, 4, 3, 3, 2, 1, 5] )

ec = np.array( [1, 2, 3, 4, 5, 7] )
        c = union1d( a, b )
        assert_array_equal( c, ec )

assert_array_equal([], union1d([],[]))

def test_setdiff1d( self ):
        a = np.array( [6, 5, 4, 7, 1, 2, 7, 4] )
        b = np.array( [2, 4, 3, 3, 2, 1, 5] )

ec = np.array( [6, 7] )
        c = setdiff1d( a, b )
        assert_array_equal( c, ec )

a = np.arange( 21 )
        b = np.arange( 19 )
        ec = np.array( [19, 20] )
        c = setdiff1d( a, b )
        assert_array_equal( c, ec )

assert_array_equal([], setdiff1d([],[]))

def test_setdiff1d_char_array(self):
        a = np.array(['a','b','c'])
        b = np.array(['a','b','s'])
        assert_array_equal(setdiff1d(a,b),np.array(['c']))

def test_manyways( self ):
        a = np.array( [5, 7, 1, 2, 8] )
        b = np.array( [9, 8, 2, 4, 3, 1, 5] )

c1 = setxor1d( a, b )
        aux1 = intersect1d( a, b )
        aux2 = union1d( a, b )
        c2 = setdiff1d( aux2, aux1 )
        assert_array_equal( c1, c2 )

if __name__ == "__main__":
    run_module_suite()

""" Test functions for 1D array set operations. """ from numpy.testing import * import numpy as np from numpy.lib.arraysetops import * import warnings class TestAso(TestCase): def test_unique( self ): a = np.array( [5, 7, 1, 2, 1, 5, 7] ) ec = np.array( [1, 2, 5, 7] ) c = unique( a ) assert_array_equal( c, ec ) warnings.simplefilter('ignore', Warning) vals, indices = unique( a, return_index=True ) warnings.resetwarnings() ed = np.array( [2, 3, 0, 1] ) assert_array_equal(vals, ec) assert_array_equal(indices, ed) warnings.simplefilter('ignore', Warning) vals, ind0, ind1 = unique( a, return_index=True, return_inverse=True ) warnings.resetwarnings() ee = np.array( [2, 3, 0, 1, 0, 2, 3] ) assert_array_equal(vals, ec) assert_array_equal(ind0, ed) assert_array_equal(ind1, ee) assert_array_equal([], unique([])) def test_intersect1d( self ): # unique inputs a = np.array( [5, 7, 1, 2] ) b = np.array( [2, 4, 3, 1, 5] ) ec = np.array( [1, 2, 5] ) c = intersect1d( a, b, assume_unique=True ) assert_array_equal( c, ec ) # non-unique inputs a = np.array( [5, 5, 7, 1, 2] ) b = np.array( [2, 1, 4, 3, 3, 1, 5] ) ed = np.array( [1, 2, 5] ) c = intersect1d( a, b ) assert_array_equal( c, ed ) assert_array_equal([], intersect1d([],[])) def test_intersect1d_nu( self ): # This should be removed when intersect1d_nu is removed. a = np.array( [5, 5, 7, 1, 2] ) b = np.array( [2, 1, 4, 3, 3, 1, 5] ) ec = np.array( [1, 2, 5] ) warnings.simplefilter('ignore', Warning) c = intersect1d_nu( a, b ) warnings.resetwarnings() assert_array_equal( c, ec ) assert_array_equal([], intersect1d_nu([],[])) def test_setxor1d( self ): a = np.array( [5, 7, 1, 2] ) b = np.array( [2, 4, 3, 1, 5] ) ec = np.array( [3, 4, 7] ) c = setxor1d( a, b ) assert_array_equal( c, ec ) a = np.array( [1, 2, 3] ) b = np.array( [6, 5, 4] ) ec = np.array( [1, 2, 3, 4, 5, 6] ) c = setxor1d( a, b ) assert_array_equal( c, ec ) a = np.array( [1, 8, 2, 3] ) b = np.array( [6, 5, 4, 8] ) ec = np.array( [1, 2, 3, 4, 5, 6] ) c = setxor1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], setxor1d([],[])) def test_ediff1d(self): zero_elem = np.array([]) one_elem = np.array([1]) two_elem = np.array([1,2]) assert_array_equal([],ediff1d(zero_elem)) assert_array_equal([0],ediff1d(zero_elem,to_begin=0)) assert_array_equal([0],ediff1d(zero_elem,to_end=0)) assert_array_equal([-1,0],ediff1d(zero_elem,to_begin=-1,to_end=0)) assert_array_equal([],ediff1d(one_elem)) assert_array_equal([1],ediff1d(two_elem)) def test_setmember1d( self ): # This should be removed when setmember1d is removed. a = np.array( [5, 7, 1, 2] ) b = np.array( [2, 4, 3, 1, 5] ) ec = np.array( [True, False, True, True] ) warnings.simplefilter('ignore', Warning) c = setmember1d( a, b ) warnings.resetwarnings() assert_array_equal( c, ec ) a[0] = 8 ec = np.array( [False, False, True, True] ) c = setmember1d( a, b ) assert_array_equal( c, ec ) a[0], a[3] = 4, 8 ec = np.array( [True, False, True, False] ) c = setmember1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], setmember1d([],[])) def test_in1d(self): a = np.array( [5, 7, 1, 2] ) b = np.array( [2, 4, 3, 1, 5] ) ec = np.array( [True, False, True, True] ) c = in1d( a, b, assume_unique=True ) assert_array_equal( c, ec ) a[0] = 8 ec = np.array( [False, False, True, True] ) c = in1d( a, b, assume_unique=True ) assert_array_equal( c, ec ) a[0], a[3] = 4, 8 ec = np.array( [True, False, True, False] ) c = in1d( a, b, assume_unique=True ) assert_array_equal( c, ec ) a = np.array([5,4,5,3,4,4,3,4,3,5,2,1,5,5]) b = [2,3,4] ec = [False, True, False, True, True, True, True, True, True, False, True, False, False, False] c = in1d(a, b) assert_array_equal(c, ec) b = b + [5, 5, 4] ec = [True, True, True, True, True, True, True, True, True, True, True, False, True, True] c = in1d(a, b) assert_array_equal(c, ec) a = np.array([5, 7, 1, 2]) b = np.array([2, 4, 3, 1, 5]) ec = np.array([True, False, True, True]) c = in1d(a, b) assert_array_equal(c, ec) a = np.array([5, 7, 1, 1, 2]) b = np.array([2, 4, 3, 3, 1, 5]) ec = np.array([True, False, True, True, True]) c = in1d(a, b) assert_array_equal(c, ec) a = np.array([5]) b = np.array([2]) ec = np.array([False]) c = in1d(a, b) assert_array_equal(c, ec) a = np.array([5, 5]) b = np.array([2, 2]) ec = np.array([False, False]) c = in1d(a, b) assert_array_equal(c, ec) assert_array_equal(in1d([], []), []) def test_in1d_char_array( self ): a = np.array(['a', 'b', 'c','d','e','c','e','b']) b = np.array(['a','c']) ec = np.array([True, False, True, False, False, True, False, False]) c = in1d(a, b) assert_array_equal(c, ec) def test_union1d( self ): a = np.array( [5, 4, 7, 1, 2] ) b = np.array( [2, 4, 3, 3, 2, 1, 5] ) ec = np.array( [1, 2, 3, 4, 5, 7] ) c = union1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], union1d([],[])) def test_setdiff1d( self ): a = np.array( [6, 5, 4, 7, 1, 2, 7, 4] ) b = np.array( [2, 4, 3, 3, 2, 1, 5] ) ec = np.array( [6, 7] ) c = setdiff1d( a, b ) assert_array_equal( c, ec ) a = np.arange( 21 ) b = np.arange( 19 ) ec = np.array( [19, 20] ) c = setdiff1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], setdiff1d([],[])) def test_setdiff1d_char_array(self): a = np.array(['a','b','c']) b = np.array(['a','b','s']) assert_array_equal(setdiff1d(a,b),np.array(['c'])) def test_manyways( self ): a = np.array( [5, 7, 1, 2, 8] ) b = np.array( [9, 8, 2, 4, 3, 1, 5] ) c1 = setxor1d( a, b ) aux1 = intersect1d( a, b ) aux2 = union1d( a, b ) c2 = setdiff1d( aux2, aux1 ) assert_array_equal( c1, c2 ) if __name__ == "__main__": run_module_suite()