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)