File tree Expand file tree Collapse file tree 2 files changed +10
-36
lines changed Expand file tree Collapse file tree 2 files changed +10
-36
lines changed Original file line number Diff line number Diff line change @@ -319,12 +319,9 @@ assert len(res.fetch(3)) == 1
319319assert len(res.fetch(3)) == 0
320320assert len(res.fetch(3)) == 0
321321try:
322- # use next() or __next__(), the method name changed in
322+ # use next() and not __next__(), the method name changed in
323323 # http://www.python.org/dev/peps/pep-3114/
324- try:
325- res.next()
326- except AttributeError:
327- res.__next__()
324+ next(res)
328325except StopIteration:
329326 pass
330327else:
@@ -334,11 +331,7 @@ CREATE FUNCTION cursor_mix_next_and_fetch() RETURNS int AS $$
334331res = plpy.cursor("select fname, lname from users order by fname")
335332assert len(res.fetch(2)) == 2
336333
337- item = None
338- try:
339- item = res.next()
340- except AttributeError:
341- item = res.__next__()
334+ item = next(res)
342335assert item['fname'] == 'rick'
343336
344337assert len(res.fetch(2)) == 1
@@ -357,10 +350,7 @@ CREATE FUNCTION next_after_close() RETURNS int AS $$
357350res = plpy.cursor("select fname, lname from users")
358351res.close()
359352try:
360- try:
361- res.next()
362- except AttributeError:
363- res.__next__()
353+ next(res)
364354except ValueError:
365355 pass
366356else:
@@ -370,10 +360,7 @@ CREATE FUNCTION cursor_fetch_next_empty() RETURNS int AS $$
370360res = plpy.cursor("select fname, lname from users where false")
371361assert len(res.fetch(1)) == 0
372362try:
373- try:
374- res.next()
375- except AttributeError:
376- res.__next__()
363+ next(res)
377364except StopIteration:
378365 pass
379366else:
Original file line number Diff line number Diff line change @@ -218,12 +218,9 @@ assert len(res.fetch(3)) == 1
218218assert len(res .fetch (3 )) == 0
219219assert len(res .fetch (3 )) == 0
220220try:
221- # use next() or __next__(), the method name changed in
221+ # use next() and not __next__(), the method name changed in
222222 # http://www.python.org/dev/peps/pep-3114/
223- try:
224- res .next ()
225- except AttributeError:
226- res .__next__ ()
223+ next(res)
227224except StopIteration:
228225 pass
229226else:
@@ -234,11 +231,7 @@ CREATE FUNCTION cursor_mix_next_and_fetch() RETURNS int AS $$
234231res = plpy .cursor (" select fname, lname from users order by fname" )
235232assert len(res .fetch (2 )) == 2
236233
237- item = None
238- try:
239- item = res .next ()
240- except AttributeError:
241- item = res .__next__ ()
234+ item = next(res)
242235assert item[' fname' ] == ' rick'
243236
244237assert len(res .fetch (2 )) == 1
@@ -259,10 +252,7 @@ CREATE FUNCTION next_after_close() RETURNS int AS $$
259252res = plpy .cursor (" select fname, lname from users" )
260253res .close ()
261254try:
262- try:
263- res .next ()
264- except AttributeError:
265- res .__next__ ()
255+ next(res)
266256except ValueError:
267257 pass
268258else:
@@ -273,10 +263,7 @@ CREATE FUNCTION cursor_fetch_next_empty() RETURNS int AS $$
273263res = plpy .cursor (" select fname, lname from users where false" )
274264assert len(res .fetch (1 )) == 0
275265try:
276- try:
277- res .next ()
278- except AttributeError:
279- res .__next__ ()
266+ next(res)
280267except StopIteration:
281268 pass
282269else:
You can’t perform that action at this time.
0 commit comments