itermerge.py | raw changeset files shortlog |
1.1 --- a/itermerge.py Sun Feb 13 02:49:55 2011 +0100 1.2 +++ b/itermerge.py Mon Feb 14 00:44:57 2011 +0100 1.3 @@ -74,8 +74,11 @@ 1.4 def next(self): 1.5 if self.iters: 1.6 value, next = self.iters[0] 1.7 - del self.iters[0] 1.8 - self._add_next(next) 1.9 + if len(self.iters) > 1: 1.10 + del self.iters[0] 1.11 + self._add_next(next) 1.12 + else: 1.13 + self.iters[0] = next(), next 1.14 return value 1.15 else: 1.16 raise StopIteration