Skip to content

Commit af3d816

Browse files
committed
Mock resampler: Improve variable names
Since the mock resampler uses a pair of channels, one to receive samples to be (fakely) resampled (input) and one to send the (fakely) resampmled samples (output), we should use clear names to identify both otherwise is really hard to follow which is which when reading the code. Signed-off-by: Leandro Lucarella <[email protected]>
1 parent 5df1db5 commit af3d816

File tree

1 file changed

+16
-13
lines changed

1 file changed

+16
-13
lines changed

tests/timeseries/mock_resampler.py

Lines changed: 16 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__( # pylint: disable=too-many-arguments
4545
self._resampler_request_channel = Broadcast[ComponentMetricRequest](
4646
"resampler-request"
4747
)
48-
self._basic_receivers: dict[str, list[Receiver[Sample[Quantity]]]] = {}
48+
self._input_channels_receivers: dict[str, list[Receiver[Sample[Quantity]]]] = {}
4949

5050
def power_senders(
5151
comp_ids: list[int],
@@ -58,7 +58,7 @@ def power_senders(
5858
Sample[Quantity], name
5959
).new_sender()
6060
)
61-
self._basic_receivers[name] = [
61+
self._input_channels_receivers[name] = [
6262
self._channel_registry.get_or_create(
6363
Sample[Quantity], name
6464
).new_receiver()
@@ -77,7 +77,7 @@ def frequency_senders(
7777
Sample[Quantity], name
7878
).new_sender()
7979
)
80-
self._basic_receivers[name] = [
80+
self._input_channels_receivers[name] = [
8181
self._channel_registry.get_or_create(
8282
Sample[Quantity], name
8383
).new_receiver(name)
@@ -119,19 +119,19 @@ def multi_phase_senders(
119119
).new_sender(),
120120
]
121121
)
122-
self._basic_receivers[p1_name] = [
122+
self._input_channels_receivers[p1_name] = [
123123
self._channel_registry.get_or_create(
124124
Sample[Quantity], p1_name
125125
).new_receiver()
126126
for _ in range(namespaces)
127127
]
128-
self._basic_receivers[p2_name] = [
128+
self._input_channels_receivers[p2_name] = [
129129
self._channel_registry.get_or_create(
130130
Sample[Quantity], p2_name
131131
).new_receiver()
132132
for _ in range(namespaces)
133133
]
134-
self._basic_receivers[p3_name] = [
134+
self._input_channels_receivers[p3_name] = [
135135
self._channel_registry.get_or_create(
136136
Sample[Quantity], p3_name
137137
).new_receiver()
@@ -222,15 +222,18 @@ async def _handle_resampling_requests(self) -> None:
222222
name = request.get_channel_name()
223223
if name in self._forward_tasks:
224224
continue
225-
basic_recv_name = f"{request.component_id}:{request.metric_id}"
226-
recv = self._basic_receivers[basic_recv_name].pop()
227-
assert recv is not None
225+
input_chan_recv_name = f"{request.component_id}:{request.metric_id}"
226+
input_chan_recv = self._input_channels_receivers[input_chan_recv_name].pop()
227+
assert input_chan_recv is not None
228+
output_chan_sender: Sender[Sample[Quantity]] = (
229+
self._channel_registry.get_or_create(
230+
Sample[Quantity], name
231+
).new_sender()
232+
)
228233
task = asyncio.create_task(
229234
self._channel_forward_messages(
230-
recv,
231-
self._channel_registry.get_or_create(
232-
Sample[Quantity], name
233-
).new_sender(),
235+
input_chan_recv,
236+
output_chan_sender,
234237
),
235238
name=name,
236239
)

0 commit comments

Comments
 (0)