Skip to content

Commit ea6e2e2

Browse files
committed
fixed tests
1 parent a79cc2a commit ea6e2e2

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

services/autoscaling/tests/unit/test_modules_cluster_scaling_dynamic.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -978,7 +978,7 @@ async def _assert_wait_for_ec2_instances_terminated() -> None:
978978
_ScaleUpParams(
979979
imposed_instance_type=None,
980980
service_resources=Resources(
981-
cpus=4, ram=TypeAdapter(ByteSize).validate_python("128Gib")
981+
cpus=4, ram=TypeAdapter(ByteSize).validate_python("114Gib")
982982
),
983983
num_services=1,
984984
expected_instance_type="r5n.4xlarge",
@@ -990,7 +990,7 @@ async def _assert_wait_for_ec2_instances_terminated() -> None:
990990
_ScaleUpParams(
991991
imposed_instance_type="t2.xlarge",
992992
service_resources=Resources(
993-
cpus=4, ram=TypeAdapter(ByteSize).validate_python("4Gib")
993+
cpus=2.6, ram=TypeAdapter(ByteSize).validate_python("4Gib")
994994
),
995995
num_services=1,
996996
expected_instance_type="t2.xlarge",
@@ -1002,7 +1002,7 @@ async def _assert_wait_for_ec2_instances_terminated() -> None:
10021002
_ScaleUpParams(
10031003
imposed_instance_type="r5n.8xlarge",
10041004
service_resources=Resources(
1005-
cpus=4, ram=TypeAdapter(ByteSize).validate_python("128Gib")
1005+
cpus=4, ram=TypeAdapter(ByteSize).validate_python("114Gib")
10061006
),
10071007
num_services=1,
10081008
expected_instance_type="r5n.8xlarge",
@@ -1165,7 +1165,7 @@ async def test_cluster_scaling_up_and_down_against_aws(
11651165
),
11661166
num_services=10,
11671167
expected_instance_type="r5n.4xlarge", # 1 GPU, 16 CPUs, 128GiB
1168-
expected_num_instances=4,
1168+
expected_num_instances=5,
11691169
),
11701170
id="sim4life-light",
11711171
),
@@ -1254,7 +1254,7 @@ async def test_cluster_scaling_up_starts_multiple_instances(
12541254
_ScaleUpParams(
12551255
imposed_instance_type="g4dn.2xlarge", # 1 GPU, 8 CPUs, 32GiB
12561256
service_resources=Resources(
1257-
cpus=8, ram=TypeAdapter(ByteSize).validate_python("15Gib")
1257+
cpus=6.6, ram=TypeAdapter(ByteSize).validate_python("15Gib")
12581258
),
12591259
num_services=12,
12601260
expected_instance_type="g4dn.2xlarge", # 1 GPU, 8 CPUs, 32GiB
@@ -1263,7 +1263,7 @@ async def test_cluster_scaling_up_starts_multiple_instances(
12631263
_ScaleUpParams(
12641264
imposed_instance_type="g4dn.8xlarge", # 32CPUs, 128GiB
12651265
service_resources=Resources(
1266-
cpus=32, ram=TypeAdapter(ByteSize).validate_python("20480MB")
1266+
cpus=30.6, ram=TypeAdapter(ByteSize).validate_python("20480MB")
12671267
),
12681268
num_services=7,
12691269
expected_instance_type="g4dn.8xlarge", # 32CPUs, 128GiB
@@ -1556,7 +1556,7 @@ async def test_cluster_adapts_machines_on_the_fly( # noqa: PLR0915
15561556
_ScaleUpParams(
15571557
imposed_instance_type=None,
15581558
service_resources=Resources(
1559-
cpus=4, ram=TypeAdapter(ByteSize).validate_python("128Gib")
1559+
cpus=4, ram=TypeAdapter(ByteSize).validate_python("114Gib")
15601560
),
15611561
num_services=1,
15621562
expected_instance_type="r5n.4xlarge",

0 commit comments

Comments
 (0)