jellyfin-kodi/tests/test_clean_none_dict_values.py

import sys

import pytest

sys.path.insert(0, 'jellyfin_kodi')

from jellyfin.utils import clean_none_dict_values  # noqa: E402


@pytest.mark.parametrize("obj,expected", [
    (None, None),
    ([None, 1, 2, 3, None, 4], [None, 1, 2, 3, None, 4]),
    ({'foo': None, 'bar': 123}, {'bar': 123}),
    ({
        'dict': {
            'empty': None,
            'string': "Hello, Woorld!",
        },
        'number': 123,
        'list': [
            None,
            123,
            "foo",
            {
                'empty': None,
                'number': 123,
                'string': "foo",
                'list': [],
                'dict': {},
            }
        ]
    }, {
        'dict': {
            'string': "Hello, Woorld!",
        },
        'number': 123,
        'list': [
            None,
            123,
            "foo",
            {
                'number': 123,
                'string': "foo",
                'list': [],
                'dict': {},
            }
        ]
    }),
])
def test_clean_none_dict_values(obj, expected):
    assert clean_none_dict_values(obj) == expected