mirror of
https://github.com/saymrwulf/uhd.git
synced 2026-05-16 21:10:10 +00:00
116 lines
3.8 KiB
Python
Executable file
116 lines
3.8 KiB
Python
Executable file
#
|
|
# Copyright 2015 Ettus Research LLC
|
|
# Copyright 2018 Ettus Research, a National Instruments Company
|
|
#
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
""" Test for gpio. """
|
|
|
|
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': {
|
|
'addl_args': [],
|
|
},}
|
|
|
|
def setup_example(self):
|
|
"""
|
|
Set args.
|
|
"""
|
|
self.test_params = gpio_test.tests
|
|
|
|
def run_test(self, test_name, test_args):
|
|
""" Run the app and scrape for the success message. """
|
|
self.log.info('Running test {n}'.format(n=test_name,))
|
|
# Run example:
|
|
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([
|
|
app.returncode == 0,
|
|
re.search('All tests passed!', app.stdout) is not None,
|
|
])
|
|
if not run_results['passed']:
|
|
print(app.stdout)
|
|
print(app.stderr)
|
|
self.report_example_results(test_name, run_results)
|
|
return run_results
|
|
|
|
|
|
class gpio_atr_readback_test(UHDPythonTestCase):
|
|
test_params = []
|
|
|
|
def test_all(self):
|
|
import uhd
|
|
usrp = uhd.usrp.MultiUSRP(self.args_str)
|
|
|
|
for gpio, source in self.test_params:
|
|
usrp.set_gpio_src("GPIO0", [source] * 12)
|
|
|
|
usrp.set_gpio_attr(gpio, "ATR_0X", 0xABC, 0xFFF)
|
|
usrp.set_gpio_attr(gpio, "ATR_RX", 0xCAB, 0xFFF)
|
|
usrp.set_gpio_attr(gpio, "ATR_TX", 0xABA, 0xFFF)
|
|
usrp.set_gpio_attr(gpio, "ATR_XX", 0xBAB, 0xFFF)
|
|
|
|
assert usrp.get_gpio_attr(gpio, "ATR_0X") == 0xABC
|
|
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}
|