[youtube, cleanup] Minor refactoring

[yt-dlp.git] / test / test_post_hooks.py

diff --git a/test/test_post_hooks.py b/test/test_post_hooks.py
index d8d2b36c311ec18ea283355cc34a5cd043155135..1555a23e0dfd617bdcfc35dc851918077042d3a0 100644 (file)
--- a/test/test_post_hooks.py
+++ b/test/test_post_hooks.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python
+#!/usr/bin/env python3
 
 from __future__ import unicode_literals
 
@@ -7,12 +7,12 @@
 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, try_rm, is_download_test
+import yt_dlp.YoutubeDL
+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)
         self.to_stderr = self.to_screen
@@ -22,6 +22,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