[FIX] Update condition to trim parameter list based on AD9081 configu…

…ration Signed-off-by: Julia Pineda <[email protected]>
analogdevicesinc · Aug 10, 2023 · 28352f4 · 28352f4
1 parent 13ddc17
commit 28352f4
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/test/test_ad9081.py b/test/test_ad9081.py
@@ -13,10 +13,9 @@ def scale_field(param_set, iio_uri):
 
     dev = adi.ad9081(uri=iio_uri)
     for field in param_set:
-        if param_set[field] is not list:
-            continue
-        existing_val = getattr(dev, field)
-        param_set[field] = param_set[field][0] * len(existing_val)
+        if isinstance(param_set[field], list):
+            existing_val = getattr(dev, field)
+            param_set[field] = [param_set[field][0]] * len(existing_val)
     return param_set