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_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)