From 69b856c6163ed426340e63b319d55784b2259cf9 Mon Sep 17 00:00:00 2001 From: Mews Date: Sat, 8 Jun 2024 17:56:01 +0100 Subject: [PATCH] Linted code --- tests/test.py | 75 ++++++++++++++++++++++++++++++++------------------- 1 file changed, 48 insertions(+), 27 deletions(-) diff --git a/tests/test.py b/tests/test.py index 4b82ab7..0dff8db 100644 --- a/tests/test.py +++ b/tests/test.py @@ -29,7 +29,6 @@ def setUp(self): 2 * np.pi * 440 * t ) # Generate a 440 Hz sine wave - def test_stretch_audio_raw(self): factor = 2 @@ -41,7 +40,6 @@ def test_stretch_audio_raw(self): self.assertEqual(round(self.sine_samplerate / factor), stretched_samplerate) - def test_speedup_audio_raw(self): factor = 2 @@ -53,7 +51,6 @@ def test_speedup_audio_raw(self): self.assertEqual(round(self.sine_samplerate / (1 / factor)), spedup_samplerate) - def test_stretch_audio_file_wav(self): factor = 2 file = "tests/sample_files/440hz.wav" @@ -66,7 +63,6 @@ def test_stretch_audio_file_wav(self): self.assertEqual(round(samplerate / factor), stretched_samplerate) - def test_stretch_audio_file_mp3(self): factor = 2 file = "tests/sample_files/440hz.mp3" @@ -79,7 +75,6 @@ def test_stretch_audio_file_mp3(self): self.assertEqual(round(samplerate / factor), stretched_samplerate) - def test_speedup_audio_file_wav(self): factor = 2 file = "tests/sample_files/440hz.wav" @@ -92,7 +87,6 @@ def test_speedup_audio_file_wav(self): self.assertEqual(round(samplerate / (1 / factor)), spedup_samplerate) - def test_speedup_audio_file_mp3(self): factor = 2 file = "tests/sample_files/440hz.wav" @@ -105,7 +99,6 @@ def test_speedup_audio_file_mp3(self): self.assertEqual(round(samplerate / (1 / factor)), spedup_samplerate) - def test_stretch_save_audio_file_wav(self): factor = 2 out_file = "tests/test_files/stretch_save_test.wav" @@ -150,33 +143,48 @@ def test_speedup_save_audio_file(self): self.assertEqual(round(self.sine_samplerate / (1 / factor)), out_samplerate) - def test_stretch_invalid_audio_type(self): factor = 2 # Pass an invalid audio data type - self.assertRaises(TypeError, lambda:stretch_audio(audio=None, factor=factor, samplerate=self.sine_samplerate)) + self.assertRaises( + TypeError, + lambda: stretch_audio( + audio=None, factor=factor, samplerate=self.sine_samplerate + ), + ) - def test_stretch_invalid_factor_zero(self): - self.assertRaises(ValueError, lambda:stretch_audio(audio=self.sine_audio, factor=0, samplerate=self.sine_samplerate)) - + self.assertRaises( + ValueError, + lambda: stretch_audio( + audio=self.sine_audio, factor=0, samplerate=self.sine_samplerate + ), + ) def test_stretch_invalid_factor_negative(self): - self.assertRaises(ValueError, lambda:stretch_audio(audio=self.sine_audio, factor=-1, samplerate=self.sine_samplerate)) + self.assertRaises( + ValueError, + lambda: stretch_audio( + audio=self.sine_audio, factor=-1, samplerate=self.sine_samplerate + ), + ) - def test_stretch_raw_data_no_samplerate(self): factor = 2 - self.assertRaises(TypeError, lambda:stretch_audio(audio=self.sine_audio, factor=factor)) + self.assertRaises( + TypeError, lambda: stretch_audio(audio=self.sine_audio, factor=factor) + ) def test_stretch_raw_data_invalid_samplerate(self): factor = 2 - self.assertRaises(TypeError, lambda:stretch_audio(audio=self.sine_audio, factor=factor, samplerate="")) + self.assertRaises( + TypeError, + lambda: stretch_audio(audio=self.sine_audio, factor=factor, samplerate=""), + ) - def test_stretch_raw_data_float_samplerate(self): factor = 2 @@ -188,27 +196,40 @@ def test_stretch_raw_data_float_samplerate(self): self.assertEqual(round(self.sine_samplerate / factor), stretched_samplerate) - def test_stretch_save_invalid_mp3_samplerate(self): factor = 0.5 out_file = "tests/test_files/stretch_invalid_mp3_save_test.mp3" - self.assertRaises(soundfile.LibsndfileError, lambda:stretch_audio(audio=self.sine_audio, factor=factor, output=out_file, samplerate=self.sine_samplerate)) - + self.assertRaises( + soundfile.LibsndfileError, + lambda: stretch_audio( + audio=self.sine_audio, + factor=factor, + output=out_file, + samplerate=self.sine_samplerate, + ), + ) + # Assert file was deleted self.assertFalse(pathlib.Path(out_file).exists()) - def test_speedup_invalid_factor_zero(self): - self.assertRaises(ValueError, lambda:speedup_audio(audio=self.sine_audio, factor=0, samplerate=self.sine_samplerate)) - + self.assertRaises( + ValueError, + lambda: speedup_audio( + audio=self.sine_audio, factor=0, samplerate=self.sine_samplerate + ), + ) def test_speedup_invalid_factor_negative(self): - self.assertRaises(ValueError, lambda:speedup_audio(audio=self.sine_audio, factor=-1, samplerate=self.sine_samplerate)) - - + self.assertRaises( + ValueError, + lambda: speedup_audio( + audio=self.sine_audio, factor=-1, samplerate=self.sine_samplerate + ), + ) if __name__ == "__main__": - unittest.main() # pragma: no cover \ No newline at end of file + unittest.main() # pragma: no cover \ No newline at end of file