diff options
Diffstat (limited to 'test/test_utils.py')
| -rw-r--r-- | test/test_utils.py | 51 | 
1 files changed, 38 insertions, 13 deletions
| diff --git a/test/test_utils.py b/test/test_utils.py index 614ef1b03..d4b4cb920 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -2,22 +2,47 @@  # Various small unit tests +import unittest + +#from youtube_dl.utils import htmlentity_transform +from youtube_dl.utils import timeconvert  from youtube_dl.utils import sanitize_filename +from youtube_dl.utils import unescapeHTML +from youtube_dl.utils import orderedSet + + +class TestUtil(unittest.TestCase): +    def test_timeconvert(self): +        self.assertIsNone(timeconvert('')) +        self.assertIsNone(timeconvert('bougrg')) + +    def test_sanitize_filename(self): +        self.assertEqual(sanitize_filename(u'abc'), u'abc') +        self.assertEqual(sanitize_filename(u'abc_d-e'), u'abc_d-e') + +        self.assertEqual(sanitize_filename(u'123'), u'123') + +        self.assertNotIn(u'/', sanitize_filename(u'abc/de')) +        self.assertNotIn(u'abc', sanitize_filename(u'abc/de')) +        self.assertIn(u'de', sanitize_filename(u'abc/de')) +        self.assertNotIn(u'/', sanitize_filename(u'abc/de///')) + +        self.assertIn(u'\\', sanitize_filename(u'abc\\de')) +        self.assertIn(u'abc', sanitize_filename(u'abc\\de')) +        self.assertIn(u'de', sanitize_filename(u'abc\\de')) -def test_sanitize_filename(): -	assert sanitize_filename(u'abc') == u'abc' -	assert sanitize_filename(u'abc_d-e') == u'abc_d-e' +        self.assertEqual(sanitize_filename(u'ä'), u'ä') +        self.assertEqual(sanitize_filename(u'кириллица'), u'кириллица') -	assert sanitize_filename(u'123') == u'123' +    def test_ordered_set(self): +        self.assertEqual(orderedSet([1,1,2,3,4,4,5,6,7,3,5]), [1,2,3,4,5,6,7]) +        self.assertEqual(orderedSet([]), []) +        self.assertEqual(orderedSet([1]), [1]) +        #keep the list ordered +        self.assertEqual(orderedSet([135,1,1,1]), [135,1]) -	assert u'/' not in sanitize_filename(u'abc/de') -	assert u'abc' in sanitize_filename(u'abc/de') -	assert u'de' in sanitize_filename(u'abc/de') -	assert u'/' not in sanitize_filename(u'abc/de///') +    def test_unescape_html(self): +        self.assertEqual(unescapeHTML(u"%20;"), u"%20;") +        self.assertEqual(unescapeHTML(u"gre&tre&yre"), [u'gre', u'tre', u'yre']) -	assert u'\\' not in sanitize_filename(u'abc\\de') -	assert u'abc' in sanitize_filename(u'abc\\de') -	assert u'de' in sanitize_filename(u'abc\\de') -	assert sanitize_filename(u'ä') == u'ä' -	assert sanitize_filename(u'кириллица') == u'кириллица' | 
