test_util.py

from util import flatten_dict
 
def test_flatten_dict_nivel_1():
   assert flatten_dict({'a': 1, 'b': 2}) == {'a': 1, 'b': 2}
 
def test_flatten_dict_nivel_3():
   assert flatten_dict({'a': 1, 'b': {'c': 2, 'd': 3}}) == \
       {'a': 1, 'b.c': 2, 'b.d': 3}
 
def test_flatten_dict_nivel_4():
   assert flatten_dict({'a': {'b': {'c': {'d': 1}}}}) == {'a.b.c.d': 1}
 
def test_flatten_dict_nivel_separador():
   assert flatten_dict({'a': {'b': 1}}, '_') == {'a_b': 1}