Skip to content

Commit 83d8f75

Browse files
committed
addressing linter complaint
1 parent d0608a1 commit 83d8f75

File tree

1 file changed

+16
-11
lines changed

1 file changed

+16
-11
lines changed

test/unit/test_edge_config.py

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515

1616
CUSTOM_REFRESH_RATE = 10.0
1717
CUSTOM_AUDIT_RATE = 0.0
18+
REFRESH_RATE_SECONDS = 15.0
1819

1920

2021
def _make_detector(detector_id: str) -> Detector:
@@ -178,30 +179,34 @@ def test_detectors_config_to_payload_shape():
178179

179180
def test_edge_endpoint_config_accepts_top_level_payload_shape():
180181
"""Accepts the top-level edge endpoint payload shape used by APIs."""
181-
config = EdgeEndpointConfig.model_validate({
182-
"global_config": {"refresh_rate": CUSTOM_REFRESH_RATE},
183-
"edge_inference_configs": {"default": {"enabled": True}},
184-
"detectors": [{"detector_id": "det_1", "edge_inference_config": "default"}],
185-
})
182+
config = EdgeEndpointConfig.model_validate(
183+
{
184+
"global_config": {"refresh_rate": CUSTOM_REFRESH_RATE},
185+
"edge_inference_configs": {"default": {"enabled": True}},
186+
"detectors": [{"detector_id": "det_1", "edge_inference_config": "default"}],
187+
}
188+
)
186189

187190
assert config.global_config.refresh_rate == CUSTOM_REFRESH_RATE
188191
assert [detector.detector_id for detector in config.detectors] == ["det_1"]
189192

190193

191194
def test_edge_endpoint_config_from_yaml_accepts_yaml_text():
192195
"""Parses edge-endpoint YAML text using EdgeEndpointConfig.from_yaml."""
193-
config = EdgeEndpointConfig.from_yaml(yaml_str="""
196+
config = EdgeEndpointConfig.from_yaml(
197+
yaml_str=f"""
194198
global_config:
195-
refresh_rate: 15.0
199+
refresh_rate: {REFRESH_RATE_SECONDS}
196200
edge_inference_configs:
197201
default:
198202
enabled: true
199203
detectors:
200204
- detector_id: det_1
201205
edge_inference_config: default
202-
""")
206+
"""
207+
)
203208

204-
assert config.global_config.refresh_rate == 15.0
209+
assert config.global_config.refresh_rate == REFRESH_RATE_SECONDS
205210
assert [detector.detector_id for detector in config.detectors] == ["det_1"]
206211

207212

@@ -272,14 +277,14 @@ def test_edge_endpoint_config_from_payload_round_trip():
272277
def test_edge_endpoint_config_from_payload_accepts_literal_payload():
273278
"""Constructs EdgeEndpointConfig from a literal payload dictionary."""
274279
payload = {
275-
"global_config": {"refresh_rate": 15.0},
280+
"global_config": {"refresh_rate": REFRESH_RATE_SECONDS},
276281
"edge_inference_configs": {"default": {"enabled": True}},
277282
"detectors": [{"detector_id": "det_1", "edge_inference_config": "default"}],
278283
}
279284

280285
config = EdgeEndpointConfig.from_payload(payload)
281286

282-
assert config.global_config.refresh_rate == 15.0
287+
assert config.global_config.refresh_rate == REFRESH_RATE_SECONDS
283288
assert config.edge_inference_configs["default"].name == "default"
284289
assert [detector.detector_id for detector in config.detectors] == ["det_1"]
285290

0 commit comments

Comments
 (0)