def testZip(self): l = Stream.zip(range(5), Stream.number(start=4, step=-1).take(8)).to_list() self.assertEqual(5, len(l)) self.assertTrue( Stream(l).map(lambda tup: tup[0] + tup[1]).all(lambda x: x == 4)) l = Stream.number(start=4, step=-1).take(8).zip_with(range(5)).to_list() self.assertEqual(5, len(l)) self.assertTrue( Stream(l).map(lambda tup: tup[0] + tup[1]).all(lambda x: x == 4))
def testToMap(self): m = Stream.zip(range(5), Stream.number(start=4, step=-1).take(3)).to_map() self.assertEqual(3, len(m)) self.assertEqual(4, m[0]) self.assertEqual(3, m[1]) self.assertEqual(2, m[2]) m = Stream.number().take(3).to_map(key=lambda it: it, value=lambda it: it * 2) self.assertEqual(2, m[1]) self.assertEqual(4, m[2]) self.assertEqual(0, m[0])