PythonBuddy/pypy/rpython/tool/test/test_identitydict.py at restrictedpython · BoragoCode/PythonBuddy · GitHub

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
46
47
48
49
50
51
52
53
54
55
56
57
58
import py
from rpython.tool.identity_dict import identity_dict, IdentityDictPurePython

class TestIdentityDictNative:
    identity_dict = identity_dict

    def test_numbers(self):
        d = self.identity_dict()
        d[0] = 1
        d[0.0] = 2
        d[long(0)] = 3

        assert d
        assert len(d) == 3
        d.clear()
        assert not d

    def test_get(self):
        d = self.identity_dict()
        d[None] = 1

        assert d.get(None, 42) == 1
        assert d.get(None) == 1
        assert d.get(1) is None
        assert d.get(1, 42) == 42

    def test_unhashable(self):
        d = self.identity_dict()
        d[[]] = 1
        d[[]] = 2
        a = []
        d[a] = 3
        assert len(d) == 3
        d[a] = 4
        assert len(d) == 3
        assert d[a] == 4

        py.test.raises(KeyError, d.__getitem__, [])

    def test_keys(self):
        d = self.identity_dict()
        d[[]] = 1
        d[[]] = 2
        d[[]] = 3

        assert d.keys() == [[], [], []]
        assert sorted(d.values()) == [1, 2, 3]

    def test_in(self):
        d = self.identity_dict()
        d[None] = 1

        assert None in d
        assert [] not in d


class TestIdentityDictPurePython(TestIdentityDictNative):
    identity_dict = IdentityDictPurePython