Stereo merge + test

penn/load.py

@@ -10,7 +10,7 @@ def audio(file):
     """Load audio from disk"""
     audio, sample_rate = torchaudio.load(file)
 
-    # if audio is stereo convert to mono
+    # If audio is stereo, convert to mono
     if audio.size(0) == 2:
         warnings.warn(f'Converting stereo audio to mono: {file}')
         audio = audio.mean(dim=0, keepdim=True)

test/conftest.py

@@ -15,7 +15,8 @@ def audio():
     """Retrieve the test audio"""
     return penn.load.audio(Path(__file__).parent / 'assets' / 'gershwin.wav')
 
+
 @pytest.fixture(scope='session')
 def audio_stereo():
     """Retrieve the test audio"""
-    return penn.load.audio(Path(__file__).parent / 'assets' / '500Hz_stereo.wav')
+    return penn.load.audio(Path(__file__).parent / 'assets' / '500Hz_stereo.wav')

test/test_core.py

@@ -15,11 +15,12 @@ def test_infer(audio):
     shape = (1, audio.shape[1] // penn.HOPSIZE)
     assert pitch.shape == periodicity.shape == shape
 
+
 def test_infer_stereo(audio_stereo):
     """Test that inference on stereo audio produces the correct shape"""
     pitch, periodicity = penn.from_audio(
         audio_stereo,
         penn.SAMPLE_RATE,
         center='half-hop')
     shape = (1, audio_stereo.shape[1] // penn.HOPSIZE)
-    assert pitch.shape == periodicity.shape == shape
+    assert pitch.shape == periodicity.shape == shape