X-Git-Url: https://jfr.im/git/yt-dlp.git/blobdiff_plain/ab8e5e516f38c3eab8947614e2347a2473e5dbbc..8e15177b4113c355989881e4e030f695a9b59c3a:/test/test_post_hooks.py

diff --git a/test/test_post_hooks.py b/test/test_post_hooks.py
index d8d2b36c3..3778d1794 100644
--- a/test/test_post_hooks.py
+++ b/test/test_post_hooks.py
@@ -1,20 +1,21 @@
-#!/usr/bin/env python
-
-from __future__ import unicode_literals
+#!/usr/bin/env python3
 
+# 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 youtube_dl.YoutubeDL
-from youtube_dl.utils import DownloadError
+
+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(youtube_dl.YoutubeDL):
+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