File: test_map_pickle.py

package info (click to toggle)
pypy3 7.0.0%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 111,848 kB
  • sloc: python: 1,291,746; ansic: 74,281; asm: 5,187; cpp: 3,017; sh: 2,533; makefile: 544; xml: 243; lisp: 45; csh: 21; awk: 4
file content (45 lines) | stat: -rw-r--r-- 1,122 bytes parent folder | download | duplicates (6)
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_)