Completed
Push — master ( 729631...83e72f )
by Satoru
01:04
created

dicts_equal()   C

Complexity

Conditions 8

Size

Total Lines 15

Duplication

Lines 0
Ratio 0 %

Importance

Changes 1
Bugs 0 Features 0
Metric Value
cc 8
c 1
b 0
f 0
dl 0
loc 15
rs 6.6666
1
#
2
# Copyright (C) 2011 - 2016 Satoru SATOH <ssato at redhat.com>
3
#
4
# pylint: disable=missing-docstring
5
import os.path
6
import tempfile
7
import unittest
8
9
import anyconfig.compat
10
11
from anyconfig.compat import OrderedDict
12
from anyconfig.mdicts import is_dict_like
13
14
15
CNF_0 = dict(name="a", a=1, b=dict(b=[1, 2], c="C"))
16
SCM_0 = {"type": "object",
17
         "properties": {
18
             "name": {"type": "string"},
19
             "a": {"type": "integer"},
20
             "b": {"type": "object",
21
                   "properties": {
22
                       "b": {"type": "array",
23
                             "items": {"type": "integer"}}}}}}
24
25
26
def selfdir():
27
    """
28
    >>> os.path.exists(selfdir())
29
    True
30
    """
31
    return os.path.dirname(__file__)
32
33
34
def setup_workdir():
35
    """
36
    >>> workdir = setup_workdir()
37
    >>> assert workdir != '.'
38
    >>> assert workdir != '/'
39
    >>> os.path.exists(workdir)
40
    True
41
    >>> os.rmdir(workdir)
42
    """
43
    return tempfile.mkdtemp(dir="/tmp", prefix="python-anyconfig-tests-")
44
45
46
def cleanup_workdir(workdir):
47
    """
48
    FIXME: Danger!
49
50
    >>> workdir = setup_workdir()
51
    >>> os.path.exists(workdir)
52
    True
53
    >>> open(os.path.join(workdir, "workdir.stamp"), 'w').write("OK!\n")
54
    >>> cleanup_workdir(workdir)
55
    >>> os.path.exists(workdir)
56
    False
57
    """
58
    assert workdir != '/'
59
    assert workdir != '.'
60
61
    os.system("rm -rf " + workdir)
62
63
64
def dicts_equal(dic, ref, ordered=False):
65
    """Compare (maybe nested) dicts.
66
    """
67
    if not is_dict_like(dic) or not is_dict_like(ref):
68
        return dic == ref
69
70
    fnc = list if ordered else len
71
    if fnc(dic.keys()) != fnc(ref.keys()):
72
        return False
73
74
    for key in ref.keys():
75
        if key not in dic or not dicts_equal(dic[key], ref[key]):
76
            return False
77
78
    return True
79
80
81
def to_bytes(astr):
82
    """
83
    Convert a string to bytes. Do nothing in python 2.6.
84
    """
85
    return bytes(astr, 'utf-8') if anyconfig.compat.IS_PYTHON_3 else astr
86
87
88
class Test(unittest.TestCase):
89
90
    def test_dicts_equal(self):
91
        dic0 = {'a': 1}
92
        dic1 = OrderedDict((('a', [1, 2, 3]),
93
                            ('b', OrderedDict((('c', "CCC"), )))))
94
        dic2 = dic1.copy()
95
        dic2["b"] = None
96
97
        dic3 = OrderedDict((('b', OrderedDict((('c', "CCC"), ))),
98
                            ('a', [1, 2, 3])))
99
100
        self.assertTrue(dicts_equal({}, {}))
101
        self.assertTrue(dicts_equal(dic0, dic0))
102
        self.assertTrue(dicts_equal(dic1, dic1))
103
        self.assertTrue(dicts_equal(dic2, dic2))
104
        self.assertTrue(dicts_equal(dic1, dic3))
105
106
        self.assertFalse(dicts_equal(dic0, {}))
107
        self.assertFalse(dicts_equal(dic0, dic1))
108
        self.assertFalse(dicts_equal(dic1, dic2))
109
        self.assertFalse(dicts_equal(dic1, dic3, ordered=True))
110
111
# vim:sw=4:ts=4:et:
112