diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_traversal.py | 39 |
1 files changed, 35 insertions, 4 deletions
diff --git a/test/test_traversal.py b/test/test_traversal.py index 21f81136f..5d08b8dbb 100644 --- a/test/test_traversal.py +++ b/test/test_traversal.py @@ -15,8 +15,11 @@ import re from youtube_dl.traversal import ( dict_get, get_first, + require, T, traverse_obj, + unpack, + value, ) from youtube_dl.compat import ( compat_chr as chr, @@ -27,7 +30,9 @@ from youtube_dl.compat import ( compat_zip as zip, ) from youtube_dl.utils import ( + ExtractorError, int_or_none, + join_nonempty, str_or_none, ) @@ -462,8 +467,8 @@ class TestTraversal(_TestCase): }), values = dict((str(k), v) for k, v in values.items()) - for key, value in values.items(): - self.assertEqual(traverse_obj(morsel, key), value, + for key, val in values.items(): + self.assertEqual(traverse_obj(morsel, key), val, msg='Morsel should provide access to all values') values = list(values.values()) self.assertMaybeCountEqual(traverse_obj(morsel, Ellipsis), values, @@ -481,8 +486,31 @@ class TestTraversal(_TestCase): [True, 1, 1.1, 'str', {0: 0}, [1]], '`filter` should filter falsy values') - def test_get_first(self): - self.assertEqual(get_first([{'a': None}, {'a': 'spam'}], 'a'), 'spam') + +class TestTraversalHelpers(_TestCase): + def test_traversal_require(self): + with self.assertRaises(ExtractorError, msg='Missing `value` should raise'): + traverse_obj(_TEST_DATA, ('None', T(require('value')))) + self.assertEqual( + traverse_obj(_TEST_DATA, ('str', T(require('value')))), 'str', + '`require` should pass through non-`None` values') + + def test_unpack(self): + self.assertEqual( + unpack(lambda *x: ''.join(map(compat_str, x)))([1, 2, 3]), '123') + self.assertEqual( + unpack(join_nonempty)([1, 2, 3]), '1-2-3') + self.assertEqual( + unpack(join_nonempty, delim=' ')([1, 2, 3]), '1 2 3') + with self.assertRaises(TypeError): + unpack(join_nonempty)() + with self.assertRaises(TypeError): + unpack() + + def test_value(self): + self.assertEqual( + traverse_obj(_TEST_DATA, ('str', T(value('other')))), 'other', + '`value` should substitute specified value') class TestDictGet(_TestCase): @@ -508,6 +536,9 @@ class TestDictGet(_TestCase): self.assertEqual(dict_get(d, ('b', 'c', key, )), None) self.assertEqual(dict_get(d, ('b', 'c', key, ), skip_false_values=False), false_value) + def test_get_first(self): + self.assertEqual(get_first([{'a': None}, {'a': 'spam'}], 'a'), 'spam') + if __name__ == '__main__': unittest.main() |