1 files changed, 57 insertions, 1 deletions
diff --git a/host/tests/devtest/gpio_test.py b/host/tests/devtest/gpio_test.py
index 179ace5b3..517f0bf5b 100755
--- a/host/tests/devtest/gpio_test.py
+++ b/host/tests/devtest/gpio_test.py
@@ -8,10 +8,13 @@
 
 import re
 from uhd_test_base import uhd_example_test_case, uhd_test_case, UHDPythonTestCase
+import random
 
 class gpio_test(uhd_example_test_case):
     """ Run gpio. """
-    tests = {'default': {},}
+    tests = {'default': {
+        'addl_args': [],
+    },}
 
     def setup_example(self):
         """
@@ -26,6 +29,7 @@ class gpio_test(uhd_example_test_case):
         args = [
             self.create_addr_args_str(),
         ]
+        args += test_args['addl_args']
         (app, run_results) = self.run_example('gpio', args)
         # Evaluate pass/fail:
         run_results['passed'] = all([
@@ -58,3 +62,55 @@ class gpio_atr_readback_test(UHDPythonTestCase):
             assert usrp.get_gpio_attr(gpio, "ATR_RX") == 0xCAB
             assert usrp.get_gpio_attr(gpio, "ATR_TX") == 0xABA
             assert usrp.get_gpio_attr(gpio, "ATR_XX") == 0xBAB
+
+
+class gpio_x410_set_get_source_test(UHDPythonTestCase):
+    test_params = {
+        "possible_sources": [],
+        "num_pins": 0,
+    }
+
+    def test_all(self):
+        import uhd
+        usrp = uhd.usrp.MultiUSRP(self.args_str)
+
+        POSSIBLE_SOURCES = self.test_params["possible_sources"]
+
+        # Assemble two lists which have at least one of each of the 7 sources,
+        # with the remaining 12-7=5 entries containing random sources.
+        sources_0 = [x for x in POSSIBLE_SOURCES]
+        sources_1 = [x for x in POSSIBLE_SOURCES]
+        for _ in range(self.test_params["num_pins"] - len(POSSIBLE_SOURCES)):
+            sources_0.append(random.choice(POSSIBLE_SOURCES))
+            sources_1.append(random.choice(POSSIBLE_SOURCES))
+        random.shuffle(sources_0)
+        random.shuffle(sources_1)
+
+        usrp.set_gpio_src("GPIO0", sources_0)
+        usrp.set_gpio_src("GPIO1", sources_1)
+
+        assert sources_0 == list(usrp.get_gpio_src("GPIO0"))
+        assert sources_1 == list(usrp.get_gpio_src("GPIO1"))
+
+
+class x410_gpio_power_test(UHDPythonTestCase):
+    """ Run gpio_power_test """
+    test_name = "x410_gpio_power_test"
+
+    def run_test(self, test_name, test_args):
+        """
+        Run test and report results.
+        """
+        import uhd
+        usrp = uhd.usrp.MultiUSRP(self.args_str)
+
+        gpio_power = usrp.get_mb_controller().get_gpio_power()
+        assert usrp.get_mb_controller().get_gpio_power().get_supported_voltages("GPIO0") == ['OFF', '1V8', '2V5', '3V3']
+
+        for port in ["GPIO0", "GPIO1"]:
+            assert gpio_power.get_port_voltage(port) == "3V3"
+            for voltage in gpio_power.get_supported_voltages(port):
+                gpio_power.set_port_voltage(port, voltage)
+                assert gpio_power.get_port_voltage(port) == voltage
+
+        return {"passed": True}