qoherent/ria-toolkit-oss
19
6
Fork 0
Code Issues Pull Requests 1 Workflows Packages Projects Releases 6 Wiki Activity

cli #15

Merged
madrigal merged 28 commits from cli into main 2025-12-22 10:42:57 -05:00
Conversation 0 Commits 28 Files Changed 116 +11 -11
5 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit a87af3d835 - Show all commits

0
src/__init__.py Normal file
View File

12
src/ria_toolkit_oss_cli/ria_toolkit_oss/common.py
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)

4
tests/ria_toolkit_oss_cli/test_capture.py
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

4
tests/ria_toolkit_oss_cli/test_transmit.py
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

2
tests/ria_toolkit_oss_cli/test_transmit_generate.py
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: