def test_parse_empty_task(self):
cluster_spec = "worker|worker0:2220,ps|"
with self.assertRaisesRegexp(ValueError,
"Empty task string at position 0"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec,
self._cluster)
def test_parse_empty_job_name(self):
cluster_spec = "worker|worker0:2220,|ps0:3220"
with self.assertRaisesRegexp(ValueError,
"Empty job_name in cluster_spec"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
print(self._cluster)
def test_parse_misused_semicolon_for_comma(self):
cluster_spec = "worker|worker0:2220;ps|ps0:3220"
with self.assertRaisesRegexp(ValueError,
"Not exactly one instance of \\'\\|\\'"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec,
self._cluster)
def test_parse_empty_job_name(self):
cluster_spec = "worker|worker0:2220,|ps0:3220"
with self.assertRaisesRegexp(ValueError,
"Empty job_name in cluster_spec"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
print(self._cluster)
def test_parse_multi_jobs_sunnyday(self):
cluster_spec = ("worker|worker0:2220;worker1:2221;worker2:2222,"
"ps|ps0:3220;ps1:3221")
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
self.assertEqual(2, len(self._cluster.job))
self.assertEqual("worker", self._cluster.job[0].name)
self.assertEqual(3, len(self._cluster.job[0].tasks))
self.assertEqual("worker0:2220", self._cluster.job[0].tasks[0])
self.assertEqual("worker1:2221", self._cluster.job[0].tasks[1])
self.assertEqual("worker2:2222", self._cluster.job[0].tasks[2])
self.assertEqual("ps", self._cluster.job[1].name)
self.assertEqual(2, len(self._cluster.job[1].tasks))
self.assertEqual("ps0:3220", self._cluster.job[1].tasks[0])
self.assertEqual("ps1:3221", self._cluster.job[1].tasks[1])
def test_parse_multi_jobs_sunnyday(self):
cluster_spec = ("worker|worker0:2220;worker1:2221;worker2:2222,"
"ps|ps0:3220;ps1:3221")
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
self.assertEqual(2, len(self._cluster.job))
self.assertEqual("worker", self._cluster.job[0].name)
self.assertEqual(3, len(self._cluster.job[0].tasks))
self.assertEqual("worker0:2220", self._cluster.job[0].tasks[0])
self.assertEqual("worker1:2221", self._cluster.job[0].tasks[1])
self.assertEqual("worker2:2222", self._cluster.job[0].tasks[2])
self.assertEqual("ps", self._cluster.job[1].name)
self.assertEqual(2, len(self._cluster.job[1].tasks))
self.assertEqual("ps0:3220", self._cluster.job[1].tasks[0])
self.assertEqual("ps1:3221", self._cluster.job[1].tasks[1])
def test_empty_cluster_spec_string(self):
cluster_spec = ""
with self.assertRaisesRegexp(ValueError, "Empty cluster_spec string"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec,
self._cluster)
def test_parse_empty_task(self):
cluster_spec = "worker|worker0:2220,ps|"
with self.assertRaisesRegexp(ValueError,
"Empty task string at position 0"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
def test_parse_misused_semicolon_for_comma(self):
cluster_spec = "worker|worker0:2220;ps|ps0:3220"
with self.assertRaisesRegexp(ValueError,
"Not exactly one instance of \\'\\|\\'"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
def test_empty_cluster_spec_string(self):
cluster_spec = ""
with self.assertRaisesRegexp(ValueError,
"Empty cluster_spec string"):
grpc_tensorflow_server.parse_cluster_spec(cluster_spec, self._cluster)
