cli #15

Merged
madrigal merged 28 commits from cli into main 2025-12-22 10:42:57 -05:00
5 changed files with 19 additions and 33 deletions
Showing only changes of commit 262d6ce9ee - Show all commits

View File

@ -338,7 +338,7 @@ def capture(
num_samples = int(duration * sample_rate) num_samples = int(duration * sample_rate)
echo_verbose(f"Duration {duration}s = {num_samples} samples at {format_sample_rate(sample_rate)}", verbose) echo_verbose(f"Duration {duration}s = {num_samples} samples at {format_sample_rate(sample_rate)}", verbose)
elif duration is None and num_samples is None: elif duration is None and num_samples is None:
raise click.ClickException(f"Must provide either --num-samples or --duration") raise click.ClickException("Must provide either --num-samples or --duration")
# Show capture parameters # Show capture parameters
echo_progress(f"Capturing from {device.upper()}...", quiet) echo_progress(f"Capturing from {device.upper()}...", quiet)

View File

@ -9,15 +9,6 @@ import yaml
import ria_toolkit_oss.signal.basic_signal_generator as basic_gen import ria_toolkit_oss.signal.basic_signal_generator as basic_gen
from ria_toolkit_oss.datatypes import Recording from ria_toolkit_oss.datatypes import Recording
from ria_toolkit_oss_cli.ria_toolkit_oss.common import (
echo_progress,
echo_verbose,
format_frequency,
format_sample_rate,
parse_metadata_args,
save_recording,
)
from ria_toolkit_oss_cli.ria_toolkit_oss.config import load_user_config
from ria_toolkit_oss.signal.block_generator.basic import FrequencyShift from ria_toolkit_oss.signal.block_generator.basic import FrequencyShift
from ria_toolkit_oss.signal.block_generator.continuous_modulation.fsk_modulator import ( from ria_toolkit_oss.signal.block_generator.continuous_modulation.fsk_modulator import (
FSKModulator, FSKModulator,
@ -51,6 +42,15 @@ from ria_toolkit_oss.signal.block_generator.symbol_modulation import (
from ria_toolkit_oss.transforms.iq_impairments import ( from ria_toolkit_oss.transforms.iq_impairments import (
iq_imbalance, iq_imbalance,
) )
from ria_toolkit_oss_cli.ria_toolkit_oss.common import (
echo_progress,
echo_verbose,
format_frequency,
format_sample_rate,
parse_metadata_args,
save_recording,
)
from ria_toolkit_oss_cli.ria_toolkit_oss.config import load_user_config
# Extend Mapper to support new types # Extend Mapper to support new types

View File

@ -10,13 +10,13 @@ import click
from ria_toolkit_oss.datatypes.recording import Recording from ria_toolkit_oss.datatypes.recording import Recording
from ria_toolkit_oss.io.recording import load_recording from ria_toolkit_oss.io.recording import load_recording
from ria_toolkit_oss.transforms import iq_augmentations, iq_impairments
from ria_toolkit_oss_cli.ria_toolkit_oss.common import ( from ria_toolkit_oss_cli.ria_toolkit_oss.common import (
echo_progress, echo_progress,
echo_verbose, echo_verbose,
format_sample_count, format_sample_count,
save_recording, save_recording,
) )
from ria_toolkit_oss.transforms import iq_augmentations, iq_impairments
def get_available_transforms(module): def get_available_transforms(module):

View File

@ -6,10 +6,10 @@ from pathlib import Path
import numpy as np import numpy as np
import pytest import pytest
from click.testing import CliRunner from click.testing import CliRunner
from ria_toolkit_oss_cli.cli import cli
from ria_toolkit_oss.datatypes import Annotation, Recording from ria_toolkit_oss.datatypes import Annotation, Recording
from ria_toolkit_oss.io import load_recording, to_npy, to_sigmf from ria_toolkit_oss.io import load_recording, to_npy, to_sigmf
from ria_toolkit_oss_cli.cli import cli
class TestCombineHelp: class TestCombineHelp:

View File

@ -63,9 +63,7 @@ class TestAutoSelectTxDevice:
patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_uhd_devices", return_value=[]), patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_uhd_devices", return_value=[]),
patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_pluto_devices", return_value=[]), patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_pluto_devices", return_value=[]),
patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices", return_value=[]), patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices", return_value=[]),
patch( patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", return_value=[]),
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", return_value=[]
),
): ):
with pytest.raises(ClickException) as exc_info: with pytest.raises(ClickException) as exc_info:
@ -83,9 +81,7 @@ class TestAutoSelectTxDevice:
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices", "ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices",
return_value=[{"type": "HackRF One", "serial": "123456"}], return_value=[{"type": "HackRF One", "serial": "123456"}],
), ),
patch( patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", return_value=[]),
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", return_value=[]
),
): ):
device_type = auto_select_tx_device(quiet=True) device_type = auto_select_tx_device(quiet=True)
@ -106,9 +102,7 @@ class TestAutoSelectTxDevice:
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices", "ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices",
return_value=[{"type": "HackRF One", "serial": "123456"}], return_value=[{"type": "HackRF One", "serial": "123456"}],
), ),
patch( patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", return_value=[]),
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", return_value=[]
),
): ):
with pytest.raises(ClickException) as exc_info: with pytest.raises(ClickException) as exc_info:
@ -129,15 +123,9 @@ class TestAutoSelectTxDevice:
for device_name, expected_type in test_cases: for device_name, expected_type in test_cases:
with ( with (
patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.load_sdr_drivers"), patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.load_sdr_drivers"),
patch( patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_uhd_devices", return_value=[]),
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_uhd_devices", return_value=[] patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_pluto_devices", return_value=[]),
), patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices", return_value=[]),
patch(
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_pluto_devices", return_value=[]
),
patch(
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_hackrf_devices", return_value=[]
),
patch( patch(
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices", "ria_toolkit_oss_cli.ria_toolkit_oss.transmit.find_bladerf_devices",
return_value=[{"type": device_name}], return_value=[{"type": device_name}],
@ -342,9 +330,7 @@ class TestTransmitCommand:
mock_recording.metadata = {} mock_recording.metadata = {}
with ( with (
patch( patch("ria_toolkit_oss_cli.ria_toolkit_oss.transmit.get_sdr_device", return_value=mock_sdr),
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.get_sdr_device", return_value=mock_sdr
),
patch( patch(
"ria_toolkit_oss_cli.ria_toolkit_oss.transmit.load_input_file", "ria_toolkit_oss_cli.ria_toolkit_oss.transmit.load_input_file",
return_value=mock_recording, return_value=mock_recording,