File: test_impl.py

package info (click to toggle)
python-convertertools 1.1.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 388 kB
  • sloc: python: 157; makefile: 19
file content (64 lines) | stat: -rw-r--r-- 1,373 bytes parent folder | download | duplicates (2)
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
import pytest

from convertertools.impl import (
    del_dict_set,
    del_dict_tuple,
    pop_dict_set,
    pop_dict_set_if_none,
    pop_dict_tuple,
)


def test_pop_dict_set_if_none():
    d = {"a": 1, "b": None, "c": 3}
    pop_dict_set_if_none(d, {"a", "b", "d"})
    assert d == {"a": 1, "c": 3}


def test_del_dict_tuple():
    d = {"a": 1, "b": 2, "c": 3}
    del_dict_tuple(d, ("a", "b"))
    assert d == {"c": 3}


def test_pop_dict_tuple():
    d = {"a": 1, "b": 2, "c": 3}
    pop_dict_tuple(d, ("a", "b"))
    assert d == {"c": 3}


def test_del_dict_set():
    d = {"a": 1, "b": 2, "c": 3}
    del_dict_set(d, {"a", "b"})
    assert d == {"c": 3}


def test_pop_dict_set():
    d = {"a": 1, "b": 2, "c": 3}
    pop_dict_set(d, {"a", "b"})
    assert d == {"c": 3}


def test_del_dict_tuple_missing_key():
    d = {"a": 1, "b": 2, "c": 3}
    with pytest.raises(KeyError):
        del_dict_tuple(d, ("a", "b", "d"))
    assert d == {"c": 3}


def test_pop_dict_tuple_missing_key():
    d = {"a": 1, "b": 2, "c": 3}
    pop_dict_tuple(d, ("a", "b", "d"))
    assert d == {"c": 3}


def test_del_dict_set_missing_key():
    d = {"a": 1, "b": 2, "c": 3}
    with pytest.raises(KeyError):
        del_dict_set(d, {"a", "b", "d"})


def test_pop_dict_set_missing_key():
    d = {"a": 1, "b": 2, "c": 3}
    pop_dict_set(d, {"a", "b", "d"})
    assert d == {"c": 3}