From 91127659dd8677be3ea41e57115fc780e3bcbed7 Mon Sep 17 00:00:00 2001
From: Trisha De Vera <Trisha.Devera@analog.com>
Date: Thu, 18 Jul 2024 13:17:22 +0800
Subject: [PATCH] Add cn0540 test

Add test for cn0540 and a new attribute test

Signed-off-by: Trisha De Vera <Trisha.Devera@analog.com>
---
 test/attr_tests.py  |  20 +++++++++
 test/conftest.py    |   5 +++
 test/test_cn0540.py | 100 ++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 125 insertions(+)
 create mode 100644 test/test_cn0540.py

diff --git a/test/attr_tests.py b/test/attr_tests.py
index 48643d784..fc10ae210 100644
--- a/test/attr_tests.py
+++ b/test/attr_tests.py
@@ -717,3 +717,23 @@ def attribute_check_range_singleval_with_depends(
             )
         else:
             assert dev_interface(uri, classname, val, attr, tol)
+
+
+def attribute_single_value_boolean_readonly(uri, classname, attr):
+    """attribute_single_value_boolean: Read boolean class property
+
+    parameters:
+        uri: type=string
+            URI of IIO context of target board/system
+        classname: type=string
+            Name of pyadi interface class which contain attribute
+        attr: type=string
+            Attribute name to be written. Must be property of classname
+    """
+    bi = eval(classname + "(uri='" + uri + "')")
+
+    if not hasattr(bi, attr):
+        raise AttributeError(f"no attribute named: {attr}")
+
+    rval = getattr(bi, attr)
+    assert type(rval) != None
diff --git a/test/conftest.py b/test/conftest.py
index 7220c1ef6..33da49aeb 100644
--- a/test/conftest.py
+++ b/test/conftest.py
@@ -242,3 +242,8 @@ def test_attribute_single_value_channel_readonly(request):
 @pytest.fixture()
 def test_attribute_check_range_singleval_with_depends(request):
     yield attribute_check_range_singleval_with_depends
+
+
+@pytest.fixture()
+def test_attribute_single_value_boolean_readonly(request):
+    yield attribute_single_value_boolean_readonly
diff --git a/test/test_cn0540.py b/test/test_cn0540.py
new file mode 100644
index 000000000..da0f1413b
--- /dev/null
+++ b/test/test_cn0540.py
@@ -0,0 +1,100 @@
+import random
+from test.attr_tests import floor_step_size
+
+import adi
+import pytest
+from adi.cn0540 import cn0540
+
+hardware = "cn0540"
+classname = "adi.cn0540.cn0540"
+
+
+#########################################
+@pytest.mark.iio_hardware(hardware, True)
+@pytest.mark.parametrize("classname", [(classname)])
+@pytest.mark.parametrize("channel", [0])
+def test_cn0540_rx_data(test_dma_rx, iio_uri, classname, channel):
+    test_dma_rx(iio_uri, classname, channel)
+
+
+#########################################
+@pytest.mark.iio_hardware(hardware)
+@pytest.mark.parametrize("classname", [(classname)])
+@pytest.mark.parametrize(
+    "attr, val",
+    [("sample_rate", [256000, 128000, 64000, 32000, 16000, 8000, 4000, 2000, 1000,],),],
+)
+def test_cn0540_attr_multiple(
+    test_attribute_multiple_values, iio_uri, classname, attr, val
+):
+    test_attribute_multiple_values(iio_uri, classname, attr, val, 0)
+
+
+#########################################
+@pytest.mark.iio_hardware(hardware)
+@pytest.mark.parametrize("classname", [(classname)])
+@pytest.mark.parametrize(
+    "attr, value",
+    [
+        ("monitor_powerup", 0),
+        ("monitor_powerup", 1),
+        ("fda_disable_status", 0),
+        ("fda_disable_status", 1),
+        ("red_led_enable", 0),
+        ("red_led_enable", 1),
+        ("sw_cc", 0),
+        ("sw_cc", 1),
+        ("fda_mode", "low-power"),
+        ("fda_mode", "full-power"),
+    ],
+)
+def test_cn0540_attr_boolean(
+    test_attribute_single_value_boolean, iio_uri, classname, attr, value
+):
+    test_attribute_single_value_boolean(iio_uri, classname, attr, value)
+
+
+#########################################
+@pytest.mark.iio_hardware(hardware)
+@pytest.mark.parametrize("classname", [(classname)])
+@pytest.mark.parametrize(
+    "attr", [("sw_ff_status"),],
+)
+def test_cn0540_attr_boolean_readonly(
+    test_attribute_single_value_boolean_readonly, iio_uri, classname, attr
+):
+    test_attribute_single_value_boolean_readonly(iio_uri, classname, attr)
+
+
+#########################################
+@pytest.mark.iio_hardware(hardware, True)
+@pytest.mark.parametrize("classname", [(classname)])
+@pytest.mark.parametrize(
+    "attr, start, stop, step, repeats ", [("shift_voltage", 0, 65535, 1, 2),],
+)
+def test_cn0540_shift_voltage(iio_uri, classname, attr, start, stop, step, repeats):
+    device = cn0540(uri=iio_uri)
+    bi = eval(classname + "(uri='" + iio_uri + "')")
+
+    if not hasattr(bi, attr):
+        raise AttributeError(f"no attribute named: {attr}")
+
+    # Pick random number in operational range
+    numints = int((stop - start) / step)
+    for _ in range(repeats):
+        ind = random.randint(0, numints)
+        val = start + step * ind
+        if isinstance(val, float):
+            val = floor_step_size(val, str(step))
+
+        setattr(bi, "raw", val)
+        assert 0 <= device.shift_voltage <= 4997.04375
+
+
+#########################################
+@pytest.mark.iio_hardware(hardware, True)
+def test_cn0540_readonly(iio_uri):
+    device = cn0540(uri=iio_uri)
+    attr = [device.input_voltage, device.sensor_voltage]
+    for attr_name in attr:
+        assert type(attr_name) != None