fix bug in dds tests librs.py with wait_for_devices()

maloel · Sep 5, 2024 · bb538e0 · bb538e0
1 parent afcdac5
commit bb538e0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/unit-tests/dds/librs.py b/unit-tests/dds/librs.py
@@ -39,7 +39,11 @@ def wait_for_devices( context, mask=product_line.any, n=1, timeout=3, throw=None
     timer = _Timer( timeout )
     devices = context.query_devices( mask )
     if len(devices) >= n:
-        if not exact or len(devices) == n:
+        if not exact:
+            return devices
+        if len(devices) == n:
+            if n == 1:
+                return devices[0]
             return devices
     #
     context.set_devices_changed_callback( _device_change_callback )