From 0574cdaeb498d9712b8b83a5c0e86abd52cb7a07 Mon Sep 17 00:00:00 2001 From: Barry Warsaw Date: Tue, 25 Sep 2001 21:40:04 +0000 Subject: 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. --- Lib/test/test_StringIO.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'Lib/test/test_StringIO.py') 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') -- cgit v1.2.1