From 24940c0de76aa8fcb2797024511652b99d7c3557 Mon Sep 17 00:00:00 2001 From: Adrian Sampson Date: Sat, 26 Nov 2016 18:27:20 -0500 Subject: [PATCH] In MediaFile tests, use simpler import path This will make it easier to copy these tests into their own repository. --- test/test_mediafile_edge.py | 76 ++++++++++++++++++------------------- 1 file changed, 38 insertions(+), 38 deletions(-) diff --git a/test/test_mediafile_edge.py b/test/test_mediafile_edge.py index af3bdfa6b..81a6c4416 100644 --- a/test/test_mediafile_edge.py +++ b/test/test_mediafile_edge.py @@ -25,11 +25,11 @@ from test import _common from test._common import unittest from test.helper import TestHelper -import beets.mediafile +from beets import mediafile import six -_sc = beets.mediafile._safe_cast +_sc = mediafile._safe_cast class EdgeTest(unittest.TestCase): @@ -37,7 +37,7 @@ class EdgeTest(unittest.TestCase): # Some files have an ID3 frame that has a list with no elements. # This is very hard to produce, so this is just the first 8192 # bytes of a file found "in the wild". - emptylist = beets.mediafile.MediaFile( + emptylist = mediafile.MediaFile( os.path.join(_common.RSRC, b'emptylist.mp3') ) genre = emptylist.genre @@ -46,7 +46,7 @@ class EdgeTest(unittest.TestCase): def test_release_time_with_space(self): # Ensures that release times delimited by spaces are ignored. # Amie Street produces such files. - space_time = beets.mediafile.MediaFile( + space_time = mediafile.MediaFile( os.path.join(_common.RSRC, b'space_time.mp3') ) self.assertEqual(space_time.year, 2009) @@ -56,7 +56,7 @@ class EdgeTest(unittest.TestCase): def test_release_time_with_t(self): # Ensures that release times delimited by Ts are ignored. # The iTunes Store produces such files. - t_time = beets.mediafile.MediaFile( + t_time = mediafile.MediaFile( os.path.join(_common.RSRC, b't_time.m4a') ) self.assertEqual(t_time.year, 1987) @@ -66,20 +66,20 @@ class EdgeTest(unittest.TestCase): def test_tempo_with_bpm(self): # Some files have a string like "128 BPM" in the tempo field # rather than just a number. - f = beets.mediafile.MediaFile(os.path.join(_common.RSRC, b'bpm.mp3')) + f = mediafile.MediaFile(os.path.join(_common.RSRC, b'bpm.mp3')) self.assertEqual(f.bpm, 128) def test_discc_alternate_field(self): # Different taggers use different vorbis comments to reflect # the disc and disc count fields: ensure that the alternative # style works. - f = beets.mediafile.MediaFile(os.path.join(_common.RSRC, b'discc.ogg')) + f = mediafile.MediaFile(os.path.join(_common.RSRC, b'discc.ogg')) self.assertEqual(f.disc, 4) self.assertEqual(f.disctotal, 5) def test_old_ape_version_bitrate(self): media_file = os.path.join(_common.RSRC, b'oldape.ape') - f = beets.mediafile.MediaFile(media_file) + f = mediafile.MediaFile(media_file) self.assertEqual(f.bitrate, 0) def test_only_magic_bytes_jpeg(self): @@ -91,13 +91,13 @@ class EdgeTest(unittest.TestCase): with open(magic_bytes_file, 'rb') as f: jpg_data = f.read() self.assertEqual( - beets.mediafile._imghdr_what_wrapper(jpg_data), 'jpeg') + mediafile._imghdr_what_wrapper(jpg_data), 'jpeg') def test_soundcheck_non_ascii(self): # Make sure we don't crash when the iTunes SoundCheck field contains # non-ASCII binary data. - f = beets.mediafile.MediaFile(os.path.join(_common.RSRC, - b'soundcheck-nonascii.m4a')) + f = mediafile.MediaFile(os.path.join(_common.RSRC, + b'soundcheck-nonascii.m4a')) self.assertEqual(f.rg_track_gain, 0.0) @@ -157,35 +157,35 @@ class SafetyTest(unittest.TestCase, TestHelper): with open(fn, 'w') as f: f.write(data) try: - self.assertRaises(exc, beets.mediafile.MediaFile, fn) + self.assertRaises(exc, mediafile.MediaFile, fn) finally: os.unlink(fn) # delete the temporary file def test_corrupt_mp3_raises_unreadablefileerror(self): # Make sure we catch Mutagen reading errors appropriately. - self._exccheck(b'corrupt.mp3', beets.mediafile.UnreadableFileError) + self._exccheck(b'corrupt.mp3', mediafile.UnreadableFileError) def test_corrupt_mp4_raises_unreadablefileerror(self): - self._exccheck(b'corrupt.m4a', beets.mediafile.UnreadableFileError) + self._exccheck(b'corrupt.m4a', mediafile.UnreadableFileError) def test_corrupt_flac_raises_unreadablefileerror(self): - self._exccheck(b'corrupt.flac', beets.mediafile.UnreadableFileError) + self._exccheck(b'corrupt.flac', mediafile.UnreadableFileError) def test_corrupt_ogg_raises_unreadablefileerror(self): - self._exccheck(b'corrupt.ogg', beets.mediafile.UnreadableFileError) + self._exccheck(b'corrupt.ogg', mediafile.UnreadableFileError) def test_invalid_ogg_header_raises_unreadablefileerror(self): - self._exccheck(b'corrupt.ogg', beets.mediafile.UnreadableFileError, + self._exccheck(b'corrupt.ogg', mediafile.UnreadableFileError, 'OggS\x01vorbis') def test_corrupt_monkeys_raises_unreadablefileerror(self): - self._exccheck(b'corrupt.ape', beets.mediafile.UnreadableFileError) + self._exccheck(b'corrupt.ape', mediafile.UnreadableFileError) def test_invalid_extension_raises_filetypeerror(self): - self._exccheck(b'something.unknown', beets.mediafile.FileTypeError) + self._exccheck(b'something.unknown', mediafile.FileTypeError) def test_magic_xml_raises_unreadablefileerror(self): - self._exccheck(b'nothing.xml', beets.mediafile.UnreadableFileError, + self._exccheck(b'nothing.xml', mediafile.UnreadableFileError, "ftyp") @unittest.skipUnless(_common.HAVE_SYMLINK, u'platform lacks symlink') @@ -193,8 +193,8 @@ class SafetyTest(unittest.TestCase, TestHelper): fn = os.path.join(_common.RSRC, b'brokenlink') os.symlink('does_not_exist', fn) try: - self.assertRaises(beets.mediafile.UnreadableFileError, - beets.mediafile.MediaFile, fn) + self.assertRaises(mediafile.UnreadableFileError, + mediafile.MediaFile, fn) finally: os.unlink(fn) @@ -205,7 +205,7 @@ class SideEffectsTest(unittest.TestCase): def test_opening_tagless_file_leaves_untouched(self): old_mtime = os.stat(self.empty).st_mtime - beets.mediafile.MediaFile(self.empty) + mediafile.MediaFile(self.empty) new_mtime = os.stat(self.empty).st_mtime self.assertEqual(old_mtime, new_mtime) @@ -217,7 +217,7 @@ class MP4EncodingTest(unittest.TestCase, TestHelper): self.path = os.path.join(self.temp_dir, b'test.m4a') shutil.copy(src, self.path) - self.mf = beets.mediafile.MediaFile(self.path) + self.mf = mediafile.MediaFile(self.path) def tearDown(self): self.remove_temp_dir() @@ -225,7 +225,7 @@ class MP4EncodingTest(unittest.TestCase, TestHelper): def test_unicode_label_in_m4a(self): self.mf.label = u'foo\xe8bar' self.mf.save() - new_mf = beets.mediafile.MediaFile(self.path) + new_mf = mediafile.MediaFile(self.path) self.assertEqual(new_mf.label, u'foo\xe8bar') @@ -236,7 +236,7 @@ class MP3EncodingTest(unittest.TestCase, TestHelper): self.path = os.path.join(self.temp_dir, b'test.mp3') shutil.copy(src, self.path) - self.mf = beets.mediafile.MediaFile(self.path) + self.mf = mediafile.MediaFile(self.path) def test_comment_with_latin1_encoding(self): # Set up the test file with a Latin1-encoded COMM frame. The encoding @@ -249,7 +249,7 @@ class MP3EncodingTest(unittest.TestCase, TestHelper): self.mf.save() -class ZeroLengthMediaFile(beets.mediafile.MediaFile): +class ZeroLengthMediaFile(mediafile.MediaFile): @property def length(self): return 0.0 @@ -270,7 +270,7 @@ class TypeTest(unittest.TestCase): def setUp(self): super(TypeTest, self).setUp() path = os.path.join(_common.RSRC, b'full.mp3') - self.mf = beets.mediafile.MediaFile(path) + self.mf = mediafile.MediaFile(path) def test_year_integer_in_string(self): self.mf.year = u'2009' @@ -302,32 +302,32 @@ class TypeTest(unittest.TestCase): class SoundCheckTest(unittest.TestCase): def test_round_trip(self): - data = beets.mediafile._sc_encode(1.0, 1.0) - gain, peak = beets.mediafile._sc_decode(data) + data = mediafile._sc_encode(1.0, 1.0) + gain, peak = mediafile._sc_decode(data) self.assertEqual(gain, 1.0) self.assertEqual(peak, 1.0) def test_decode_zero(self): data = b' 80000000 80000000 00000000 00000000 00000000 00000000 ' \ b'00000000 00000000 00000000 00000000' - gain, peak = beets.mediafile._sc_decode(data) + gain, peak = mediafile._sc_decode(data) self.assertEqual(gain, 0.0) self.assertEqual(peak, 0.0) def test_malformatted(self): - gain, peak = beets.mediafile._sc_decode(b'foo') + gain, peak = mediafile._sc_decode(b'foo') self.assertEqual(gain, 0.0) self.assertEqual(peak, 0.0) def test_special_characters(self): - gain, peak = beets.mediafile._sc_decode(u'caf\xe9'.encode('utf-8')) + gain, peak = mediafile._sc_decode(u'caf\xe9'.encode('utf-8')) self.assertEqual(gain, 0.0) self.assertEqual(peak, 0.0) def test_decode_handles_unicode(self): # Most of the time, we expect to decode the raw bytes. But some formats # might give us text strings, which we need to handle. - gain, peak = beets.mediafile._sc_decode(u'caf\xe9') + gain, peak = mediafile._sc_decode(u'caf\xe9') self.assertEqual(gain, 0.0) self.assertEqual(peak, 0.0) @@ -340,7 +340,7 @@ class ID3v23Test(unittest.TestCase, TestHelper): self.path = os.path.join(self.temp_dir, b'test.' + ext) shutil.copy(src, self.path) - return beets.mediafile.MediaFile(self.path, id3v23=id3v23) + return mediafile.MediaFile(self.path, id3v23=id3v23) def _delete_test(self): self.remove_temp_dir() @@ -385,9 +385,9 @@ class ID3v23Test(unittest.TestCase, TestHelper): mf = self._make_test(id3v23=v23) try: mf.images = [ - beets.mediafile.Image(b'data', desc=u""), - beets.mediafile.Image(b'data', desc=u"foo"), - beets.mediafile.Image(b'data', desc=u"\u0185"), + mediafile.Image(b'data', desc=u""), + mediafile.Image(b'data', desc=u"foo"), + mediafile.Image(b'data', desc=u"\u0185"), ] mf.save() apic_frames = mf.mgfile.tags.getall('APIC')