From 796447b1a564ac3452274c0a10acd2923d14f028 Mon Sep 17 00:00:00 2001 From: Alex Merkel Date: Tue, 29 Dec 2020 16:09:32 +0100 Subject: [PATCH] Beautify test code --- test/test_post_hooks.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/test/test_post_hooks.py b/test/test_post_hooks.py index 8a30f54a2..d8d2b36c3 100644 --- a/test/test_post_hooks.py +++ b/test/test_post_hooks.py @@ -11,14 +11,17 @@ from test.helper import get_params, try_rm import youtube_dl.YoutubeDL from youtube_dl.utils import DownloadError + class YoutubeDL(youtube_dl.YoutubeDL): def __init__(self, *args, **kwargs): super(YoutubeDL, self).__init__(*args, **kwargs) self.to_stderr = self.to_screen + TEST_ID = 'gr51aVj-mLg' EXPECTED_NAME = 'gr51aVj-mLg' + class TestPostHooks(unittest.TestCase): def setUp(self): self.stored_name_1 = None @@ -44,15 +47,15 @@ class TestPostHooks(unittest.TestCase): ydl = YoutubeDL(self.params) self.assertRaises(DownloadError, ydl.download, [TEST_ID]) - def hook_one(self,filename): + def hook_one(self, filename): self.stored_name_1, _ = os.path.splitext(os.path.basename(filename)) self.files.append(filename) - def hook_two(self,filename): + def hook_two(self, filename): self.stored_name_2, _ = os.path.splitext(os.path.basename(filename)) self.files.append(filename) - def hook_three(self,filename): + def hook_three(self, filename): self.files.append(filename) raise Exception('Test exception for \'%s\'' % filename) @@ -60,5 +63,6 @@ class TestPostHooks(unittest.TestCase): for f in self.files: try_rm(f) + if __name__ == '__main__': unittest.main()