From: Barry Warsaw <barry@python.org>
Date: Tue, 25 Sep 2001 21:40:04 +0000 (+0000)
Subject: test_iterator(): Don't do a type comparison to see if it's an
X-Git-Tag: v2.2.1c1~1618
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=45653503ec0a0f8db383aa3532bc09450970c74a;p=python

test_iterator(): Don't do a type comparison to see if it's an
iterator, just test to make sure it has the two required iterator
protocol methods __iter__() and next() -- actually just test
hasattr-ness.
---

diff --git a/Lib/test/test_StringIO.py b/Lib/test/test_StringIO.py
index 4a0a814ddd..33db4ba5bc 100644
--- a/Lib/test/test_StringIO.py
+++ b/Lib/test/test_StringIO.py
@@ -57,8 +57,11 @@ class TestGenericStringIO(unittest.TestCase):
 
     def test_iterator(self):
         eq = self.assertEqual
+        unless = self.failUnless
         it = iter(self._fp)
-        self.failUnless(isinstance(it, types.FunctionIterType))
+        # Does this object support the iteration protocol?
+        unless(hasattr(it, '__iter__'))
+        unless(hasattr(it, 'next'))
         i = 0
         for line in self._fp:
             eq(line, self._line + '\n')