diff options
author | Alan McIntyre <alan.mcintyre@local> | 2008-07-03 03:57:29 +0000 |
---|---|---|
committer | Alan McIntyre <alan.mcintyre@local> | 2008-07-03 03:57:29 +0000 |
commit | 590babe4646a3435f8a709d6230d05c10f085be1 (patch) | |
tree | 51e77ce9c79c6f58d4cadc7a819c5b00064cf503 /numpy/lib/tests/test_arraysetops.py | |
parent | 94bc330296e01d8fc0cb14aba1354ddb6d8587cc (diff) | |
download | numpy-590babe4646a3435f8a709d6230d05c10f085be1.tar.gz |
Remove uses of set_package_path, set_local_path, restore_path.
Clean up and (somewhat) standardize test module imports.
Removed unneeded reload calls.
Diffstat (limited to 'numpy/lib/tests/test_arraysetops.py')
-rw-r--r-- | numpy/lib/tests/test_arraysetops.py | 113 |
1 files changed, 42 insertions, 71 deletions
diff --git a/numpy/lib/tests/test_arraysetops.py b/numpy/lib/tests/test_arraysetops.py index 0a15aa8b0..bbfaeb1fa 100644 --- a/numpy/lib/tests/test_arraysetops.py +++ b/numpy/lib/tests/test_arraysetops.py @@ -3,83 +3,67 @@ """ from numpy.testing import * -set_package_path() -import numpy +import numpy as np from numpy.lib.arraysetops import * -restore_path() - -################################################## class TestAso(TestCase): - ## - # 03.11.2005, c def test_unique1d( self ): + a = np.array( [5, 7, 1, 2, 1, 5, 7] ) - a = numpy.array( [5, 7, 1, 2, 1, 5, 7] ) - - ec = numpy.array( [1, 2, 5, 7] ) + ec = np.array( [1, 2, 5, 7] ) c = unique1d( a ) assert_array_equal( c, ec ) assert_array_equal([], unique1d([])) - ## - # 03.11.2005, c def test_intersect1d( self ): + a = np.array( [5, 7, 1, 2] ) + b = np.array( [2, 4, 3, 1, 5] ) - a = numpy.array( [5, 7, 1, 2] ) - b = numpy.array( [2, 4, 3, 1, 5] ) - - ec = numpy.array( [1, 2, 5] ) + ec = np.array( [1, 2, 5] ) c = intersect1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], intersect1d([],[])) - ## - # 03.11.2005, c def test_intersect1d_nu( self ): + a = np.array( [5, 5, 7, 1, 2] ) + b = np.array( [2, 1, 4, 3, 3, 1, 5] ) - a = numpy.array( [5, 5, 7, 1, 2] ) - b = numpy.array( [2, 1, 4, 3, 3, 1, 5] ) - - ec = numpy.array( [1, 2, 5] ) + ec = np.array( [1, 2, 5] ) c = intersect1d_nu( a, b ) assert_array_equal( c, ec ) assert_array_equal([], intersect1d_nu([],[])) - ## - # 03.11.2005, c def test_setxor1d( self ): + a = np.array( [5, 7, 1, 2] ) + b = np.array( [2, 4, 3, 1, 5] ) - a = numpy.array( [5, 7, 1, 2] ) - b = numpy.array( [2, 4, 3, 1, 5] ) - - ec = numpy.array( [3, 4, 7] ) + ec = np.array( [3, 4, 7] ) c = setxor1d( a, b ) assert_array_equal( c, ec ) - a = numpy.array( [1, 2, 3] ) - b = numpy.array( [6, 5, 4] ) + a = np.array( [1, 2, 3] ) + b = np.array( [6, 5, 4] ) - ec = numpy.array( [1, 2, 3, 4, 5, 6] ) + ec = np.array( [1, 2, 3, 4, 5, 6] ) c = setxor1d( a, b ) assert_array_equal( c, ec ) - a = numpy.array( [1, 8, 2, 3] ) - b = numpy.array( [6, 5, 4, 8] ) + a = np.array( [1, 8, 2, 3] ) + b = np.array( [6, 5, 4, 8] ) - ec = numpy.array( [1, 2, 3, 4, 5, 6] ) + 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 = numpy.array([]) - one_elem = numpy.array([1]) - two_elem = numpy.array([1,2]) + 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)) @@ -88,81 +72,68 @@ class TestAso(TestCase): assert_array_equal([],ediff1d(one_elem)) assert_array_equal([1],ediff1d(two_elem)) - ## - # 03.11.2005, c def test_setmember1d( self ): + a = np.array( [5, 7, 1, 2] ) + b = np.array( [2, 4, 3, 1, 5] ) - a = numpy.array( [5, 7, 1, 2] ) - b = numpy.array( [2, 4, 3, 1, 5] ) - - ec = numpy.array( [True, False, True, True] ) + ec = np.array( [True, False, True, True] ) c = setmember1d( a, b ) assert_array_equal( c, ec ) a[0] = 8 - ec = numpy.array( [False, False, True, True] ) + ec = np.array( [False, False, True, True] ) c = setmember1d( a, b ) assert_array_equal( c, ec ) a[0], a[3] = 4, 8 - ec = numpy.array( [True, False, True, False] ) + ec = np.array( [True, False, True, False] ) c = setmember1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], setmember1d([],[])) - ## - # 03.11.2005, c def test_union1d( self ): + a = np.array( [5, 4, 7, 1, 2] ) + b = np.array( [2, 4, 3, 3, 2, 1, 5] ) - a = numpy.array( [5, 4, 7, 1, 2] ) - b = numpy.array( [2, 4, 3, 3, 2, 1, 5] ) - - ec = numpy.array( [1, 2, 3, 4, 5, 7] ) + ec = np.array( [1, 2, 3, 4, 5, 7] ) c = union1d( a, b ) assert_array_equal( c, ec ) assert_array_equal([], union1d([],[])) - ## - # 03.11.2005, c - # 09.01.2006 def test_setdiff1d( self ): + a = np.array( [6, 5, 4, 7, 1, 2] ) + b = np.array( [2, 4, 3, 3, 2, 1, 5] ) - a = numpy.array( [6, 5, 4, 7, 1, 2] ) - b = numpy.array( [2, 4, 3, 3, 2, 1, 5] ) - - ec = numpy.array( [6, 7] ) + ec = np.array( [6, 7] ) c = setdiff1d( a, b ) assert_array_equal( c, ec ) - a = numpy.arange( 21 ) - b = numpy.arange( 19 ) - ec = numpy.array( [19, 20] ) + 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 = numpy.array(['a','b','c']) - b = numpy.array(['a','b','s']) - assert_array_equal(setdiff1d(a,b),numpy.array(['c'])) + a = np.array(['a','b','c']) + b = np.array(['a','b','s']) + assert_array_equal(setdiff1d(a,b),np.array(['c'])) - ## - # 03.11.2005, c def test_manyways( self ): - nItem = 100 - a = numpy.fix( nItem / 10 * numpy.random.random( nItem ) ) - b = numpy.fix( nItem / 10 * numpy.random.random( nItem ) ) + a = np.fix( nItem / 10 * np.random.random( nItem ) ) + b = np.fix( nItem / 10 * np.random.random( nItem ) ) c1 = intersect1d_nu( a, b ) c2 = unique1d( intersect1d( a, b ) ) assert_array_equal( c1, c2 ) - a = numpy.array( [5, 7, 1, 2, 8] ) - b = numpy.array( [9, 8, 2, 4, 3, 1, 5] ) + 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 ) |