52 lines
1.1 KiB
Python
52 lines
1.1 KiB
Python
|
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
|