1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45
|
class AppTestMapPickle:
spaceconfig = dict(usemodules=['struct'])
def test_map_pickle(self):
"""Pickle a map with one sequence."""
import pickle
def pickle_unpickle(obj):
d = pickle.dumps(obj)
return pickle.loads(d)
m1 = map(ord, "Is this the real life?")
m1_ = pickle_unpickle(m1)
assert list(m1) == list(m1_)
def test_map2_pickle(self):
"""Pickle a map with multiple sequences."""
import pickle
def pickle_unpickle(obj):
d = pickle.dumps(obj)
return pickle.loads(d)
m1 = map(max, "abc", "def")
m1_ = pickle_unpickle(m1)
assert list(m1) == list(m1_)
def test_map2_adv_pickle(self):
"""If some iterator was advanced, the pickling preserves it."""
import pickle
def pickle_unpickle(obj):
d = pickle.dumps(obj)
return pickle.loads(d)
s1 = iter("abc")
s2 = iter("defg")
next(s2)
m1 = map(max, s1, s2)
m1_ = pickle_unpickle(m1)
assert list(m1) == list(m1_)
|