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
|
import pytest
from munch import Munch, DefaultMunch
def test_from_yaml(yaml):
data = yaml.load('''
Flow style: !munch.Munch { Clark: Evans, Brian: Ingerson, Oren: Ben-Kiki }
Block style: !munch
Clark : Evans
Brian : Ingerson
Oren : Ben-Kiki
''', Loader=yaml.FullLoader)
assert data == {
'Flow style': Munch(Brian='Ingerson', Clark='Evans', Oren='Ben-Kiki'),
'Block style': Munch(Brian='Ingerson', Clark='Evans', Oren='Ben-Kiki'),
}
def test_to_yaml_safe(yaml):
b = Munch(foo=['bar', Munch(lol=True)], hello=42)
dumped = yaml.safe_dump(b, default_flow_style=True)
assert dumped == '{foo: [bar, {lol: true}], hello: 42}\n'
def test_to_yaml(yaml):
b = Munch(foo=['bar', Munch(lol=True)], hello=42)
dumped = yaml.dump(b, default_flow_style=True)
assert dumped == '!munch.Munch {foo: [bar, !munch.Munch {lol: true}], hello: 42}\n'
def test_toYAML(yaml):
b = Munch(foo=['bar', Munch(lol=True)], hello=42)
assert yaml.safe_dump(b, default_flow_style=True) == '{foo: [bar, {lol: true}], hello: 42}\n'
assert b.toYAML(default_flow_style=True) == '{foo: [bar, {lol: true}], hello: 42}\n'
assert yaml.dump(b, default_flow_style=True) == '!munch.Munch {foo: [bar, !munch.Munch {lol: true}], hello: 42}\n'
assert b.toYAML(Dumper=yaml.Dumper, default_flow_style=True) == \
'!munch.Munch {foo: [bar, !munch.Munch {lol: true}], hello: 42}\n'
@pytest.mark.usefixtures('yaml')
def test_fromYAML():
# pylint: disable=unidiomatic-typecheck
yaml_str = 'foo:\n bar:\n - 1\n - 2\n hello: world\n'
obj = Munch.fromYAML(yaml_str)
assert type(obj) == Munch
assert obj == Munch(foo=Munch(bar=[1, 2], hello='world'))
assert obj.toYAML() == yaml_str
@pytest.mark.usefixtures('yaml')
def test_fromYAML_default_munch():
# pylint: disable=unidiomatic-typecheck
yaml_str = 'foo:\n bar:\n - 1\n - 2\n hello: world\n'
default_value = object()
obj = DefaultMunch.fromYAML(yaml_str, default_value)
assert type(obj) == DefaultMunch
assert obj == DefaultMunch(foo=Munch(bar=[1, 2], hello='world'))
assert obj['not_exist'] is default_value
assert obj.not_exist is default_value
assert obj.toYAML() == yaml_str
|
