@@ -617,6 +617,16 @@ async def gen():
617617 applied_twice = aiter (applied_once )
618618 self .assertIs (applied_once , applied_twice )
619619
620+ def test_aiter_callable_errors (self ):
621+ async def gen ():
622+ yield 1
623+ self .assertRaises (TypeError , aiter , gen (), 1 )
624+ self .assertRaises (TypeError , aiter , [1 , 2 ], stop_exception = LookupError )
625+ self .assertRaises (TypeError , aiter , len , stop_exception = 42 )
626+ self .assertRaises (TypeError , aiter , len ,
627+ stop_exception = (LookupError , 42 ))
628+ self .assertRaises (TypeError , aiter , len , stop_exception = LookupError ())
629+
620630 def test_anext_iter (self ):
621631 class MyError (Exception ):
622632 pass
@@ -1119,16 +1129,6 @@ async def main():
11191129 self .loop .run_until_complete (main ())
11201130 self .assertEqual (cancelled , [1 ])
11211131
1122- def test_aiter_callable_errors (self ):
1123- async def gen ():
1124- yield 1
1125- self .assertRaises (TypeError , aiter , gen (), 1 )
1126- self .assertRaises (TypeError , aiter , [1 , 2 ], stop_exception = LookupError )
1127- self .assertRaises (TypeError , aiter , len , stop_exception = 42 )
1128- self .assertRaises (TypeError , aiter , len ,
1129- stop_exception = (LookupError , 42 ))
1130- self .assertRaises (TypeError , aiter , len , stop_exception = LookupError ())
1131-
11321132 def test_anext_bad_args (self ):
11331133 async def gen ():
11341134 yield 1
0 commit comments