mirror of
https://github.com/saymrwulf/uhd.git
synced 2026-05-15 21:01:26 +00:00
3642ac013e
Pull some general functionality out of a specific test case and create a TestBase class. Future test cases will inherit from TestBase and have access to this general functionality.
43 lines
1.3 KiB
Python
Executable file
43 lines
1.3 KiB
Python
Executable file
#
|
|
# Copyright 2019 Ettus Research, a National Instruments Brand
|
|
#
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
#
|
|
"""
|
|
Base Test Case classes
|
|
"""
|
|
|
|
import platform
|
|
import unittest
|
|
|
|
class TestBase(unittest.TestCase):
|
|
"""
|
|
Base Test Case class which contains commonly required functionality
|
|
"""
|
|
def skipUnlessOnLinux():
|
|
"""
|
|
Test function decorator which skips tests unless the current
|
|
execution environment is a linux OS.
|
|
"""
|
|
if 'linux' in platform.system().lower():
|
|
return lambda func: func
|
|
return unittest.skip("This test is only valid when run on a Linux system.")
|
|
|
|
def skipUnlessOnUsrp():
|
|
"""
|
|
Test function decorator which skips tests unless the current
|
|
execution environment is a USRP.
|
|
|
|
Assumes that 'arm' in the machine name constitutes an ARM
|
|
processor, aka a USRP.
|
|
"""
|
|
if 'arm' in platform.machine().lower():
|
|
return lambda func: func
|
|
return unittest.skip("This test is only valid when run on the USRP.")
|
|
|
|
def set_device_name(self, device_name):
|
|
"""
|
|
Stores a device name attribute for tests whose success condition
|
|
depends on the current device.
|
|
"""
|
|
self.device_name = device_name
|