Skip to content

Commit a684dae

Browse files
[Test] Make test_job_level_scaling head node instance type architecture agnostic
The test is run both with x86 and ARM Signed-off-by: Hanwen <[email protected]>
1 parent 30a2d01 commit a684dae

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

tests/integration-tests/tests/scaling/test_scaling.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828
from tests.schedulers.test_slurm import _assert_job_state
2929

3030

31-
@pytest.mark.usefixtures("os", "instance")
31+
@pytest.mark.usefixtures("os")
3232
@pytest.mark.parametrize(
3333
"scaling_strategy, scaling_behaviour_by_capacity",
3434
[
@@ -185,6 +185,7 @@
185185
)
186186
def test_job_level_scaling(
187187
region,
188+
instance,
188189
pcluster_config_reader,
189190
clusters_factory,
190191
scheduler_commands_factory,
@@ -201,6 +202,7 @@ def test_job_level_scaling(
201202
scaledown_idletime=scaledown_idletime,
202203
scaling_strategy=scaling_strategy,
203204
no_of_queues=no_of_queues,
205+
head_node_instance_type=instance.replace(".xlarge", ".2xlarge"),
204206
)
205207
cluster = clusters_factory(cluster_config)
206208
remote_command_executor = RemoteCommandExecutor(cluster)

tests/integration-tests/tests/scaling/test_scaling/test_job_level_scaling/pcluster.config.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
Image:
22
Os: {{ os }}
33
HeadNode:
4-
InstanceType: "c5.2xlarge"
4+
InstanceType: {{ head_node_instance_type }}
55
Networking:
66
SubnetId: {{ public_subnet_id }}
77
Ssh:

0 commit comments

Comments
 (0)