def testTerminatingQuotientStream(self):
"""Check a quotient that terminates."""
q_str = quotient_stream(1, 4)
self.assertEquals(q_str.next(), 0)
self.assertEquals(q_str.next(), 2)
self.assertEquals(q_str.next(), 5)
self.assertRaises(StopIteration, q_str.next)
def testEndlessQuotientStream(self):
"""Check a quotient that goes on forever."""
q_str = quotient_stream(1, 6)
self.assertEquals(q_str.next(), 0)
self.assertEquals(q_str.next(), 1)
self.assertEquals(q_str.next(), 6)
self.assertEquals(q_str.next(), 6)
self.assertEquals(q_str.next(), 6)
