Modify tests to work with heterogeneous MPS counts

These tests confirms correct CUDA percentages in an environment
where the gres/mps counts are not always 100 and can very from
device to device within a node.