def test_infinite_repeat(self):
ret = Q.repeat(1).take(5).to_list()
self.assertSequenceEqual([1, 1, 1, 1, 1], ret)
def test_repeat_with_count(self):
ret = Q.repeat('A', 5).to_list()
self.assertSequenceEqual(['A', 'A', 'A', 'A', 'A'], ret)
