Format code using black

python_apps/airtime_analyzer/tests/metadata_analyzer_tests.py

@@ -6,78 +6,101 @@ import mock
 from nose.tools import *
 from airtime_analyzer.metadata_analyzer import MetadataAnalyzer
 
+
 def setup():
     pass
 
+
 def teardown():
     pass
 
+
 def check_default_metadata(metadata):
-    assert metadata['track_title'] == 'Test Title'
-    assert metadata['artist_name'] == 'Test Artist'
-    assert metadata['album_title'] == 'Test Album'
-    assert metadata['year'] == '1999'
-    assert metadata['genre'] == 'Test Genre'
-    assert metadata['track_number'] == '1'
-    assert metadata["length"] == str(datetime.timedelta(seconds=metadata["length_seconds"]))
+    assert metadata["track_title"] == "Test Title"
+    assert metadata["artist_name"] == "Test Artist"
+    assert metadata["album_title"] == "Test Album"
+    assert metadata["year"] == "1999"
+    assert metadata["genre"] == "Test Genre"
+    assert metadata["track_number"] == "1"
+    assert metadata["length"] == str(
+        datetime.timedelta(seconds=metadata["length_seconds"])
+    )
+
 
 def test_mp3_mono():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-mono.mp3', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-mono.mp3", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 1
-    assert metadata['bit_rate'] == 63998
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['mime'] == 'audio/mp3' # Not unicode because MIMEs aren't.
-    assert metadata['track_total'] == '10' # MP3s can have a track_total
-    #Mutagen doesn't extract comments from mp3s it seems
+    assert metadata["channels"] == 1
+    assert metadata["bit_rate"] == 63998
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["mime"] == "audio/mp3"  # Not unicode because MIMEs aren't.
+    assert metadata["track_total"] == "10"  # MP3s can have a track_total
+    # Mutagen doesn't extract comments from mp3s it seems
+
 
 def test_mp3_jointstereo():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-jointstereo.mp3', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-jointstereo.mp3", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 2
-    assert metadata['bit_rate'] == 127998
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['mime'] == 'audio/mp3'
-    assert metadata['track_total'] == '10' # MP3s can have a track_total
+    assert metadata["channels"] == 2
+    assert metadata["bit_rate"] == 127998
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["mime"] == "audio/mp3"
+    assert metadata["track_total"] == "10"  # MP3s can have a track_total
+
 
 def test_mp3_simplestereo():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-simplestereo.mp3', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-simplestereo.mp3", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 2
-    assert metadata['bit_rate'] == 127998
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['mime'] == 'audio/mp3'
-    assert metadata['track_total'] == '10' # MP3s can have a track_total
+    assert metadata["channels"] == 2
+    assert metadata["bit_rate"] == 127998
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["mime"] == "audio/mp3"
+    assert metadata["track_total"] == "10"  # MP3s can have a track_total
+
 
 def test_mp3_dualmono():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-dualmono.mp3', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-dualmono.mp3", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 2
-    assert metadata['bit_rate'] == 127998
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['mime'] == 'audio/mp3'
-    assert metadata['track_total'] == '10' # MP3s can have a track_total
+    assert metadata["channels"] == 2
+    assert metadata["bit_rate"] == 127998
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["mime"] == "audio/mp3"
+    assert metadata["track_total"] == "10"  # MP3s can have a track_total
 
 
 def test_ogg_mono():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-mono.ogg', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-mono.ogg", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 1
-    assert metadata['bit_rate'] == 80000
-    assert abs(metadata['length_seconds'] - 3.8) < 0.1
-    assert metadata['mime'] == 'audio/vorbis'
-    assert metadata['comment'] == 'Test Comment'
+    assert metadata["channels"] == 1
+    assert metadata["bit_rate"] == 80000
+    assert abs(metadata["length_seconds"] - 3.8) < 0.1
+    assert metadata["mime"] == "audio/vorbis"
+    assert metadata["comment"] == "Test Comment"
+
 
 def test_ogg_stereo():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-stereo.ogg', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-stereo.ogg", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 2
-    assert metadata['bit_rate'] == 112000
-    assert abs(metadata['length_seconds'] - 3.8) < 0.1
-    assert metadata['mime'] == 'audio/vorbis'
-    assert metadata['comment'] == 'Test Comment'
+    assert metadata["channels"] == 2
+    assert metadata["bit_rate"] == 112000
+    assert abs(metadata["length_seconds"] - 3.8) < 0.1
+    assert metadata["mime"] == "audio/vorbis"
+    assert metadata["comment"] == "Test Comment"
 
-''' faac and avconv can't seem to create a proper mono AAC file... ugh
+
+""" faac and avconv can't seem to create a proper mono AAC file... ugh
 def test_aac_mono():
     metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-mono.m4a')
     print("Mono AAC metadata:")
@@ -88,78 +111,93 @@ def test_aac_mono():
     assert abs(metadata['length_seconds'] - 3.8) < 0.1
     assert metadata['mime'] == 'audio/mp4'
     assert metadata['comment'] == 'Test Comment'
-'''
+"""
+
 
 def test_aac_stereo():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-stereo.m4a', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-stereo.m4a", dict()
+    )
     check_default_metadata(metadata)
-    assert metadata['channels'] == 2
-    assert metadata['bit_rate'] == 102619
-    assert abs(metadata['length_seconds'] - 3.8) < 0.1
-    assert metadata['mime'] == 'audio/mp4'
-    assert metadata['comment'] == 'Test Comment'
+    assert metadata["channels"] == 2
+    assert metadata["bit_rate"] == 102619
+    assert abs(metadata["length_seconds"] - 3.8) < 0.1
+    assert metadata["mime"] == "audio/mp4"
+    assert metadata["comment"] == "Test Comment"
+
 
 def test_mp3_utf8():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-stereo-utf8.mp3', dict())
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-stereo-utf8.mp3", dict()
+    )
     # Using a bunch of different UTF-8 codepages here. Test data is from:
     #   http://winrus.com/utf8-jap.htm
-    assert metadata['track_title'] == 'ｱｲｳｴｵｶｷｸｹｺｻｼｽｾｿﾀﾁﾂﾃ'
-    assert metadata['artist_name'] == 'てすと'
-    assert metadata['album_title'] == 'Ä ä Ü ü ß'
-    assert metadata['year'] == '1999'
-    assert metadata['genre'] == 'Я Б Г Д Ж Й'
-    assert metadata['track_number'] == '1'
-    assert metadata['channels'] == 2
-    assert metadata['bit_rate'] < 130000
-    assert metadata['bit_rate'] > 127000
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['mime'] == 'audio/mp3'
-    assert metadata['track_total'] == '10' # MP3s can have a track_total
+    assert metadata["track_title"] == "ｱｲｳｴｵｶｷｸｹｺｻｼｽｾｿﾀﾁﾂﾃ"
+    assert metadata["artist_name"] == "てすと"
+    assert metadata["album_title"] == "Ä ä Ü ü ß"
+    assert metadata["year"] == "1999"
+    assert metadata["genre"] == "Я Б Г Д Ж Й"
+    assert metadata["track_number"] == "1"
+    assert metadata["channels"] == 2
+    assert metadata["bit_rate"] < 130000
+    assert metadata["bit_rate"] > 127000
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["mime"] == "audio/mp3"
+    assert metadata["track_total"] == "10"  # MP3s can have a track_total
+
 
 def test_invalid_wma():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-stereo-invalid.wma', dict())
-    assert metadata['mime'] == 'audio/x-ms-wma'
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-stereo-invalid.wma", dict()
+    )
+    assert metadata["mime"] == "audio/x-ms-wma"
+
 
 def test_wav_stereo():
-    metadata = MetadataAnalyzer.analyze('tests/test_data/44100Hz-16bit-stereo.wav', dict())
-    assert metadata['mime'] == 'audio/x-wav'
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['channels'] == 2
-    assert metadata['sample_rate'] == 44100
+    metadata = MetadataAnalyzer.analyze(
+        "tests/test_data/44100Hz-16bit-stereo.wav", dict()
+    )
+    assert metadata["mime"] == "audio/x-wav"
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["channels"] == 2
+    assert metadata["sample_rate"] == 44100
 
 
 # Make sure the parameter checking works
 @raises(FileNotFoundError)
 def test_move_wrong_string_param1():
-    not_unicode = 'asdfasdf'
+    not_unicode = "asdfasdf"
     MetadataAnalyzer.analyze(not_unicode, dict())
 
+
 @raises(TypeError)
 def test_move_wrong_metadata_dict():
     not_a_dict = list()
-    MetadataAnalyzer.analyze('asdfasdf', not_a_dict)
+    MetadataAnalyzer.analyze("asdfasdf", not_a_dict)
+
 
 # Test an mp3 file where the number of channels is invalid or missing:
 def test_mp3_bad_channels():
-    filename = 'tests/test_data/44100Hz-16bit-mono.mp3'
-    '''
+    filename = "tests/test_data/44100Hz-16bit-mono.mp3"
+    """
         It'd be a pain in the ass to construct a real MP3 with an invalid number
         of channels by hand because that value is stored in every MP3 frame in the file
-    '''
+    """
     audio_file = mutagen.File(filename, easy=True)
     audio_file.info.mode = 1777
-    with mock.patch('airtime_analyzer.metadata_analyzer.mutagen') as mock_mutagen:
+    with mock.patch("airtime_analyzer.metadata_analyzer.mutagen") as mock_mutagen:
         mock_mutagen.File.return_value = audio_file
-        #mock_mutagen.side_effect = lambda *args, **kw: audio_file #File(*args, **kw)
+        # mock_mutagen.side_effect = lambda *args, **kw: audio_file #File(*args, **kw)
 
     metadata = MetadataAnalyzer.analyze(filename, dict())
     check_default_metadata(metadata)
-    assert metadata['channels'] == 1
-    assert metadata['bit_rate'] == 63998
-    assert abs(metadata['length_seconds'] - 3.9) < 0.1
-    assert metadata['mime'] == 'audio/mp3' # Not unicode because MIMEs aren't.
-    assert metadata['track_total'] == '10' # MP3s can have a track_total
-    #Mutagen doesn't extract comments from mp3s it seems
+    assert metadata["channels"] == 1
+    assert metadata["bit_rate"] == 63998
+    assert abs(metadata["length_seconds"] - 3.9) < 0.1
+    assert metadata["mime"] == "audio/mp3"  # Not unicode because MIMEs aren't.
+    assert metadata["track_total"] == "10"  # MP3s can have a track_total
+    # Mutagen doesn't extract comments from mp3s it seems
+
 
 def test_unparsable_file():
-    MetadataAnalyzer.analyze('tests/test_data/unparsable.txt', dict())
+    MetadataAnalyzer.analyze("tests/test_data/unparsable.txt", dict())