diff --git a/tests/integration/prebuilt_presets/test_prebuilt_presets.py b/tests/integration/prebuilt_presets/test_prebuilt_presets.py index 29f38349..dfa98848 100644 --- a/tests/integration/prebuilt_presets/test_prebuilt_presets.py +++ b/tests/integration/prebuilt_presets/test_prebuilt_presets.py @@ -3,7 +3,6 @@ from pathlib import Path from typing import Dict from typing import List - import pytest from expected_download import assert_expected_downloads from expected_transaction_log import assert_transaction_log_matches @@ -17,7 +16,6 @@ from ytdl_sub.subscriptions.subscription import Subscription from ytdl_sub.utils.exceptions import ValidationException - @pytest.mark.parametrize("media_player_preset", TvShowCollectionPresets.preset_names) @pytest.mark.parametrize( "tv_show_structure_preset", TvShowCollectionEpisodeFormattingPresets.preset_names diff --git a/tests/integration/prebuilt_presets/test_tv_show_by_date.py b/tests/integration/prebuilt_presets/test_tv_show_by_date.py index 6d949137..3dfe1d6e 100644 --- a/tests/integration/prebuilt_presets/test_tv_show_by_date.py +++ b/tests/integration/prebuilt_presets/test_tv_show_by_date.py @@ -1,10 +1,10 @@ import pytest - from expected_download import assert_expected_downloads from expected_transaction_log import assert_transaction_log_matches + from ytdl_sub.config.config_file import ConfigFile -from ytdl_sub.prebuilt_presets.tv_show import TvShowByDateEpisodeFormattingPresets, \ - TvShowByDatePresets +from ytdl_sub.prebuilt_presets.tv_show import TvShowByDateEpisodeFormattingPresets +from ytdl_sub.prebuilt_presets.tv_show import TvShowByDatePresets from ytdl_sub.subscriptions.subscription import Subscription DEPRECATED_TV_SHOW_PRESET_EQUIVALENTS = { @@ -18,6 +18,7 @@ DEPRECATED_TV_SHOW_PRESET_EQUIVALENTS = { DEFAULT_TV_SHOW_STRUCTURE = "season_by_year__episode_by_month_day" + class TestPrebuiltTVShowPresets: @staticmethod @@ -119,9 +120,7 @@ class TestPrebuiltTVShowPresets: tv_show_preset: str, ): - with mock_download_collection_entries( - is_youtube_channel=True - ): + with mock_download_collection_entries(is_youtube_channel=True): self.run( config=config, subscription_name=subscription_name, @@ -130,7 +129,9 @@ class TestPrebuiltTVShowPresets: episode_ordering_preset=DEFAULT_TV_SHOW_STRUCTURE, ) - @pytest.mark.parametrize("episode_ordering_preset", TvShowByDateEpisodeFormattingPresets.preset_names) + @pytest.mark.parametrize( + "episode_ordering_preset", TvShowByDateEpisodeFormattingPresets.preset_names + ) def test_episode_ordering_presets( self, config, @@ -149,4 +150,4 @@ class TestPrebuiltTVShowPresets: output_directory=output_directory, tv_show_preset="Kodi TV Show by Date", episode_ordering_preset=episode_ordering_preset, - ) \ No newline at end of file + )