summaryrefslogtreecommitdiff
path: root/tests/run/set_item.pyx
blob: 4086351cac3ebeb6c537f864380c8f55c66ff2c7 (plain)
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
# mode: run
# tag: list, dict, setitem, delitem

def set_item(obj, key, value):
    """
    >>> set_item([1, 2, 3], 1, -1)
    [1, -1, 3]
    >>> set_item([1, 2, 3], -1, -1)
    [1, 2, -1]
    >>> set_item({}, 'abc', 5)
    {'abc': 5}
    >>> set_item({}, -1, 5)
    {-1: 5}
    >>> class D(dict): pass
    >>> set_item(D({}), 'abc', 5)
    {'abc': 5}
    >>> set_item(D({}), -1, 5)
    {-1: 5}
    """
    obj[key] = value
    return obj


def set_item_int(obj, int key, value):
    """
    >>> set_item_int([1, 2, 3], 1, -1)
    [1, -1, 3]
    >>> set_item_int([1, 2, 3], -1, -1)
    [1, 2, -1]
    >>> set_item_int({}, 1, 5)
    {1: 5}
    >>> set_item_int({}, -1, 5)
    {-1: 5}
    >>> class D(dict): pass
    >>> set_item_int(D({}), 1, 5)
    {1: 5}
    >>> set_item_int(D({}), -1, 5)
    {-1: 5}
    """
    obj[key] = value
    return obj


def del_item(obj, key):
    """
    >>> del_item([1, 2, 3], 1)
    [1, 3]
    >>> del_item([1, 2, 3], -3)
    [2, 3]
    >>> class D(dict): pass
    >>> del_item({'abc': 1, 'def': 2}, 'abc')
    {'def': 2}
    >>> del_item(D({'abc': 1, 'def': 2}), 'abc')
    {'def': 2}
    >>> del_item(D({-1: 1, -2: 2}), -1)
    {-2: 2}
    """
    del obj[key]
    return obj


def del_item_int(obj, int key):
    """
    >>> del_item_int([1, 2, 3], 1)
    [1, 3]
    >>> del_item_int([1, 2, 3], -3)
    [2, 3]
    >>> class D(dict): pass
    >>> del_item_int(D({-1: 1, 1: 2}), 1)
    {-1: 1}
    >>> del_item_int(D({-1: 1, -2: 2}), -1)
    {-2: 2}
    """
    del obj[key]
    return obj