@Test
public void testSimpleDistributedTFConfigGeneratorMaster() {

  String json = TensorFlowConfigEnvGenerator.getTFConfigEnv("master", 2, 1,
      "wtan", "tf-job-001", "example.com");
  assertEquals(json, "{\\\"cluster\\\":{\\\"master\\\":[\\\"master-0.wtan" +
      ".tf-job-001.example.com:8000\\\"],\\\"worker\\\":[\\\"worker-0.wtan" +
      ".tf-job-001.example.com:8000\\\"],\\\"ps\\\":[\\\"ps-0.wtan" +
      ".tf-job-001.example.com:8000\\\"]},\\\"task\\\":{ " +
      "\\\"type\\\":\\\"master\\\", \\\"index\\\":$_TASK_INDEX}," +
      "\\\"environment\\\":\\\"cloud\\\"}");
}