Allow extra error message in mpi ssh for when no protocol is specified

aws · Dec 2, 2024 · 83047ca · 83047ca
1 parent 320abfc
commit 83047ca
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/integration-tests/tests/basic/test_essential_features.py b/tests/integration-tests/tests/basic/test_essential_features.py
@@ -147,7 +147,7 @@ def _test_mpi_ssh(remote_command_executor, test_datadir, scheduler_commands_fact
 
     # mpirun_out_ip = ["Warning: Permanently added '192.168.60.89' (ECDSA) to the list of known hosts.",
     # '', 'ip-192-168-60-89']
-    assert_that(len(mpirun_out_ip)).is_equal_to(3)
+    assert_that(len(mpirun_out_ip)).is_greater_than_or_equal_to(3)
     assert_that(mpirun_out_ip[-1]).is_equal_to(remote_host)
 
     mpirun_out = remote_command_executor.run_remote_script(
@@ -156,7 +156,7 @@ def _test_mpi_ssh(remote_command_executor, test_datadir, scheduler_commands_fact
 
     # mpirun_out = ["Warning: Permanently added 'ip-192-168-60-89,192.168.60.89' (ECDSA) to the list of known hosts.",
     # '', 'ip-192-168-60-89']
-    assert_that(len(mpirun_out)).is_equal_to(3)
+    assert_that(len(mpirun_out)).is_greater_than_or_equal_to(3)
     assert_that(mpirun_out[-1]).is_equal_to(remote_host)