Python IterIO.next示例

示例#1

文件： iterio.py 项目： 0xJCG/dubtrack-technical-test

 def test_basic(self):
     def producer(out):
         out.write('1\n')
         out.write('2\n')
         out.flush()
         out.write('3\n')
     iterable = IterIO(producer)
     self.assert_equal(iterable.next(), '1\n2\n')
     self.assert_equal(iterable.next(), '3\n')
     self.assert_raises(StopIteration, iterable.next)

示例#2

    def test_basic(self):
        def producer(out):
            out.write('1\n')
            out.write('2\n')
            out.flush()
            out.write('3\n')

        iterable = IterIO(producer)
        self.assert_equal(iterable.next(), '1\n2\n')
        self.assert_equal(iterable.next(), '3\n')
        self.assert_raises(StopIteration, iterable.next)