@Test
public void testPSComponentNumPSIsOne() throws IOException {

  testCommons.yarnConfig.set("hadoop.registry.dns.domain-name", "testDomain");
  TensorFlowRunJobParameters parameters = new TensorFlowRunJobParameters();
  parameters.setPsResource(testCommons.resource);
  parameters.setName("testJobName");
  parameters.setNumPS(1);
  parameters.setPsDockerImage("testPSDockerImage");
  TensorFlowPsComponent psComponent =
      createPsComponent(parameters);
  Component component = psComponent.createComponent();
  assertEquals(1L, (long) component.getNumberOfContainers());
  verifyCommons(component);
}