diff options
| author | Stephan Hoyer <shoyer@gmail.com> | 2016-10-24 11:12:34 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-10-24 11:12:34 -0400 |
| commit | 3bd79ab568856678139fa4d7ca272e7cbd05e1ad (patch) | |
| tree | 6ad12614b5222ef9f22b5785c4fda1c8d080c8b1 /numpy/lib/tests | |
| parent | e7d6f36de23061fe3b1f3c1aabe273fc4999acc4 (diff) | |
| parent | 484827189756d45e6231bfd2cd3cf6db9c2993df (diff) | |
| download | numpy-3bd79ab568856678139fa4d7ca272e7cbd05e1ad.tar.gz | |
Merge pull request #8183 from mattharrigan/ediff1d-performance
Ediff1d performance
Diffstat (limited to 'numpy/lib/tests')
| -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 |