diff options
Diffstat (limited to 'test/test_compat.py')
| -rw-r--r-- | test/test_compat.py | 21 | 
1 files changed, 21 insertions, 0 deletions
| diff --git a/test/test_compat.py b/test/test_compat.py index c3ba8ad2e..b6bfad05e 100644 --- a/test/test_compat.py +++ b/test/test_compat.py @@ -13,7 +13,10 @@ sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))  from youtube_dl.utils import get_filesystem_encoding  from youtube_dl.compat import (      compat_getenv, +    compat_etree_fromstring,      compat_expanduser, +    compat_shlex_split, +    compat_str,      compat_urllib_parse_unquote,      compat_urllib_parse_unquote_plus,  ) @@ -67,5 +70,23 @@ class TestCompat(unittest.TestCase):          self.assertEqual(compat_urllib_parse_unquote_plus('abc%20def'), 'abc def')          self.assertEqual(compat_urllib_parse_unquote_plus('%7e/abc+def'), '~/abc def') +    def test_compat_shlex_split(self): +        self.assertEqual(compat_shlex_split('-option "one two"'), ['-option', 'one two']) + +    def test_compat_etree_fromstring(self): +        xml = ''' +            <root foo="bar" spam="中文"> +                <normal>foo</normal> +                <chinese>中文</chinese> +                <foo><bar>spam</bar></foo> +            </root> +        ''' +        doc = compat_etree_fromstring(xml.encode('utf-8')) +        self.assertTrue(isinstance(doc.attrib['foo'], compat_str)) +        self.assertTrue(isinstance(doc.attrib['spam'], compat_str)) +        self.assertTrue(isinstance(doc.find('normal').text, compat_str)) +        self.assertTrue(isinstance(doc.find('chinese').text, compat_str)) +        self.assertTrue(isinstance(doc.find('foo/bar').text, compat_str)) +  if __name__ == '__main__':      unittest.main() | 
