summaryrefslogtreecommitdiffhomepage
path: root/tests/subliminal_patch/test_video.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/subliminal_patch/test_video.py')
-rw-r--r--tests/subliminal_patch/test_video.py48
1 files changed, 48 insertions, 0 deletions
diff --git a/tests/subliminal_patch/test_video.py b/tests/subliminal_patch/test_video.py
new file mode 100644
index 000000000..44f63957a
--- /dev/null
+++ b/tests/subliminal_patch/test_video.py
@@ -0,0 +1,48 @@
+from subliminal import Episode, Movie
+from subliminal_patch.video import Video
+
+
+def test_video_fromguess_episode():
+ video = Video.fromguess(
+ "Breaking.Bad.S01E01.Bluray.mkv",
+ {"type": "episode", "streaming_service": "foo", "random_key": "bar"},
+ )
+ assert video.streaming_service == "foo"
+ assert video.other is None
+ assert isinstance(video, Episode)
+
+
+def test_video_fromguess_movie():
+ video = Video.fromguess(
+ "Taxi.Driver.1976.Bluray.mkv",
+ {"type": "movie", "edition": "foo", "random_key": "bar", "other": "Proper"},
+ )
+ assert video.edition == "foo"
+ assert video.other == "Proper"
+ assert isinstance(video, Movie)
+
+
+def test_video_fromname_episode():
+ video = Video.fromname("Breaking.Bad.S01E01.NF.WEB-DL.1080p.x264-FOO.mkv")
+
+ assert video.series == "Breaking Bad"
+ assert video.title is None
+ assert video.season == 1
+ assert video.episode == 1
+ assert video.source == "Web"
+ assert video.streaming_service == "Netflix"
+ assert video.resolution == "1080p"
+ assert video.video_codec == "H.264"
+ assert video.release_group == "FOO"
+
+
+def test_video_fromname_movie():
+ video = Video.fromname("Some.Flick.2022.UHD.Bluray.Proper.2160p.FLAC.HEVC-FOO.mkv")
+
+ assert video.source == "Ultra HD Blu-ray"
+ assert video.title == "Some Flick"
+ assert video.year == 2022
+ assert video.other == "Proper"
+ assert video.resolution == "2160p"
+ assert video.video_codec == "H.265"
+ assert video.audio_codec == "FLAC"