diff options
| author | MilesCranmer <miles.cranmer@gmail.com> | 2022-06-09 20:48:45 -0400 |
|---|---|---|
| committer | MilesCranmer <miles.cranmer@gmail.com> | 2022-06-09 20:48:45 -0400 |
| commit | 179d1575ebdf850add391db87da6b1f4d3209e3c (patch) | |
| tree | 83c1bc29e64a5986891df3f00e1d2907d80c1abe /numpy/lib/tests/test_arraysetops.py | |
| parent | afbcdf27cd04128b1f297a9503a4bdfbccbddd28 (diff) | |
| download | numpy-179d1575ebdf850add391db87da6b1f4d3209e3c.tar.gz | |
MAINT: Fix linting errors in in1d tests
Diffstat (limited to 'numpy/lib/tests/test_arraysetops.py')
| -rw-r--r-- | numpy/lib/tests/test_arraysetops.py | 26 |
1 files changed, 18 insertions, 8 deletions
diff --git a/numpy/lib/tests/test_arraysetops.py b/numpy/lib/tests/test_arraysetops.py index 8de1cee6f..60d9b7c87 100644 --- a/numpy/lib/tests/test_arraysetops.py +++ b/numpy/lib/tests/test_arraysetops.py @@ -435,10 +435,16 @@ class TestSetOps: assert_array_equal(in1d(a, b, assume_unique=False), ec) assert_array_equal(in1d(a, long_b, assume_unique=True), ec) assert_array_equal(in1d(a, long_b, assume_unique=False), ec) - assert_array_equal(in1d(a, b, assume_unique=True, _slow_integer=True), ec) - assert_array_equal(in1d(a, b, assume_unique=False, _slow_integer=True), ec) - assert_array_equal(in1d(a, long_b, assume_unique=True, _slow_integer=True), ec) - assert_array_equal(in1d(a, long_b, assume_unique=False, _slow_integer=True), ec) + assert_array_equal(in1d(a, b, assume_unique=True, _slow_integer=True), + ec) + assert_array_equal(in1d(a, b, assume_unique=False, _slow_integer=True), + ec) + assert_array_equal(in1d(a, long_b, assume_unique=True, + _slow_integer=True), + ec) + assert_array_equal(in1d(a, long_b, assume_unique=False, + _slow_integer=True), + ec) def test_in1d_hit_alternate_algorithm(self): """Hit the standard isin code with integers""" @@ -461,10 +467,14 @@ class TestSetOps: a = np.array([True, False]) b = np.array([False, False, False]) expected = np.array([False, True]) - assert_array_equal(expected, in1d(a, b)) - assert_array_equal(expected, in1d(a, b, _slow_integer=True)) - assert_array_equal(np.invert(expected), in1d(a, b, invert=True)) - assert_array_equal(np.invert(expected), in1d(a, b, invert=True, _slow_integer=True)) + assert_array_equal(expected, + in1d(a, b)) + assert_array_equal(expected, + in1d(a, b, _slow_integer=True)) + assert_array_equal(np.invert(expected), + in1d(a, b, invert=True)) + assert_array_equal(np.invert(expected), + in1d(a, b, invert=True, _slow_integer=True)) def test_in1d_first_array_is_object(self): ar1 = [None] |