removed src formatting

This commit is contained in:
G gillian 2025-12-15 15:19:54 -05:00
parent b33c0c2e01
commit a87af3d835
5 changed files with 11 additions and 11 deletions

0
src/__init__.py Normal file
View File

View File

@ -357,7 +357,7 @@ def get_sdr_device(device_type: str, ident: Optional[str] = None, tx=False):
try: try:
if device_type == "pluto": if device_type == "pluto":
from src.ria_toolkit_oss.sdr.pluto import Pluto from ria_toolkit_oss.sdr.pluto import Pluto
if ip_addr: if ip_addr:
return Pluto(identifier=ip_addr) return Pluto(identifier=ip_addr)
@ -365,17 +365,17 @@ def get_sdr_device(device_type: str, ident: Optional[str] = None, tx=False):
return Pluto() return Pluto()
elif device_type == "hackrf": elif device_type == "hackrf":
from src.ria_toolkit_oss.sdr.hackrf import HackRF from ria_toolkit_oss.sdr.hackrf import HackRF
return HackRF() return HackRF()
elif device_type == "bladerf": elif device_type == "bladerf":
from src.ria_toolkit_oss.sdr.blade import Blade from ria_toolkit_oss.sdr.blade import Blade
return Blade() return Blade()
elif device_type == "usrp": elif device_type == "usrp":
from src.ria_toolkit_oss.sdr.usrp import USRP from ria_toolkit_oss.sdr.usrp import USRP
if ip_addr: if ip_addr:
return USRP(identifier=f"addr={ip_addr}") return USRP(identifier=f"addr={ip_addr}")
@ -385,12 +385,12 @@ def get_sdr_device(device_type: str, ident: Optional[str] = None, tx=False):
return USRP() return USRP()
elif device_type == "rtlsdr": elif device_type == "rtlsdr":
from src.ria_toolkit_oss.sdr.rtlsdr import RTLSDR from ria_toolkit_oss.sdr.rtlsdr import RTLSDR
return RTLSDR() return RTLSDR()
elif device_type == "thinkrf": elif device_type == "thinkrf":
from src.ria_toolkit_oss.sdr.thinkrf import ThinkRF from ria_toolkit_oss.sdr.thinkrf import ThinkRF
if ip_addr: if ip_addr:
return ThinkRF(identifier=ip_addr) return ThinkRF(identifier=ip_addr)

View File

@ -27,7 +27,7 @@ class TestGetSdrDevice:
mock_sdr_instance = MagicMock() mock_sdr_instance = MagicMock()
mock_sdr_class.return_value = mock_sdr_instance mock_sdr_class.return_value = mock_sdr_instance
with patch.dict("sys.modules", {"src.ria_toolkit_oss.sdr.pluto": MagicMock(Pluto=mock_sdr_class)}): with patch.dict("sys.modules", {"ria_toolkit_oss.sdr.pluto": MagicMock(Pluto=mock_sdr_class)}):
device = get_sdr_device("pluto") device = get_sdr_device("pluto")
assert device is mock_sdr_instance assert device is mock_sdr_instance
@ -37,7 +37,7 @@ class TestGetSdrDevice:
mock_sdr_instance = MagicMock() mock_sdr_instance = MagicMock()
mock_sdr_class.return_value = mock_sdr_instance mock_sdr_class.return_value = mock_sdr_instance
with patch.dict("sys.modules", {"src.ria_toolkit_oss.sdr.hackrf": MagicMock(HackRF=mock_sdr_class)}): with patch.dict("sys.modules", {"ria_toolkit_oss.sdr.hackrf": MagicMock(HackRF=mock_sdr_class)}):
device = get_sdr_device("hackrf") device = get_sdr_device("hackrf")
assert device is mock_sdr_instance assert device is mock_sdr_instance

View File

@ -27,7 +27,7 @@ class TestGetTxDevice:
mock_sdr_instance = MagicMock() mock_sdr_instance = MagicMock()
mock_sdr_class.return_value = mock_sdr_instance mock_sdr_class.return_value = mock_sdr_instance
with patch.dict("sys.modules", {"src.ria_toolkit_oss.sdr.pluto": MagicMock(Pluto=mock_sdr_class)}): with patch.dict("sys.modules", {"ria_toolkit_oss.sdr.pluto": MagicMock(Pluto=mock_sdr_class)}):
device = get_sdr_device("pluto") device = get_sdr_device("pluto")
assert device is mock_sdr_instance assert device is mock_sdr_instance
@ -37,7 +37,7 @@ class TestGetTxDevice:
mock_sdr_instance = MagicMock() mock_sdr_instance = MagicMock()
mock_sdr_class.return_value = mock_sdr_instance mock_sdr_class.return_value = mock_sdr_instance
with patch.dict("sys.modules", {"src.ria_toolkit_oss.sdr.hackrf": MagicMock(HackRF=mock_sdr_class)}): with patch.dict("sys.modules", {"ria_toolkit_oss.sdr.hackrf": MagicMock(HackRF=mock_sdr_class)}):
device = get_sdr_device("hackrf") device = get_sdr_device("hackrf")
assert device is mock_sdr_instance assert device is mock_sdr_instance

View File

@ -2,7 +2,7 @@
from click.testing import CliRunner from click.testing import CliRunner
from src.ria_toolkit_oss.ria_toolkit_oss_cli.cli import cli from ria_toolkit_oss_cli.cli import cli
class TestTransmitGenerate: class TestTransmitGenerate: