mirror of
https://github.com/morpheus65535/bazarr
synced 2024-12-27 01:57:33 +00:00
607 lines
36 KiB
Python
607 lines
36 KiB
Python
# -*- coding: utf-8 -*-
|
|
from datetime import timedelta, datetime
|
|
from enzyme.mkv import MKV, VIDEO_TRACK, AUDIO_TRACK, SUBTITLE_TRACK
|
|
import io
|
|
import os.path
|
|
import requests
|
|
import unittest
|
|
import zipfile
|
|
|
|
|
|
# Test directory
|
|
TEST_DIR = os.path.join(os.path.dirname(__file__), os.path.splitext(__file__)[0])
|
|
|
|
|
|
def setUpModule():
|
|
if not os.path.exists(TEST_DIR):
|
|
r = requests.get('http://downloads.sourceforge.net/project/matroska/test_files/matroska_test_w1_1.zip')
|
|
with zipfile.ZipFile(io.BytesIO(r.content), 'r') as f:
|
|
f.extractall(TEST_DIR)
|
|
|
|
|
|
class MKVTestCase(unittest.TestCase):
|
|
def test_test1(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test1.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(minutes=1, seconds=27, milliseconds=336))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2010, 8, 21, 7, 23, 3))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml2 v0.10.0 + libmatroska2 v0.10.1')
|
|
self.assertTrue(mkv.info.writing_app == 'mkclean 0.5.5 ru from libebml v1.0.0 + libmatroska v1.0.0 + mkvmerge v4.1.1 (\'Bouncin\' Back\') built on Jul 3 2010 22:54:08')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == True)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MS/VFW/FOURCC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 854)
|
|
self.assertTrue(mkv.video_tracks[0].height == 480)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_height is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio track
|
|
self.assertTrue(len(mkv.audio_tracks) == 1)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == True)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_MPEG/L3')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 0)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Big Buck Bunny - test 1')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File1, basic MPEG4.2 and MP3 with only SimpleBlock')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
def test_test2(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test2.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(seconds=47, milliseconds=509))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2011, 6, 2, 12, 45, 20))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml2 v0.21.0 + libmatroska2 v0.22.1')
|
|
self.assertTrue(mkv.info.writing_app == 'mkclean 0.8.3 ru from libebml2 v0.10.0 + libmatroska2 v0.10.1 + mkclean 0.5.5 ru from libebml v1.0.0 + libmatroska v1.0.0 + mkvmerge v4.1.1 (\'Bouncin\' Back\') built on Jul 3 2010 22:54:08')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == True)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MPEG4/ISO/AVC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 1024)
|
|
self.assertTrue(mkv.video_tracks[0].height == 576)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width == 1354)
|
|
self.assertTrue(mkv.video_tracks[0].display_height is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio track
|
|
self.assertTrue(len(mkv.audio_tracks) == 1)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == True)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_AAC')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 0)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Elephant Dream - test 2')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File 2, 100,000 timecode scale, odd aspect ratio, and CRC-32. Codecs are AVC and AAC')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
def test_test3(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test3.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(seconds=49, milliseconds=64))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2010, 8, 21, 21, 43, 25))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml2 v0.11.0 + libmatroska2 v0.10.1')
|
|
self.assertTrue(mkv.info.writing_app == 'mkclean 0.5.5 ro from libebml v1.0.0 + libmatroska v1.0.0 + mkvmerge v4.1.1 (\'Bouncin\' Back\') built on Jul 3 2010 22:54:08')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == True)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MPEG4/ISO/AVC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 1024)
|
|
self.assertTrue(mkv.video_tracks[0].height == 576)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_height is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio track
|
|
self.assertTrue(len(mkv.audio_tracks) == 1)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'eng')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == True)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_MPEG/L3')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 0)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Elephant Dream - test 3')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File 3, header stripping on the video track and no SimpleBlock')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
def test_test5(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test5.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(seconds=46, milliseconds=665))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2010, 8, 21, 18, 6, 43))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml v1.0.0 + libmatroska v1.0.0')
|
|
self.assertTrue(mkv.info.writing_app == 'mkvmerge v4.0.0 (\'The Stars were mine\') built on Jun 6 2010 16:18:42')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == True)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MPEG4/ISO/AVC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 1024)
|
|
self.assertTrue(mkv.video_tracks[0].height == 576)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width == 1024)
|
|
self.assertTrue(mkv.video_tracks[0].display_height == 576)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio tracks
|
|
self.assertTrue(len(mkv.audio_tracks) == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == True)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_AAC')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
self.assertTrue(mkv.audio_tracks[1].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[1].number == 10)
|
|
self.assertTrue(mkv.audio_tracks[1].name == 'Commentary')
|
|
self.assertTrue(mkv.audio_tracks[1].language == 'eng')
|
|
self.assertTrue(mkv.audio_tracks[1].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[1].default == False)
|
|
self.assertTrue(mkv.audio_tracks[1].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[1].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[1].codec_id == 'A_AAC')
|
|
self.assertTrue(mkv.audio_tracks[1].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[1].sampling_frequency == 22050.0)
|
|
self.assertTrue(mkv.audio_tracks[1].channels == 1)
|
|
self.assertTrue(mkv.audio_tracks[1].output_sampling_frequency == 44100.0)
|
|
self.assertTrue(mkv.audio_tracks[1].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 8)
|
|
self.assertTrue(mkv.subtitle_tracks[0].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[0].number == 3)
|
|
self.assertTrue(mkv.subtitle_tracks[0].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[0].language == 'eng')
|
|
self.assertTrue(mkv.subtitle_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[0].default == True)
|
|
self.assertTrue(mkv.subtitle_tracks[0].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[0].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[1].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[1].number == 4)
|
|
self.assertTrue(mkv.subtitle_tracks[1].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[1].language == 'hun')
|
|
self.assertTrue(mkv.subtitle_tracks[1].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[1].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[1].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[1].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[1].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[1].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[2].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[2].number == 5)
|
|
self.assertTrue(mkv.subtitle_tracks[2].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[2].language == 'ger')
|
|
self.assertTrue(mkv.subtitle_tracks[2].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[2].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[2].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[2].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[2].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[2].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[3].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[3].number == 6)
|
|
self.assertTrue(mkv.subtitle_tracks[3].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[3].language == 'fre')
|
|
self.assertTrue(mkv.subtitle_tracks[3].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[3].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[3].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[3].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[3].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[3].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[4].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[4].number == 8)
|
|
self.assertTrue(mkv.subtitle_tracks[4].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[4].language == 'spa')
|
|
self.assertTrue(mkv.subtitle_tracks[4].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[4].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[4].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[4].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[4].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[4].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[5].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[5].number == 9)
|
|
self.assertTrue(mkv.subtitle_tracks[5].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[5].language == 'ita')
|
|
self.assertTrue(mkv.subtitle_tracks[5].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[5].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[5].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[5].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[5].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[5].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[6].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[6].number == 11)
|
|
self.assertTrue(mkv.subtitle_tracks[6].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[6].language == 'jpn')
|
|
self.assertTrue(mkv.subtitle_tracks[6].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[6].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[6].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[6].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[6].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[6].codec_name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[7].type == SUBTITLE_TRACK)
|
|
self.assertTrue(mkv.subtitle_tracks[7].number == 7)
|
|
self.assertTrue(mkv.subtitle_tracks[7].name is None)
|
|
self.assertTrue(mkv.subtitle_tracks[7].language == 'und')
|
|
self.assertTrue(mkv.subtitle_tracks[7].enabled == True)
|
|
self.assertTrue(mkv.subtitle_tracks[7].default == False)
|
|
self.assertTrue(mkv.subtitle_tracks[7].forced == False)
|
|
self.assertTrue(mkv.subtitle_tracks[7].lacing == False)
|
|
self.assertTrue(mkv.subtitle_tracks[7].codec_id == 'S_TEXT/UTF8')
|
|
self.assertTrue(mkv.subtitle_tracks[7].codec_name is None)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Big Buck Bunny - test 8')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File 8, secondary audio commentary track, misc subtitle tracks')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
def test_test6(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test6.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(seconds=87, milliseconds=336))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2010, 8, 21, 16, 31, 55))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml2 v0.10.1 + libmatroska2 v0.10.1')
|
|
self.assertTrue(mkv.info.writing_app == 'mkclean 0.5.5 r from libebml v1.0.0 + libmatroska v1.0.0 + mkvmerge v4.0.0 (\'The Stars were mine\') built on Jun 6 2010 16:18:42')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == False)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MS/VFW/FOURCC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 854)
|
|
self.assertTrue(mkv.video_tracks[0].height == 480)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_height is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio track
|
|
self.assertTrue(len(mkv.audio_tracks) == 1)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == False)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_MPEG/L3')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 0)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Big Buck Bunny - test 6')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File 6, random length to code the size of Clusters and Blocks, no Cues for seeking')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
def test_test7(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test7.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(seconds=37, milliseconds=43))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2010, 8, 21, 17, 0, 23))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml2 v0.10.1 + libmatroska2 v0.10.1')
|
|
self.assertTrue(mkv.info.writing_app == 'mkclean 0.5.5 r from libebml v1.0.0 + libmatroska v1.0.0 + mkvmerge v4.0.0 (\'The Stars were mine\') built on Jun 6 2010 16:18:42')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == False)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MPEG4/ISO/AVC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 1024)
|
|
self.assertTrue(mkv.video_tracks[0].height == 576)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_height is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio track
|
|
self.assertTrue(len(mkv.audio_tracks) == 1)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == False)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_AAC')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 0)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Big Buck Bunny - test 7')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File 7, junk elements are present at the beggining or end of clusters, the parser should skip it. There is also a damaged element at 451418')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
def test_test8(self):
|
|
stream = io.open(os.path.join(TEST_DIR, 'test8.mkv'), 'rb')
|
|
mkv = MKV(stream)
|
|
# info
|
|
self.assertTrue(mkv.info.title is None)
|
|
self.assertTrue(mkv.info.duration == timedelta(seconds=47, milliseconds=341))
|
|
self.assertTrue(mkv.info.date_utc == datetime(2010, 8, 21, 17, 22, 14))
|
|
self.assertTrue(mkv.info.muxing_app == 'libebml2 v0.10.1 + libmatroska2 v0.10.1')
|
|
self.assertTrue(mkv.info.writing_app == 'mkclean 0.5.5 r from libebml v1.0.0 + libmatroska v1.0.0 + mkvmerge v4.0.0 (\'The Stars were mine\') built on Jun 6 2010 16:18:42')
|
|
# video track
|
|
self.assertTrue(len(mkv.video_tracks) == 1)
|
|
self.assertTrue(mkv.video_tracks[0].type == VIDEO_TRACK)
|
|
self.assertTrue(mkv.video_tracks[0].number == 1)
|
|
self.assertTrue(mkv.video_tracks[0].name is None)
|
|
self.assertTrue(mkv.video_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.video_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.video_tracks[0].default == False)
|
|
self.assertTrue(mkv.video_tracks[0].forced == False)
|
|
self.assertTrue(mkv.video_tracks[0].lacing == False)
|
|
self.assertTrue(mkv.video_tracks[0].codec_id == 'V_MPEG4/ISO/AVC')
|
|
self.assertTrue(mkv.video_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.video_tracks[0].width == 1024)
|
|
self.assertTrue(mkv.video_tracks[0].height == 576)
|
|
self.assertTrue(mkv.video_tracks[0].interlaced == False)
|
|
self.assertTrue(mkv.video_tracks[0].stereo_mode is None)
|
|
self.assertTrue(mkv.video_tracks[0].crop == {})
|
|
self.assertTrue(mkv.video_tracks[0].display_width is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_height is None)
|
|
self.assertTrue(mkv.video_tracks[0].display_unit is None)
|
|
self.assertTrue(mkv.video_tracks[0].aspect_ratio_type is None)
|
|
# audio track
|
|
self.assertTrue(len(mkv.audio_tracks) == 1)
|
|
self.assertTrue(mkv.audio_tracks[0].type == AUDIO_TRACK)
|
|
self.assertTrue(mkv.audio_tracks[0].number == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].language == 'und')
|
|
self.assertTrue(mkv.audio_tracks[0].enabled == True)
|
|
self.assertTrue(mkv.audio_tracks[0].default == False)
|
|
self.assertTrue(mkv.audio_tracks[0].forced == False)
|
|
self.assertTrue(mkv.audio_tracks[0].lacing == True)
|
|
self.assertTrue(mkv.audio_tracks[0].codec_id == 'A_AAC')
|
|
self.assertTrue(mkv.audio_tracks[0].codec_name is None)
|
|
self.assertTrue(mkv.audio_tracks[0].sampling_frequency == 48000.0)
|
|
self.assertTrue(mkv.audio_tracks[0].channels == 2)
|
|
self.assertTrue(mkv.audio_tracks[0].output_sampling_frequency is None)
|
|
self.assertTrue(mkv.audio_tracks[0].bit_depth is None)
|
|
# subtitle track
|
|
self.assertTrue(len(mkv.subtitle_tracks) == 0)
|
|
# chapters
|
|
self.assertTrue(len(mkv.chapters) == 0)
|
|
# tags
|
|
self.assertTrue(len(mkv.tags) == 1)
|
|
self.assertTrue(len(mkv.tags[0].simpletags) == 3)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].name == 'TITLE')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[0].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].string == 'Big Buck Bunny - test 8')
|
|
self.assertTrue(mkv.tags[0].simpletags[0].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].name == 'DATE_RELEASED')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[1].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].string == '2010')
|
|
self.assertTrue(mkv.tags[0].simpletags[1].binary is None)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].name == 'COMMENT')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].default == True)
|
|
self.assertTrue(mkv.tags[0].simpletags[2].language == 'und')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].string == 'Matroska Validation File 8, audio missing between timecodes 6.019s and 6.360s')
|
|
self.assertTrue(mkv.tags[0].simpletags[2].binary is None)
|
|
|
|
|
|
def suite():
|
|
suite = unittest.TestSuite()
|
|
suite.addTest(unittest.TestLoader().loadTestsFromTestCase(MKVTestCase))
|
|
return suite
|
|
|
|
if __name__ == '__main__':
|
|
unittest.TextTestRunner().run(suite())
|