diff --git a/tests/everflow/everflow_test_utilities.py b/tests/everflow/everflow_test_utilities.py index 754d8349328..170675d3872 100644 --- a/tests/everflow/everflow_test_utilities.py +++ b/tests/everflow/everflow_test_utilities.py @@ -355,8 +355,11 @@ def setup_info(duthosts, rand_one_dut_hostname, tbinfo, request, topo_scenario): duthost = None topo = tbinfo['topo']['name'] if 't2' in topo: - pytest_assert(len(duthosts) > 1, "Test must run on whole chassis") - downstream_duthost, upstream_duthost = get_t2_duthost(duthosts, tbinfo) + if len(duthosts) == 1: + downstream_duthost = upstream_duthost = duthost = duthosts[rand_one_dut_hostname] + else: + pytest_assert(len(duthosts) > 2, "Test must run on whole chassis") + downstream_duthost, upstream_duthost = get_t2_duthost(duthosts, tbinfo) else: downstream_duthost = upstream_duthost = duthost = duthosts[rand_one_dut_hostname]