Vitiko
2 years ago
parent
58868cff17
commit
c06dd620b7
@ -0,0 +1,229 @@
|
||||
import logging
|
||||
|
||||
import pytest
|
||||
|
||||
from bazarr.utilities import video_analyzer
|
||||
|
||||
logging.getLogger("knowit").setLevel(logging.WARNING)
|
||||
|
||||
M_INFO = {
|
||||
"creatingLibrary": {
|
||||
"name": "MediaInfoLib",
|
||||
"version": "23.03",
|
||||
"url": "https://mediaarea.net/MediaInfo",
|
||||
},
|
||||
"media": {
|
||||
"@ref": "/mnt/media/Hocus.Pocus.1993.1080p.DSNP.WEB-DL.DDP5.1.H.264.DUAL-PD.mkv",
|
||||
"track": [
|
||||
{
|
||||
"@type": "General",
|
||||
"UniqueID": "177986280948425821736023466260510750529",
|
||||
"VideoCount": "1",
|
||||
"AudioCount": "2",
|
||||
"TextCount": "31",
|
||||
"FileExtension": "mkv",
|
||||
"Format": "Matroska",
|
||||
"Format_Version": "4",
|
||||
"FileSize": "6468058376",
|
||||
"Duration": "5766.219",
|
||||
"OverallBitRate_Mode": "VBR",
|
||||
"OverallBitRate": "8973726",
|
||||
"FrameRate": "23.976",
|
||||
"FrameCount": "138251",
|
||||
"StreamSize": "2607619",
|
||||
"IsStreamable": "Yes",
|
||||
"Encoded_Date": "2023-04-22 16:46:57 UTC",
|
||||
"File_Modified_Date": "2023-05-17 01:49:55 UTC",
|
||||
"File_Modified_Date_Local": "2023-05-16 21:49:55",
|
||||
"Encoded_Application": "mkvmerge v75.0.0 ('Goliath') 64-bit",
|
||||
"Encoded_Library": "libebml v1.4.4 + libmatroska v1.7.1",
|
||||
},
|
||||
{
|
||||
"@type": "Video",
|
||||
"StreamOrder": "0",
|
||||
"ID": "1",
|
||||
"UniqueID": "9393509843335289949",
|
||||
"Format": "AVC",
|
||||
"Format_Profile": "High",
|
||||
"Format_Level": "4",
|
||||
"Format_Settings_CABAC": "Yes",
|
||||
"Format_Settings_RefFrames": "4",
|
||||
"CodecID": "V_MPEG4/ISO/AVC",
|
||||
"Duration": "5766.219000000",
|
||||
"BitRate_Mode": "VBR",
|
||||
"BitRate": "8458733",
|
||||
"BitRate_Maximum": "12749952",
|
||||
"Width": "1920",
|
||||
"Height": "1080",
|
||||
"Stored_Height": "1088",
|
||||
"Sampled_Width": "1920",
|
||||
"Sampled_Height": "1080",
|
||||
"PixelAspectRatio": "1.000",
|
||||
"DisplayAspectRatio": "1.778",
|
||||
"FrameRate_Mode": "CFR",
|
||||
"FrameRate": "23.976",
|
||||
"FrameCount": "138251",
|
||||
"ColorSpace": "YUV",
|
||||
"ChromaSubsampling": "4:2:0",
|
||||
"BitDepth": "8",
|
||||
"ScanType": "Progressive",
|
||||
"Delay": "0.000",
|
||||
"Delay_Source": "Container",
|
||||
"StreamSize": "6096863666",
|
||||
"Default": "Yes",
|
||||
"Forced": "No",
|
||||
"BufferSize": "17000000",
|
||||
"colour_description_present": "Yes",
|
||||
"colour_description_present_Source": "Container / Stream",
|
||||
"colour_range": "Limited",
|
||||
"colour_range_Source": "Stream",
|
||||
"colour_primaries": "BT.709",
|
||||
"colour_primaries_Source": "Container / Stream",
|
||||
"transfer_characteristics": "BT.709",
|
||||
"transfer_characteristics_Source": "Container / Stream",
|
||||
"matrix_coefficients": "BT.709",
|
||||
"matrix_coefficients_Source": "Container / Stream",
|
||||
},
|
||||
{
|
||||
"@type": "Text",
|
||||
"@typeorder": "7",
|
||||
"StreamOrder": "9",
|
||||
"ID": "10",
|
||||
"UniqueID": "2233390560797234737",
|
||||
"Format": "UTF-8",
|
||||
"CodecID": "S_TEXT/UTF8",
|
||||
"Duration": "5480.360000000",
|
||||
"BitRate": "45",
|
||||
"FrameRate": "0.206",
|
||||
"FrameCount": "1129",
|
||||
"ElementCount": "1129",
|
||||
"StreamSize": "31194",
|
||||
"Language": "es-419",
|
||||
"Default": "No",
|
||||
"Forced": "No",
|
||||
},
|
||||
{
|
||||
"@type": "Text",
|
||||
"@typeorder": "9",
|
||||
"StreamOrder": "11",
|
||||
"ID": "12",
|
||||
"UniqueID": "1345374948683222936",
|
||||
"Format": "UTF-8",
|
||||
"CodecID": "S_TEXT/UTF8",
|
||||
"Duration": "5561.600000000",
|
||||
"BitRate": "46",
|
||||
"FrameRate": "0.164",
|
||||
"FrameCount": "914",
|
||||
"ElementCount": "914",
|
||||
"StreamSize": "32145",
|
||||
"Language": "es-ES",
|
||||
"Default": "No",
|
||||
"Forced": "No",
|
||||
},
|
||||
{
|
||||
"@type": "Text",
|
||||
"@typeorder": "11",
|
||||
"StreamOrder": "13",
|
||||
"ID": "14",
|
||||
"UniqueID": "17039172451186467602",
|
||||
"Format": "UTF-8",
|
||||
"CodecID": "S_TEXT/UTF8",
|
||||
"Duration": "4966.120000000",
|
||||
"BitRate": "1",
|
||||
"FrameRate": "0.007",
|
||||
"FrameCount": "35",
|
||||
"ElementCount": "35",
|
||||
"StreamSize": "1011",
|
||||
"Language": "fr-CA",
|
||||
"Default": "No",
|
||||
"Forced": "No",
|
||||
},
|
||||
{
|
||||
"@type": "Text",
|
||||
"@typeorder": "24",
|
||||
"StreamOrder": "26",
|
||||
"ID": "27",
|
||||
"UniqueID": "16221047442617815320",
|
||||
"Format": "UTF-8",
|
||||
"CodecID": "S_TEXT/UTF8",
|
||||
"Duration": "4961.520000000",
|
||||
"BitRate": "0",
|
||||
"FrameRate": "0.002",
|
||||
"FrameCount": "11",
|
||||
"ElementCount": "11",
|
||||
"StreamSize": "379",
|
||||
"Language": "pt-BR",
|
||||
"Default": "No",
|
||||
"Forced": "No",
|
||||
},
|
||||
{
|
||||
"@type": "Text",
|
||||
"@typeorder": "30",
|
||||
"StreamOrder": "32",
|
||||
"ID": "33",
|
||||
"UniqueID": "4259582444071016270",
|
||||
"Format": "UTF-8",
|
||||
"CodecID": "S_TEXT/UTF8",
|
||||
"Duration": "5507.508000000",
|
||||
"BitRate": "50",
|
||||
"FrameRate": "0.253",
|
||||
"FrameCount": "1392",
|
||||
"ElementCount": "1392",
|
||||
"StreamSize": "34539",
|
||||
"Language": "zh-Hans",
|
||||
"Default": "No",
|
||||
"Forced": "No",
|
||||
},
|
||||
{
|
||||
"@type": "Text",
|
||||
"@typeorder": "31",
|
||||
"StreamOrder": "33",
|
||||
"ID": "34",
|
||||
"UniqueID": "4890027048965677919",
|
||||
"Format": "UTF-8",
|
||||
"CodecID": "S_TEXT/UTF8",
|
||||
"Duration": "5730.725000000",
|
||||
"BitRate": "43",
|
||||
"FrameRate": "0.207",
|
||||
"FrameCount": "1186",
|
||||
"ElementCount": "1186",
|
||||
"StreamSize": "31154",
|
||||
"Language": "zh-Hant",
|
||||
"Default": "No",
|
||||
"Forced": "No",
|
||||
},
|
||||
],
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def video_file():
|
||||
return "tests/subliminal_patch/data/file_1.mkv"
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def mediainfo_data(mocker, video_file):
|
||||
mocker.patch(
|
||||
"knowit.providers.mediainfo.MediaInfoCTypesExecutor._execute",
|
||||
return_value=M_INFO,
|
||||
)
|
||||
data = video_analyzer.know(
|
||||
video_path=video_file,
|
||||
context={"provider": "mediainfo"},
|
||||
)
|
||||
yield data
|
||||
|
||||
|
||||
def test_embedded_subs_reader(mocker, mediainfo_data, video_file):
|
||||
mocker.patch(
|
||||
"bazarr.utilities.video_analyzer.parse_video_metadata",
|
||||
return_value={"mediainfo": mediainfo_data},
|
||||
)
|
||||
mocker.patch(
|
||||
"bazarr.utilities.video_analyzer.alpha3_from_alpha2", return_value=None
|
||||
)
|
||||
result = video_analyzer.embedded_subs_reader(1e6, video_file)
|
||||
assert ["spl", False, False, "SubRip"] in result
|
||||
assert ["pob", False, False, "SubRip"] in result
|
||||
assert ["zht", False, False, "SubRip"] in result
|
||||
Loading…
Reference in new issue