summary refs log tree commit diff
path: root/test
diff options
context:
space:
mode:
authorSergey M․ <dstftw@gmail.com>2019-03-17 09:07:47 +0700
committerSergey M․ <dstftw@gmail.com>2019-03-17 09:07:47 +0700
commit0dc41787af0fb011f01d88c1f2ecb686a8416df5 (patch)
tree42bdaecb36f30b6be28de42a5a94471d42f8c3db /test
parente5cfb779ea0814971324d693e02f8c4d4ca78600 (diff)
downloadyoutube-dl-0dc41787af0fb011f01d88c1f2ecb686a8416df5.tar.gz
youtube-dl-0dc41787af0fb011f01d88c1f2ecb686a8416df5.tar.xz
youtube-dl-0dc41787af0fb011f01d88c1f2ecb686a8416df5.zip
[utils] Introduce parse_bitrate
Diffstat (limited to 'test')
-rw-r--r--test/test_utils.py8
1 files changed, 8 insertions, 0 deletions
diff --git a/test/test_utils.py b/test/test_utils.py
index 409482c3b..acd994bd7 100644
--- a/test/test_utils.py
+++ b/test/test_utils.py
@@ -55,6 +55,7 @@ from youtube_dl.utils import (
     parse_count,
     parse_iso8601,
     parse_resolution,
+    parse_bitrate,
     pkcs1pad,
     read_batch_urls,
     sanitize_filename,
@@ -1030,6 +1031,13 @@ class TestUtil(unittest.TestCase):
         self.assertEqual(parse_resolution('4k'), {'height': 2160})
         self.assertEqual(parse_resolution('8K'), {'height': 4320})
 
+    def test_parse_bitrate(self):
+        self.assertEqual(parse_bitrate(None), None)
+        self.assertEqual(parse_bitrate(''), None)
+        self.assertEqual(parse_bitrate('300kbps'), 300)
+        self.assertEqual(parse_bitrate('1500kbps'), 1500)
+        self.assertEqual(parse_bitrate('300 kbps'), 300)
+
     def test_version_tuple(self):
         self.assertEqual(version_tuple('1'), (1,))
         self.assertEqual(version_tuple('10.23.344'), (10, 23, 344))