Rooba
/
python-benedict
mirror of https://github.com/fabiocaccamo/python-benedict.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fixed benedict yaml representer. #43

This commit is contained in:
Fabio Caccamo 2020-11-30 11:26:20 +01:00
parent c1172d8cc0
commit a121c9da9e
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
benedict/dicts/__init__.py
View File

@ -270,3 +270,10 @@ class benedict(KeypathDict, IODict, ParseDict):
Remove duplicated values from the current dict instance. Remove duplicated values from the current dict instance.
""" """
_unique(self) _unique(self)
# fix benedict yaml representer - #43
from yaml import SafeDumper
from yaml.representer import SafeRepresenter
SafeDumper.yaml_representers[cls] = SafeRepresenter.represent_dict

6
tests/github/test_issue_0043.py
View File

@ -3,6 +3,7 @@
from benedict import benedict from benedict import benedict
import unittest import unittest
import yaml
class github_issue_0043_test_case(unittest.TestCase): class github_issue_0043_test_case(unittest.TestCase):
@ -21,3 +22,8 @@ class github_issue_0043_test_case(unittest.TestCase):
level2: Hello world level2: Hello world
""" """
self.assertEqual(s, r) self.assertEqual(s, r)
def test_dict_compatibility(self):
yaml.safe_dump(dict(benedict({})))
yaml.safe_dump(dict(benedict({"level1": None})))
yaml.safe_dump(dict(benedict({"level1": {"level2": "blablabla"}})))