def test_setxor1d(self):
     # Test setxor1d
     a = array([1, 2, 5, 7, -1], mask=[0, 0, 0, 0, 1])
     b = array([1, 2, 3, 4, 5, -1], mask=[0, 0, 0, 0, 0, 1])
     test = setxor1d(a, b)
     assert_equal(test, array([3, 4, 7]))
     #
     a = array([1, 2, 5, 7, -1], mask=[0, 0, 0, 0, 1])
     b = [1, 2, 3, 4, 5]
     test = setxor1d(a, b)
     assert_equal(test, array([3, 4, 7, -1], mask=[0, 0, 0, 1]))
     #
     a = array([1, 2, 3])
     b = array([6, 5, 4])
     test = setxor1d(a, b)
     assert_(isinstance(test, MaskedArray))
     assert_equal(test, [1, 2, 3, 4, 5, 6])
     #
     a = array([1, 8, 2, 3], mask=[0, 1, 0, 0])
     b = array([6, 5, 4, 8], mask=[0, 0, 0, 1])
     test = setxor1d(a, b)
     assert_(isinstance(test, MaskedArray))
     assert_equal(test, [1, 2, 3, 4, 5, 6])
     #
     assert_array_equal([], setxor1d([], []))
示例#2
0
 def test_setxor1d(self):
     # Test setxor1d
     a = array([1, 2, 5, 7, -1], mask=[0, 0, 0, 0, 1])
     b = array([1, 2, 3, 4, 5, -1], mask=[0, 0, 0, 0, 0, 1])
     test = setxor1d(a, b)
     assert_equal(test, array([3, 4, 7]))
     #
     a = array([1, 2, 5, 7, -1], mask=[0, 0, 0, 0, 1])
     b = [1, 2, 3, 4, 5]
     test = setxor1d(a, b)
     assert_equal(test, array([3, 4, 7, -1], mask=[0, 0, 0, 1]))
     #
     a = array([1, 2, 3])
     b = array([6, 5, 4])
     test = setxor1d(a, b)
     assert_(isinstance(test, MaskedArray))
     assert_equal(test, [1, 2, 3, 4, 5, 6])
     #
     a = array([1, 8, 2, 3], mask=[0, 1, 0, 0])
     b = array([6, 5, 4, 8], mask=[0, 0, 0, 1])
     test = setxor1d(a, b)
     assert_(isinstance(test, MaskedArray))
     assert_equal(test, [1, 2, 3, 4, 5, 6])
     #
     assert_array_equal([], setxor1d([], []))