def test_sanity(self):
mp4filename = os.path.join(TESTDATA, 'loop_circle.mp4')
with open(mp4filename, 'rb') as mp4file:
isofile = isomedia.load(mp4file)
moov_atom = [atom for atom in isofile.atoms if atom.type == 'moov']
self.assertEqual(len(moov_atom), 1, 'there should be 1 moov atom.')
def test_fail_to_generic_atom(self):
mp4filename = os.path.join(TESTDATA, 'broken_mvhd.mp4')
with open(mp4filename, 'rb') as infile:
isofile = isomedia.load(infile)
self.assertTrue(isofile.atoms)
self.assertEqual(len(isofile.atoms), 1)
self.assertTrue(isinstance(isofile.atoms[0], GenericAtom))
def test_lossless_write(self):
mp4filename = os.path.join(TESTDATA, 'loop_circle.mp4')
with open(mp4filename, 'rb') as infile, tempfile.NamedTemporaryFile(delete=False) as outfile:
isofile = isomedia.load(infile)
isofile.write(outfile)
self.assertTrue(filecmp.cmp(infile.name, outfile.name))
os.remove(outfile.name)
def test_meta_with_children(self):
mp4filename = os.path.join(TESTDATA, 'meta_with_children.mp4')
with open(mp4filename, 'rb') as infile:
isofile = isomedia.load(infile)
moov = [atom for atom in isofile.atoms if atom.type == 'moov'][0]
udta = [atom for atom in moov.children if atom.type == 'udta'][0]
meta = [atom for atom in udta.children if atom.type == 'meta'][0]
ilst = [atom for atom in meta.children if atom.type == 'ilst'][0]
cust = [atom for atom in ilst.children if atom.type == '----'][0]
def test_property_change(self):
mp4filename = os.path.join(TESTDATA, 'guitar.mp4')
faked_duration = 424242
with open(mp4filename, 'rb') as infile, tempfile.NamedTemporaryFile(delete=False) as outfile:
isofile = isomedia.load(infile)
moov = [atom for atom in isofile.atoms if atom.type == 'moov'][0]
mvhd = [atom for atom in moov.children if atom.type == 'mvhd'][0]
mvhd.properties['duration'] = faked_duration
isofile.write(outfile)
reopenedfile = open(outfile.name, 'rb')
reopenediso = isomedia.load(reopenedfile)
new_moov = [atom for atom in reopenediso.atoms if atom.type == 'moov'][0]
new_mvhd = [atom for atom in new_moov.children if atom.type == 'mvhd'][0]
self.assertEqual(new_mvhd.properties['duration'], faked_duration)
reopenedfile.close()
os.remove(outfile.name)