X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/cc52de43568d8cd58c7e2ef4e5cecf609da28a9c..9d7ded6419089c1bf252496073f73ad90ed71004:/test/test_post_hooks.py

diff --git a/test/test_post_hooks.py b/test/test_post_hooks.py
index 3f9a61c1e..3778d1794 100644
--- a/test/test_post_hooks.py
+++ b/test/test_post_hooks.py
@@ -1,20 +1,21 @@
 #!/usr/bin/env python3
 
-from __future__ import unicode_literals
-
+# Allow direct execution
 import os
 import sys
 import unittest
+
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from test.helper import get_params, try_rm
-import yt_dlp.YoutubeDL
+
+from test.helper import get_params, is_download_test, try_rm
+import yt_dlp.YoutubeDL  # isort: split
 from yt_dlp.utils import DownloadError
 
 
 class YoutubeDL(yt_dlp.YoutubeDL):
     def __init__(self, *args, **kwargs):
-        super(YoutubeDL, self).__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)
         self.to_stderr = self.to_screen
 
 
@@ -22,6 +23,7 @@ def __init__(self, *args, **kwargs):
 EXPECTED_NAME = 'gr51aVj-mLg'
 
 
+@is_download_test
 class TestPostHooks(unittest.TestCase):
     def setUp(self):
         self.stored_name_1 = None