def testStreamPickling(self):
sio = BytesIO()
expected = slist(slist((i, )) for i in xrange(10))
expected.dumpToPickle(sio)
sio = BytesIO(sio.getvalue())
result = stream.loadFromPickled(sio)
self.assertEquals(list(expected), list(result))
def test_reduceUsesInitProperly(self):
self.assertEquals(
slist([sset((1, 2)), sset(
(3, 4))]).reduce(lambda x, y: x.update(y)), set((1, 2, 3, 4)))
self.assertEquals(
slist([sset((1, 2)), sset(
(3, 4))]).reduce(lambda x, y: x.update(y), sset()),
set((1, 2, 3, 4)))
def testStreamPickling(self):
sio = BytesIO()
expected = slist(slist((i,)) for i in xrange(10))
expected.dumpToPickle(sio)
sio = BytesIO(sio.getvalue())
result = stream.loadFromPickled(sio)
self.assertEquals(list(expected), list(result))
def testStreamList(self):
l = lambda: slist((1, 2, 3))
self.assertEqual(l().toList(), [1, 2, 3])
self.assertEqual(l()[-1], 3)
def test_flatMap_defaultIdentityFunction(self):
l = slist(({1: 2, 3: 4}, {5: 6, 7: 8}))
self.assertEquals(l.flatMap().toSet(), set((1, 3, 5, 7)))
def testStreamList(self):
l = lambda: slist((1, 2, 3))
self.assertEqual(l().toList(), [1, 2, 3])
self.assertEqual(l()[-1], 3)
def test_reduceUsesInitProperly(self):
self.assertEquals(slist([sset((1, 2)), sset((3, 4))]).reduce(lambda x, y: x.update(y)), set((1, 2, 3, 4)))
self.assertEquals(slist([sset((1, 2)), sset((3, 4))]).reduce(lambda x, y: x.update(y), sset()),
set((1, 2, 3, 4)))
def test_flatMap_defaultIdentityFunction(self):
l = slist(({1: 2, 3: 4}, {5: 6, 7: 8}))
self.assertEquals(l.flatMap().toSet(), set((1, 3, 5, 7)))
