コード例 #1
0
    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))
コード例 #2
0
    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])