Skip to content

Commit 3a1f9ed

Browse files
style: auto-format code with ruff
1 parent a64d34e commit 3a1f9ed

6 files changed

Lines changed: 20 additions & 17 deletions

tests/providers/aws/live/test_cleanup_e2e_onaws.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -776,7 +776,14 @@ async def test_run_instances_cleanup(self, setup_cleanup_e2e):
776776

777777
# 2. Poll until provisioning complete
778778
deadline = time.time() + SDK_TIMEOUTS["request_completion"]
779-
terminal = {"complete", "complete_with_error", "partial", "failed", "cancelled", "timeout"}
779+
terminal = {
780+
"complete",
781+
"complete_with_error",
782+
"partial",
783+
"failed",
784+
"cancelled",
785+
"timeout",
786+
}
780787
status_response = None
781788
while True:
782789
status_response = await sdk.get_request_status(request_id=request_id) # type: ignore[attr-defined]

tests/providers/aws/live/test_multi_asg_termination.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -533,9 +533,7 @@ def test_multi_asg_termination(setup_multi_asg_templates):
533533
f"status=complete but got {len(machines)}/{capacity_to_request} machines"
534534
)
535535
else:
536-
assert len(machines) >= 1, (
537-
f"status={_final_status!r} with zero machines"
538-
)
536+
assert len(machines) >= 1, f"status={_final_status!r} with zero machines"
539537

540538
instance_ids = [
541539
machine.get("machineId") or machine.get("machine_id") for machine in machines

tests/providers/aws/live/test_multi_ec2_fleet_termination.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -557,9 +557,7 @@ def test_multi_ec2_fleet_termination(setup_multi_ec2_fleet_templates):
557557
f"status=complete but got {len(machines)}/{capacity_to_request} machines"
558558
)
559559
else:
560-
assert len(machines) >= 1, (
561-
f"status={_final_status!r} with zero machines"
562-
)
560+
assert len(machines) >= 1, f"status={_final_status!r} with zero machines"
563561

564562
instance_ids = [
565563
machine.get("machineId") or machine.get("machine_id") for machine in machines

tests/providers/aws/live/test_multi_resource_termination.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -570,9 +570,7 @@ def test_multi_resource_termination(setup_multi_resource_templates):
570570
f"status=complete but got {len(machines)}/{capacity_to_request} machines"
571571
)
572572
else:
573-
assert len(machines) >= 1, (
574-
f"status={_final_status!r} with zero machines"
575-
)
573+
assert len(machines) >= 1, f"status={_final_status!r} with zero machines"
576574

577575
instance_ids = [
578576
machine.get("machineId") or machine.get("machine_id") for machine in machines

tests/providers/aws/live/test_multi_spot_fleet_termination.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -539,9 +539,7 @@ def test_multi_spot_fleet_termination(setup_multi_spot_fleet_templates):
539539
f"status=complete but got {len(machines)}/{capacity_to_request} machines"
540540
)
541541
else:
542-
assert len(machines) >= 1, (
543-
f"status={_final_status!r} with zero machines"
544-
)
542+
assert len(machines) >= 1, f"status={_final_status!r} with zero machines"
545543

546544
instance_ids = [
547545
machine.get("machineId") or machine.get("machine_id") for machine in machines

tests/providers/aws/live/test_onaws.py

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1157,8 +1157,10 @@ def _check_request_machines_response_status(status_response, requested_count=Non
11571157
# instance came up and the message names the shortfall. Zero-fulfilment
11581158
# or non-capacity-shaped errors still fail loudly.
11591159
_req = status_response["requests"][0]
1160-
_fallback = requested_count if requested_count is not None else len(
1161-
_req.get("machines") or _req.get("machine_ids") or []
1160+
_fallback = (
1161+
requested_count
1162+
if requested_count is not None
1163+
else len(_req.get("machines") or _req.get("machine_ids") or [])
11621164
)
11631165
assert_terminal_ok(status_response, _fallback)
11641166
for machine in status_response["requests"][0]["machines"]:
@@ -1421,8 +1423,10 @@ def _wait_for_request_completion(hfm, request_id: str, scheduler_type: str):
14211423
from tests.providers.aws.live._capacity_helpers import assert_terminal_ok
14221424

14231425
_req = status_response["requests"][0]
1424-
_target = _req.get("target_units") or _req.get("requested_count") or len(
1425-
_req.get("machines") or _req.get("machine_ids") or []
1426+
_target = (
1427+
_req.get("target_units")
1428+
or _req.get("requested_count")
1429+
or len(_req.get("machines") or _req.get("machine_ids") or [])
14261430
)
14271431
assert_terminal_ok(status_response, int(_target))
14281432
return status_response

0 commit comments

Comments
 (0)