Rewrote tests using pytest

tests/test_speedup.py

@@ -0,0 +1,94 @@
+"""
+Audio tone files from https://www.mediacollege.com/audio/tone/download/
+"""
+
+import os
+import sys
+
+sys.path.insert(
+    0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "src"))
+)
+
+import pathlib
+
+import numpy as np
+import pytest
+import soundfile
+from simplestretch import speedup_audio
+
+
+class TestStretch:
+    # Create a simple audio signal for testing
+    sine_samplerate = 44100
+    duration = 2  # seconds
+    t = np.linspace(0, duration, int(sine_samplerate * duration), endpoint=False)
+    sine_audio = 0.5 * np.sin(2 * np.pi * 440 * t)  # Generate a 440 Hz sine wave
+
+    def test_speedup_audio_raw(self):
+        factor = 2
+
+        spedup_audio, spedup_samplerate = speedup_audio(
+            audio=self.sine_audio, factor=factor, samplerate=self.sine_samplerate
+        )
+
+        assert self.sine_audio.all() == spedup_audio.all()
+
+        assert round(self.sine_samplerate / (1 / factor)) == spedup_samplerate
+
+    def test_speedup_audio_file_wav(self):
+        factor = 2
+        file = "tests/sample_files/440hz.wav"
+
+        audio, samplerate = soundfile.read(file)
+
+        spedup_audio, spedup_samplerate = speedup_audio(audio=file, factor=factor)
+
+        assert audio.all() == spedup_audio.all()
+
+        assert round(samplerate / (1 / factor)) == spedup_samplerate
+
+    def test_speedup_audio_file_mp3(self):
+        factor = 2
+        file = "tests/sample_files/440hz.wav"
+
+        audio, samplerate = soundfile.read(file)
+
+        spedup_audio, spedup_samplerate = speedup_audio(audio=file, factor=factor)
+
+        assert audio.all() == spedup_audio.all()
+
+        assert round(samplerate / (1 / factor)) == spedup_samplerate
+
+    def test_speedup_save_audio_file(self):
+        factor = 2
+        out_file = "tests/test_files/speedup_save_test.wav"
+
+        speedup_audio(
+            audio=self.sine_audio,
+            factor=factor,
+            output=out_file,
+            samplerate=self.sine_samplerate,
+        )
+
+        # Assert the file exists
+        path = pathlib.Path(out_file)
+        assert path.is_file()
+
+        # Assert the file contents are correct
+        out_audio, out_samplerate = soundfile.read(out_file)
+
+        assert self.sine_audio.all() == out_audio.all()
+
+        assert round(self.sine_samplerate / (1 / factor)) == out_samplerate
+
+    def test_speedup_invalid_factor_zero(self):
+        with pytest.raises(ValueError):
+            speedup_audio(
+                audio=self.sine_audio, factor=0, samplerate=self.sine_samplerate
+            )
+
+    def test_speedup_invalid_factor_negative(self):
+        with pytest.raises(ValueError):
+            speedup_audio(
+                audio=self.sine_audio, factor=-1, samplerate=self.sine_samplerate
+            )