File tree Expand file tree Collapse file tree 1 file changed +11
-10
lines changed Expand file tree Collapse file tree 1 file changed +11
-10
lines changed Original file line number Diff line number Diff line change @@ -81,6 +81,16 @@ class BadIterableClass:
8181 def __iter__ (self ):
8282 raise ZeroDivisionError
8383
84+ class CallableIterClass :
85+ def __init__ (self ):
86+ self .i = 0
87+ def __call__ (self ):
88+ i = self .i
89+ self .i = i + 1
90+ if i > 100 :
91+ raise IndexError # Emergency stop
92+ return i
93+
8494# Main test suite
8595
8696class TestCase (unittest .TestCase ):
@@ -237,16 +247,7 @@ def __iter__(self):
237247
238248 # Test two-argument iter() with callable instance
239249 def test_iter_callable (self ):
240- class C :
241- def __init__ (self ):
242- self .i = 0
243- def __call__ (self ):
244- i = self .i
245- self .i = i + 1
246- if i > 100 :
247- raise IndexError # Emergency stop
248- return i
249- self .check_iterator (iter (C (), 10 ), list (range (10 )), pickle = False )
250+ self .check_iterator (iter (CallableIterClass (), 10 ), list (range (10 )), pickle = True )
250251
251252 # Test two-argument iter() with function
252253 def test_iter_function (self ):
You can’t perform that action at this time.
0 commit comments