Fix config for 3rd device in valid range test

Author: BitForge95

tests/create_experiment.py

@@ -1,6 +1,8 @@
 import copy
 import shutil
 from contextlib import contextmanager
+from pathlib import Path
+
 import numpy as np
 import yaml
 
@@ -30,13 +32,15 @@ def get_default_config():
 
 
 @contextmanager
-def create_experiment(
+def setup_test_experiment(
     tmp_path,
     n_devices=2,
     devices_kwargs=None,
+    default_sampling_rate=1.0,
 ):
     devices_kwargs = devices_kwargs or [{}] * n_devices
-    default_params = {"sampling_rate": 1.0}
+    default_params = {"sampling_rate": default_sampling_rate}
+
     devices_kwargs = [default_params | kwargs for kwargs in devices_kwargs]
 
     try:
@@ -97,12 +101,12 @@ def make_modality_config(*device_names, sampling_rates=None, offsets=None):
     elif isinstance(offsets, (int, float)):
         offsets = [offsets] * len(device_names)
 
-    assert len(device_names) == len(
-        sampling_rates
-    ), f"sampling_rates length {len(sampling_rates)} does not match device_names length {len(device_names)}"
-    assert len(device_names) == len(
-        offsets
-    ), f"offsets length {len(offsets)} does not match device_names length {len(device_names)}"
+    assert len(device_names) == len(sampling_rates), (
+        f"sampling_rates length {len(sampling_rates)} does not match device_names length {len(device_names)}"
+    )
+    assert len(device_names) == len(offsets), (
+        f"offsets length {len(offsets)} does not match device_names length {len(device_names)}"
+    )
 
     return {
         name: {"interpolation": {"sampling_rate": sr, "offset": off}}