diff tests/test_core.py @ 416:14efef6f4bae

implement 'split_and_join_with_silence'

--- a/tests/test_core.py	Wed Oct 16 20:08:37 2024 +0200
+++ b/tests/test_core.py	Wed Oct 16 20:42:58 2024 +0200
@@ -8,7 +8,14 @@
 import numpy as np
 import pytest
 
-from auditok import AudioParameterError, AudioRegion, load, make_silence, split
+from auditok import (
+    AudioParameterError,
+    AudioRegion,
+    load,
+    make_silence,
+    split,
+    split_and_join_with_silence,
+)
 from auditok.core import (
     _duration_to_nb_windows,
     _make_audio_region,
@@ -105,6 +112,63 @@
 
 
 @pytest.mark.parametrize(
+    "duration",
+    [
+        (0,),  # zero_second
+        (1,),  # one_second
+        (1.0001,),  # 1.0001_second
+    ],
+    ids=[
+        "zero_second",
+        "one_second",
+        "1.0001_second",
+    ],
+)
+def test_split_and_join_with_silence(duration):
+    duration = 1.0
+    sampling_rate = 10
+    sample_width = 2
+    channels = 1
+
+    regions = split(
+        input="tests/data/test_split_10HZ_mono.raw",
+        min_dur=0.2,
+        max_dur=5,
+        max_silence=0.2,
+        drop_trailing_silence=False,
+        strict_min_dur=False,
+        analysis_window=0.1,
+        sr=sampling_rate,
+        sw=sample_width,
+        ch=channels,
+        eth=50,
+    )
+
+    size = round(duration * sampling_rate) * sample_width * channels
+    join_data = b"\0" * size
+    expected_data = join_data.join(region.data for region in regions)
+    expected_region = AudioRegion(
+        expected_data, sampling_rate, sample_width, channels
+    )
+
+    region_with_silence = split_and_join_with_silence(
+        input="tests/data/test_split_10HZ_mono.raw",
+        silence_duration=duration,
+        min_dur=0.2,
+        max_dur=5,
+        max_silence=0.2,
+        drop_trailing_silence=False,
+        strict_min_dur=False,
+        analysis_window=0.1,
+        sr=sampling_rate,
+        sw=sample_width,
+        ch=channels,
+        eth=50,
+    )
+    assert region_with_silence == expected_region
+
+
[e-mail](
     "duration, analysis_window, round_fn, expected, kwargs",
     [
         (0, 1, None, 0, None),  # zero_duration