Merge branch 'shahid' of https://github.com/remitamine/youtube-dl into remitamine...

[youtube-dl] / test / test_utils.py
diff --git a/test/test_utils.py b/test/test_utils.py

index b401070371bfcea183abc2b08419ddd0c75dd3fe..a759b2da99ea9f2c7676171d3bc0d5d93aa2149d 100644 (file)
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -235,12 +235,21 @@ class TestUtil(unittest.TestCase):
              <node x="a"/>
              <node x="a" y="c" />
              <node x="b" y="d" />
+            <node x="" />
          </root>'''
          doc = xml.etree.ElementTree.fromstring(testxml)
  
+        self.assertEqual(find_xpath_attr(doc, './/fourohfour', 'n'), None)
          self.assertEqual(find_xpath_attr(doc, './/fourohfour', 'n', 'v'), None)
+        self.assertEqual(find_xpath_attr(doc, './/node', 'n'), None)
+        self.assertEqual(find_xpath_attr(doc, './/node', 'n', 'v'), None)
+        self.assertEqual(find_xpath_attr(doc, './/node', 'x'), doc[1])
          self.assertEqual(find_xpath_attr(doc, './/node', 'x', 'a'), doc[1])
+        self.assertEqual(find_xpath_attr(doc, './/node', 'x', 'b'), doc[3])
+        self.assertEqual(find_xpath_attr(doc, './/node', 'y'), doc[2])
          self.assertEqual(find_xpath_attr(doc, './/node', 'y', 'c'), doc[2])
+        self.assertEqual(find_xpath_attr(doc, './/node', 'y', 'd'), doc[3])
+        self.assertEqual(find_xpath_attr(doc, './/node', 'x', ''), doc[4])
  
      def test_xpath_with_ns(self):
          testxml = '''<root xmlns:media="http://example.com/">
@@ -324,6 +333,7 @@ class TestUtil(unittest.TestCase):
          self.assertEqual(parse_duration('02:03:04'), 7384)
          self.assertEqual(parse_duration('01:02:03:04'), 93784)
          self.assertEqual(parse_duration('1 hour 3 minutes'), 3780)
+        self.assertEqual(parse_duration('87 Min.'), 5220)
  
      def test_fix_xml_ampersands(self):
          self.assertEqual(
@@ -621,6 +631,21 @@ Line
  '''
          self.assertEqual(dfxp2srt(dfxp_data), srt_data)
  
+        dfxp_data_no_default_namespace = '''<?xml version="1.0" encoding="UTF-8"?>
+            <tt xml:lang="en" xmlns:tts="http://www.w3.org/ns/ttml#parameter">
+            <body>
+                <div xml:lang="en">
+                    <p begin="0" end="1">The first line</p>
+                </div>
+            </body>
+            </tt>'''
+        srt_data = '''1
+00:00:00,000 --> 00:00:01,000
+The first line
+
+'''
+        self.assertEqual(dfxp2srt(dfxp_data_no_default_namespace), srt_data)
+
  
  if __name__ == '__main__':
      unittest.main()