PsychoticNinja/Limnoria
1
0
Fork 0
mirror of https://github.com/progval/Limnoria.git synced 2025-04-26 04:51:06 -05:00
Code Issues Projects Releases Wiki Activity

Added utils.iter.one, to grab one value from an iterable.

Browse Source
This commit is contained in:
Jeremy Fincher 2005-06-12 04:54:15 +00:00
parent 201d4acd26
commit b369039507
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/utils/iter.py
View File

@ -148,6 +148,11 @@ def ilen(iterable):
i += 1 i += 1
return i return i
def one(iterable):
for x in iterable:
return x
raise ValueError, 'Empty iterable: %r' % iterable
def startswith(long, short): def startswith(long, short):
longI = iter(long) longI = iter(long)
shortI = iter(short) shortI = iter(short)

8
test/test_utils.py
View File

@ -384,6 +384,14 @@ class StrTest(SupyTestCase):
class IterTest(SupyTestCase): class IterTest(SupyTestCase):
def testOne(self):
L = range(3)
i = iter(L)
self.assertEqual(utils.iter.one(i), 0)
self.assertEqual(utils.iter.one(i), 1)
self.assertEqual(utils.iter.one(i), 2)
self.assertRaises(ValueError, utils.iter.one, i)
def testLimited(self): def testLimited(self):
L = range(10) L = range(10)
self.assertEqual([], list(utils.iter.limited(L, 0))) self.assertEqual([], list(utils.iter.limited(L, 0)))