File tree 1 file changed +11
-10
lines changed
1 file changed +11
-10
lines changed Original file line number Diff line number Diff line change @@ -81,6 +81,16 @@ class BadIterableClass:
81
81
def __iter__ (self ):
82
82
raise ZeroDivisionError
83
83
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
+
84
94
# Main test suite
85
95
86
96
class TestCase (unittest .TestCase ):
@@ -237,16 +247,7 @@ def __iter__(self):
237
247
238
248
# Test two-argument iter() with callable instance
239
249
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 )
250
251
251
252
# Test two-argument iter() with function
252
253
def test_iter_function (self ):
You can’t perform that action at this time.
0 commit comments