black formatting for tests

.idea/dictionaries/haroldmartin.xml

@@ -1,8 +1,12 @@
 <component name="ProjectDictionaryState">
   <dictionary name="haroldmartin">
     <words>
+      <w>descramble</w>
       <w>descrambler</w>
+      <w>ficano</w>
       <w>itag</w>
+      <w>noqa</w>
+      <w>pytube</w>
     </words>
   </dictionary>
 </component>

pytube/captions.py

@@ -1,5 +1,5 @@
 # -*- coding: utf-8 -*-
-"""This module contrains a container for caption tracks."""
+"""This module contains a container for caption tracks."""
 import math
 import time
 import xml.etree.ElementTree as ElementTree

tests/conftest.py

@@ -14,19 +14,19 @@ def load_playback_file(filename):
     """Load a gzip json playback file."""
     cur_fp = os.path.realpath(__file__)
     cur_dir = os.path.dirname(cur_fp)
-    fp = os.path.join(cur_dir, 'mocks', filename)
-    with gzip.open(fp, 'rb') as fh:
-        content = fh.read().decode('utf-8')
+    fp = os.path.join(cur_dir, "mocks", filename)
+    with gzip.open(fp, "rb") as fh:
+        content = fh.read().decode("utf-8")
         return json.loads(content)
 
 
 def load_and_init_from_playback_file(filename):
     """Load a gzip json playback file and create YouTube instance."""
     pb = load_playback_file(filename)
-    yt = YouTube(pb['url'], defer_prefetch_init=True)
-    yt.watch_html = pb['watch_html']
-    yt.js = pb['js']
-    yt.vid_info = pb['video_info']
+    yt = YouTube(pb["url"], defer_prefetch_init=True)
+    yt.watch_html = pb["watch_html"]
+    yt.js = pb["js"]
+    yt.vid_info = pb["video_info"]
     yt.descramble()
     return yt
 
@@ -34,19 +34,19 @@ def load_and_init_from_playback_file(filename):
 @pytest.fixture
 def cipher_signature():
     """Youtube instance initialized with video id 9bZkp7q19f0."""
-    filename = 'yt-video-9bZkp7q19f0.json.gz'
+    filename = "yt-video-9bZkp7q19f0.json.gz"
     return load_and_init_from_playback_file(filename)
 
 
 @pytest.fixture
 def presigned_video():
     """Youtube instance initialized with video id QRS8MkLhQmM."""
-    filename = 'yt-video-QRS8MkLhQmM.json.gz'
+    filename = "yt-video-QRS8MkLhQmM.json.gz"
     return load_and_init_from_playback_file(filename)
 
 
 @pytest.fixture
 def age_restricted():
     """Youtube instance initialized with video id zRbsm3e2ltw."""
-    filename = 'yt-video-zRbsm3e2ltw-1507777044.json.gz'
+    filename = "yt-video-zRbsm3e2ltw-1507777044.json.gz"
     return load_playback_file(filename)

tests/contrib/test_playlist.py

@@ -3,8 +3,8 @@ from pytube import Playlist
 
 
 def test_title():
-    list_key = 'PLsyeobzWxl7poL9JTVyndKe62ieoN-MZ3'
-    url = 'https://www.youtube.com/playlist?list=' + list_key
+    list_key = "PLsyeobzWxl7poL9JTVyndKe62ieoN-MZ3"
+    url = "https://www.youtube.com/playlist?list=" + list_key
     pl = Playlist(url)
     pl_title = pl.title()
-    assert pl_title == 'Python Tutorial for Beginners'
+    assert pl_title == "Python Tutorial for Beginners"

tests/generate_fixture.py

@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+# flake8: noqa: E402
+
 from os import path
 import sys
 import json

tests/test_cipher.py

@@ -7,4 +7,4 @@ from pytube.exceptions import RegexMatchError
 
 def test_map_functions():
     with pytest.raises(RegexMatchError):
-        cipher.map_functions('asdf')
+        cipher.map_functions("asdf")

tests/test_cli.py

@@ -4,10 +4,10 @@ from unittest import mock
 from pytube import cli
 
 
-@mock.patch('pytube.cli.YouTube')
-@mock.patch('pytube.cli.sys')
+@mock.patch("pytube.cli.YouTube")
+@mock.patch("pytube.cli.sys")
 def test_download(MockYouTube, mock_sys):
     instance = MockYouTube.return_value
     instance.prefetch_descramble.return_value = None
     instance.streams = mock.Mock()
-    cli.download('asdf', 'asdf')
+    cli.download("asdf", "asdf")

tests/test_exceptions.py

@@ -4,6 +4,6 @@ from pytube.exceptions import ExtractError
 
 def test_is_expected():
     try:
-        raise ExtractError('ppfff', video_id='YLnZklYFe7E')
+        raise ExtractError("ppfff", video_id="YLnZklYFe7E")
     except ExtractError as e:
-        assert e.video_id == 'YLnZklYFe7E'
+        assert e.video_id == "YLnZklYFe7E"

tests/test_extract.py

@@ -4,15 +4,15 @@ from pytube import extract
 
 
 def test_extract_video_id():
-    url = 'https://www.youtube.com/watch?v=9bZkp7q19f0'
+    url = "https://www.youtube.com/watch?v=9bZkp7q19f0"
     video_id = extract.video_id(url)
-    assert video_id == '9bZkp7q19f0'
+    assert video_id == "9bZkp7q19f0"
 
 
 def test_extract_watch_url():
-    video_id = '9bZkp7q19f0'
+    video_id = "9bZkp7q19f0"
     watch_url = extract.watch_url(video_id)
-    assert watch_url == 'https://youtube.com/watch?v=9bZkp7q19f0'
+    assert watch_url == "https://youtube.com/watch?v=9bZkp7q19f0"
 
 
 def test_info_url(cipher_signature):
@@ -20,25 +20,25 @@ def test_info_url(cipher_signature):
         video_id=cipher_signature.video_id,
         watch_url=cipher_signature.watch_url,
         watch_html=cipher_signature.watch_html,
-        embed_html='',
+        embed_html="",
         age_restricted=False,
     )
     expected = (
-        'https://youtube.com/get_video_info?video_id=9bZkp7q19f0&el=%24el'
-        '&ps=default&eurl=https%253A%2F%2Fyoutube.com%2Fwatch%253Fv%'
-        '253D9bZkp7q19f0&hl=en_US'
+        "https://youtube.com/get_video_info?video_id=9bZkp7q19f0&el=%24el"
+        "&ps=default&eurl=https%253A%2F%2Fyoutube.com%2Fwatch%253Fv%"
+        "253D9bZkp7q19f0&hl=en_US"
     )
     assert video_info_url == expected
 
 
 def test_js_url(cipher_signature):
-    expected = 'https://youtube.com/yts/jsbin/player-vflOdyxa4/en_US/base.js'
+    expected = "https://youtube.com/yts/jsbin/player-vflOdyxa4/en_US/base.js"
     result = extract.js_url(cipher_signature.watch_html)
     assert expected == result
 
 
 def test_age_restricted(age_restricted):
-    assert extract.is_age_restricted(age_restricted['watch_html'])
+    assert extract.is_age_restricted(age_restricted["watch_html"])
 
 
 def test_non_age_restricted(cipher_signature):

tests/test_helpers.py

@@ -7,15 +7,15 @@ from pytube.exceptions import RegexMatchError
 
 def test_regex_search_no_match():
     with pytest.raises(RegexMatchError):
-        helpers.regex_search('^a$', '', groups=True)
+        helpers.regex_search("^a$", "", groups=True)
 
 
 def test_regex_search():
     # TODO(nficano): should check isinstance
-    assert helpers.regex_search('^a$', 'a') is not None
+    assert helpers.regex_search("^a$", "a") is not None
 
 
 def test_safe_filename():
     """Unsafe characters get stripped from generated filename"""
-    assert helpers.safe_filename('abc1245$$') == 'abc1245'
-    assert helpers.safe_filename('abc##') == 'abc'
+    assert helpers.safe_filename("abc1245$$") == "abc1245"
+    assert helpers.safe_filename("abc##") == "abc"

tests/test_itags.py

@@ -4,9 +4,9 @@ from pytube import itags
 
 def test_get_format_profile():
     profile = itags.get_format_profile(22)
-    assert profile['resolution'] == '720p'
+    assert profile["resolution"] == "720p"
 
 
 def test_get_format_profile_non_existant():
     profile = itags.get_format_profile(2239)
-    assert profile['resolution'] is None
+    assert profile["resolution"] is None

tests/test_main.py

@@ -4,9 +4,9 @@ from unittest import mock
 from pytube import YouTube
 
 
-@mock.patch('pytube.__main__.YouTube')
+@mock.patch("pytube.__main__.YouTube")
 def test_prefetch_deferred(MockYouTube):
     instance = MockYouTube.return_value
     instance.prefetch_descramble.return_value = None
-    YouTube('https://www.youtube.com/watch?v=9bZkp7q19f0', True)
+    YouTube("https://www.youtube.com/watch?v=9bZkp7q19f0", True)
     assert not instance.prefetch_descramble.called

tests/test_request.py

@@ -6,7 +6,7 @@ from unittest import mock
 from pytube import request
 
 
-@mock.patch('pytube.request.urlopen')
+@mock.patch("pytube.request.urlopen")
 def test_get_streaming(mock_urlopen):
     fake_stream_binary = [
         iter(os.urandom(8 * 1024)),
@@ -17,26 +17,26 @@ def test_get_streaming(mock_urlopen):
     response = mock.Mock()
     response.read.side_effect = fake_stream_binary
     mock_urlopen.return_value = response
-    response = request.get('http://fakeassurl.gov', streaming=True)
+    response = request.get("http://fakeassurl.gov", streaming=True)
     call_count = 0
     for i in response:
         call_count += 1
     assert call_count == 3
 
 
-@mock.patch('pytube.request.urlopen')
+@mock.patch("pytube.request.urlopen")
 def test_get_headers(mock_urlopen):
     response = mock.Mock()
-    response.info.return_value = {'content-length': '16384'}
+    response.info.return_value = {"content-length": "16384"}
     mock_urlopen.return_value = response
-    response = request.get('http://fakeassurl.gov', headers=True)
-    assert response == {'content-length': '16384'}
+    response = request.get("http://fakeassurl.gov", headers=True)
+    assert response == {"content-length": "16384"}
 
 
-@mock.patch('pytube.request.urlopen')
+@mock.patch("pytube.request.urlopen")
 def test_get(mock_urlopen):
     response = mock.Mock()
-    response.read.return_value = '<html></html>'.encode('utf-8')
+    response.read.return_value = "<html></html>".encode("utf-8")
     mock_urlopen.return_value = response
-    response = request.get('http://fakeassurl.gov')
-    assert response == '<html></html>'
+    response = request.get("http://fakeassurl.gov")
+    assert response == "<html></html>"

tests/test_streams.py

@@ -8,41 +8,41 @@ from pytube import Stream
 
 
 def test_filesize(cipher_signature, mocker):
-    mocker.patch.object(request, 'get')
-    request.get.return_value = {'content-length': '6796391'}
+    mocker.patch.object(request, "get")
+    request.get.return_value = {"content-length": "6796391"}
     assert cipher_signature.streams.first().filesize == 6796391
 
 
 def test_default_filename(cipher_signature):
-    expected = 'PSY - GANGNAM STYLE(강남스타일) MV.mp4'
+    expected = "PSY - GANGNAM STYLE(강남스타일) MV.mp4"
     stream = cipher_signature.streams.first()
     assert stream.default_filename == expected
 
 
 def test_title(cipher_signature):
-    expected = 'PSY - GANGNAM STYLE(강남스타일) M/V'
+    expected = "PSY - GANGNAM STYLE(강남스타일) M/V"
     stream = cipher_signature.streams.first()
     assert stream.title == expected
 
-    expected = 'PSY - GANGNAM STYLE(강남스타일)'
+    expected = "PSY - GANGNAM STYLE(강남스타일)"
     stream.player_config_args = {
-        'player_response': {'videoDetails': {'title': expected}},
+        "player_response": {"videoDetails": {"title": expected}},
     }
     assert stream.title == expected
 
-    expected = 'Unknown YouTube Video Title'
+    expected = "Unknown YouTube Video Title"
     stream.player_config_args = {}
     assert stream.title == expected
 
 
 def test_download(cipher_signature, mocker):
-    mocker.patch.object(request, 'get')
+    mocker.patch.object(request, "get")
     request.get.side_effect = [
-        {'content-length': '16384'},
-        {'content-length': '16384'},
+        {"content-length": "16384"},
+        {"content-length": "16384"},
         iter([str(random.getrandbits(8 * 1024))]),
     ]
-    with mock.patch('pytube.streams.open', mock.mock_open(), create=True):
+    with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
         stream = cipher_signature.streams.first()
         stream.download()
 
@@ -61,13 +61,13 @@ def test_on_progress_hook(cipher_signature, mocker):
     callback_fn = mock.MagicMock()
     cipher_signature.register_on_progress_callback(callback_fn)
 
-    mocker.patch.object(request, 'get')
+    mocker.patch.object(request, "get")
     request.get.side_effect = [
-        {'content-length': '16384'},
-        {'content-length': '16384'},
+        {"content-length": "16384"},
+        {"content-length": "16384"},
         iter([str(random.getrandbits(8 * 1024))]),
     ]
-    with mock.patch('pytube.streams.open', mock.mock_open(), create=True):
+    with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
         stream = cipher_signature.streams.first()
         stream.download()
     assert callback_fn.called
@@ -81,13 +81,13 @@ def test_on_complete_hook(cipher_signature, mocker):
     callback_fn = mock.MagicMock()
     cipher_signature.register_on_complete_callback(callback_fn)
 
-    mocker.patch.object(request, 'get')
+    mocker.patch.object(request, "get")
     request.get.side_effect = [
-        {'content-length': '16384'},
-        {'content-length': '16384'},
+        {"content-length": "16384"},
+        {"content-length": "16384"},
         iter([str(random.getrandbits(8 * 1024))]),
     ]
-    with mock.patch('pytube.streams.open', mock.mock_open(), create=True):
+    with mock.patch("pytube.streams.open", mock.mock_open(), create=True):
         stream = cipher_signature.streams.first()
         stream.download()
     assert callback_fn.called
@@ -96,8 +96,7 @@ def test_on_complete_hook(cipher_signature, mocker):
 def test_repr_for_audio_streams(cipher_signature):
     stream = str(cipher_signature.streams.filter(only_audio=True).first())
     expected = (
-        '<Stream: itag="140" mime_type="audio/mp4" abr="128kbps" '
-        'acodec="mp4a.40.2">'
+        '<Stream: itag="140" mime_type="audio/mp4" abr="128kbps" ' 'acodec="mp4a.40.2">'
     )
     assert stream == expected