Skip to content

Commit

Permalink
Linted code
Browse files Browse the repository at this point in the history
  • Loading branch information
Mews committed Jun 8, 2024
1 parent ae43528 commit 69b856c
Showing 1 changed file with 48 additions and 27 deletions.
75 changes: 48 additions & 27 deletions tests/test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand All @@ -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

Expand All @@ -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"
Expand All @@ -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"
Expand All @@ -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"
Expand All @@ -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"
Expand All @@ -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"
Expand Down Expand Up @@ -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

Expand All @@ -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
unittest.main() # pragma: no cover

0 comments on commit 69b856c

Please sign in to comment.