Beispiel #1
0
def test_add_none_to_empty_fields():
    edf_test = EDF()

    test_dict = {'a': ['a'], 'b': [1, 3], 'c': [1]}
    output = edf_test._add_none_to_empty_fields(test_dict)
    assert output == {'a': ['a', None], 'b': [1, 3], 'c': [1, None]}

    test_dict = {'a': [], 'b': [], 'c': []}
    output = edf_test._add_none_to_empty_fields(test_dict)
    assert output == {'a': [], 'b': [], 'c': []}

    test_dict = {'a': [1, 2], 'b': [1, 3], 'c': [1, 'a']}
    output = edf_test._add_none_to_empty_fields(test_dict)
    assert output == {'a': [1, 2], 'b': [1, 3], 'c': [1, 'a']}