Skip to content

Commit eacef41

Browse files
committed
Fix lint errors
1 parent 7a46b32 commit eacef41

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

numpy/_core/tests/test_numeric.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -324,15 +324,17 @@ def test_take(self):
324324
out = np.take(a, indices)
325325
assert_equal(out, tgt)
326326

327-
pairs = [(np.int32, np.int32), (np.int32, np.int64), (np.int64, np.int32), (np.int64, np.int64)]
327+
pairs = [
328+
(np.int32, np.int32), (np.int32, np.int64),
329+
(np.int64, np.int32), (np.int64, np.int64)
330+
]
328331
for array_type, indices_type in pairs:
329-
x = np.array([1, 2, 3, 4, 5], dtype=array_type)
330-
ind = np.array([0, 2, 2, 3], dtype=indices_type )
331-
tgt = np.array([1, 3, 3, 4], dtype=array_type)
332-
out = np.take(x, ind)
333-
assert_equal(out, tgt)
334-
assert_equal(out.dtype, tgt.dtype)
335-
332+
x = np.array([1, 2, 3, 4, 5], dtype=array_type)
333+
ind = np.array([0, 2, 2, 3], dtype=indices_type)
334+
tgt = np.array([1, 3, 3, 4], dtype=array_type)
335+
out = np.take(x, ind)
336+
assert_equal(out, tgt)
337+
assert_equal(out.dtype, tgt.dtype)
336338

337339
def test_trace(self):
338340
c = [[1, 2], [3, 4], [5, 6]]

0 commit comments

Comments
 (0)