diff options
| author | MattHarrigan <harrigan.matthew@gmail.com> | 2016-10-19 19:57:43 -0400 |
|---|---|---|
| committer | MattHarrigan <harrigan.matthew@gmail.com> | 2016-10-21 15:16:10 -0400 |
| commit | 484827189756d45e6231bfd2cd3cf6db9c2993df (patch) | |
| tree | cd95769ca3e379ccc171b909936d35ccedf8ff65 | |
| parent | 16a98c0efe9e09258946f8469f7b556c568f1e21 (diff) | |
| download | python-numpy-484827189756d45e6231bfd2cd3cf6db9c2993df.tar.gz python-numpy-484827189756d45e6231bfd2cd3cf6db9c2993df.tar.bz2 python-numpy-484827189756d45e6231bfd2cd3cf6db9c2993df.zip | |
TST: Added cases for better coverage of ediff1d
| -rw-r--r-- | numpy/lib/tests/test_arraysetops.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/numpy/lib/tests/test_arraysetops.py b/numpy/lib/tests/test_arraysetops.py index 852183ffe..b75a2b060 100644 --- a/numpy/lib/tests/test_arraysetops.py +++ b/numpy/lib/tests/test_arraysetops.py @@ -169,6 +169,12 @@ class TestSetOps(TestCase): 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)) + assert_array_equal([7,1,9], ediff1d(two_elem, to_begin=7, to_end=9)) + assert_array_equal([5,6,1,7,8], ediff1d(two_elem, to_begin=[5,6], to_end=[7,8])) + assert_array_equal([1,9], ediff1d(two_elem, to_end=9)) + assert_array_equal([1,7,8], ediff1d(two_elem, to_end=[7,8])) + assert_array_equal([7,1], ediff1d(two_elem, to_begin=7)) + assert_array_equal([5,6,1], ediff1d(two_elem, to_begin=[5,6])) def test_in1d(self): # we use two different sizes for the b array here to test the |