diff --git a/tests/integration-tests/tests/efa/test_efa.py b/tests/integration-tests/tests/efa/test_efa.py index 1e3059db87..79c0beeb06 100644 --- a/tests/integration-tests/tests/efa/test_efa.py +++ b/tests/integration-tests/tests/efa/test_efa.py @@ -59,11 +59,17 @@ def _test_efa_installed(scheduler_commands, remote_command_executor): # Check EFA interface is present on compute node result = remote_command_executor.run_remote_command("cat /shared/lspci.out") - assert_that(result.stdout).contains("00:06.0 Ethernet controller: Amazon.com, Inc. Device efa0") + assert_that( + ("Ethernet controller: Amazon.com, Inc. Elastic Fabric Adapter (EFA)" in result.stdout) + or ("Ethernet controller: Amazon.com, Inc. Device efa0" in result.stdout) + ).is_true() # Check EFA interface not present on master result = remote_command_executor.run_remote_command("lspci") - assert_that(result.stdout).does_not_contain("00:06.0 Ethernet controller: Amazon.com, Inc. Device efa0") + assert_that( + ("Ethernet controller: Amazon.com, Inc. Elastic Fabric Adapter (EFA)" in result.stdout) + or ("Ethernet controller: Amazon.com, Inc. Device efa0" in result.stdout) + ).is_false() def _test_osu_benchmarks(mpi_version, remote_command_executor, scheduler_commands, test_datadir, slots_per_instance):