projects
/
youtube-dl
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
[utils] Add get_element_by_class
[youtube-dl]
/
test
/
test_utils.py
diff --git
a/test/test_utils.py
b/test/test_utils.py
index 85928dbc2357567c51a2fe87468c3a8361825a03..afd273a6533b68915c3b9cbcd547591882f9b37a 100644
(file)
--- a/
test/test_utils.py
+++ b/
test/test_utils.py
@@
-33,6
+33,7
@@
from youtube_dl.utils import (
ExtractorError,
find_xpath_attr,
fix_xml_ampersands,
ExtractorError,
find_xpath_attr,
fix_xml_ampersands,
+ get_element_by_class,
InAdvancePagedList,
intlist_to_bytes,
is_html,
InAdvancePagedList,
intlist_to_bytes,
is_html,
@@
-991,5
+992,13
@@
The first line
self.assertEqual(urshift(3, 1), 1)
self.assertEqual(urshift(-3, 1), 2147483646)
self.assertEqual(urshift(3, 1), 1)
self.assertEqual(urshift(-3, 1), 2147483646)
+ def test_get_element_by_class(self):
+ html = '''
+ <span class="foo bar">nice</span>
+ '''
+
+ self.assertEqual(get_element_by_class('foo', html), 'nice')
+ self.assertEqual(get_element_by_class('no-such-class', html), None)
+
if __name__ == '__main__':
unittest.main()
if __name__ == '__main__':
unittest.main()