Tweak test for certain configurations

Depending upon topology and node weight, ordering of nodes may not
  be as test expected. Explicitly pick nodes to use in srun within
  an salloc.