test: add benches for TheAlgorithms/audio_filters

Author: art049

.github/workflows/codspeed.yml

@@ -10,7 +10,9 @@ jobs:
   benchmarks:
     runs-on: ubuntu-latest
     steps:
-      - uses: actions/checkout@v3
+      - uses: actions/checkout@v4
+        with:
+          submodules: "recursive"
       - name: Set up Python 3.12
         uses: actions/setup-python@v2
         with:

.gitmodules

@@ -0,0 +1,3 @@
+[submodule "tests/benchmarks/TheAlgorithms"]
+	path = tests/benchmarks/TheAlgorithms
+	url = [email protected]:TheAlgorithms/Python.git

pyproject.toml

@@ -87,8 +87,9 @@ force_grid_wrap = 0
 float_to_top = true
 
 [tool.pytest.ini_options]
-addopts = "--ignore=tests/benchmarks --ignore=tests/examples"
+addopts = "--ignore=tests/benchmarks --ignore=tests/examples --ignore=tests/benchmarks/TheAlgorithms"
 filterwarnings = ["ignore::DeprecationWarning:pytest_benchmark.utils.*:"]
+pythonpath = ["tests/benchmarks/TheAlgorithms"]
 
 [tool.coverage.run]
 branch = true

tests/benchmarks/TheAlgorithms

@@ -0,0 +1,37 @@
+from audio_filters.butterworth_filter import (
+    make_allpass,
+    make_bandpass,
+    make_highpass,
+    make_highshelf,
+    make_lowpass,
+    make_lowshelf,
+    make_peak,
+)
+
+
+def test_make_lowpass(benchmark):
+    benchmark(make_lowpass, 1000, 48000)
+
+
+def test_make_highpass(benchmark):
+    benchmark(make_highpass, 1000, 48000)
+
+
+def test_make_bandpass(benchmark):
+    benchmark(make_bandpass, 1000, 48000)
+
+
+def test_make_allpass(benchmark):
+    benchmark(make_allpass, 1000, 48000)
+
+
+def test_make_peak(benchmark):
+    benchmark(make_peak, 1000, 48000, 6)
+
+
+def test_make_lowshelf(benchmark):
+    benchmark(make_lowshelf, 1000, 48000, 6)
+
+
+def test_make_highshelf(benchmark):
+    benchmark(make_highshelf, 1000, 48000, 6)

tests/benchmarks/TheAlgorithms_bench/__init__.py

@@ -0,0 +1,13 @@
+from audio_filters.iir_filter import IIRFilter
+
+
+def test_set_coefficients(benchmark):
+    filt = IIRFilter(2)
+    a_coeffs = [1.0, -1.8, 0.81]
+    b_coeffs = [0.9, -1.8, 0.81]
+    benchmark(filt.set_coefficients, a_coeffs, b_coeffs)
+
+
+def test_process(benchmark):
+    filt = IIRFilter(2)
+    benchmark(filt.process, 0)