diff options
| author | Mike Taves <mwtoews@gmail.com> | 2020-01-28 13:12:38 +1300 |
|---|---|---|
| committer | Sebastian Berg <sebastian@sipsolutions.net> | 2020-01-27 16:12:38 -0800 |
| commit | f398a0df8a2105b2fdeaeab54505451169b0a869 (patch) | |
| tree | fe041dbec09eb1a303f7b027cc98e7423476de3f /numpy/core/tests | |
| parent | 6d889e7eca0f7ae6d640c380bd0b604e6530f049 (diff) | |
| download | numpy-f398a0df8a2105b2fdeaeab54505451169b0a869.tar.gz | |
STY: use 'yield from <expr>' for simple cases (#15444)
This PR uses simple cases of PEP 380 to rewrite:
for v in g:
yield v
into:
yield from <expr>
Diffstat (limited to 'numpy/core/tests')
| -rw-r--r-- | numpy/core/tests/test_numeric.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/numpy/core/tests/test_numeric.py b/numpy/core/tests/test_numeric.py index 33e71a064..3bc4cd187 100644 --- a/numpy/core/tests/test_numeric.py +++ b/numpy/core/tests/test_numeric.py @@ -954,8 +954,7 @@ class NIterError(Exception): class TestFromiter: def makegen(self): - for x in range(24): - yield x**2 + return (x**2 for x in range(24)) def test_types(self): ai32 = np.fromiter(self.makegen(), np.int32) |