gr-mcp/tests/utils.py

18 lines
585 B
Python

from gnuradio_mcp.models import BlockModel
def get_block_keys(platform):
return list(platform.blocks.keys())
def add_block(flowgraph_middleware, block_key, block_name):
flowgraph_middleware.add_block(block_key, block_name)
def remove_block(flowgraph_middleware, block_name):
flowgraph_middleware._flowgraph.remove_element(
flowgraph_middleware._flowgraph.get_block(block_name)
)
def get_current_blocks(flowgraph_middleware):
return [
BlockModel(key=block.key, label=block.label)
for block in flowgraph_middleware._flowgraph.blocks
]