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 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)))