fix test case to workaround vmem ratio bug in Yarn.