def testRecoverableNoState(self): tpu_map = { 'projects/test-project/locations/us-central1-c/nodes/tpu_name': { 'ipAddress': '10.1.2.3', 'port': '8470', } } client = cloud_tpu_client.CloudTPUClient( tpu='tpu_name', service=self.mock_service_client(tpu_map=tpu_map)) self.assertEqual(True, client.recoverable())
def testInitializeWithoutMetadata(self): client = cloud_tpu_client.CloudTPUClient(tpu='tpu_name', project='project', zone='zone') self.assertEqual('tpu_name', client._tpu) self.assertEqual(True, client._use_api) self.assertEqual(None, client._service) self.assertEqual(None, client._credentials) self.assertEqual('project', client._project) self.assertEqual('zone', client._zone) self.assertEqual(None, client._discovery_url)
def testInitializeTpuName(self): tpu_map = { 'projects/test-project/locations/us-central1-c/nodes/tpu_name': { 'ipAddress': '10.1.2.3', 'port': '8470', 'health': 'HEALTHY' } } client = cloud_tpu_client.CloudTPUClient( tpu='tpu_name', service=self.mock_service_client(tpu_map=tpu_map)) self.assertClientContains(client)
def testInitializeNoArgumentsWithEnvironmentVariable(self): os.environ['TPU_NAME'] = 'tpu_name' tpu_map = { 'projects/test-project/locations/us-central1-c/nodes/tpu_name': { 'ipAddress': '10.1.2.3', 'port': '8470', 'health': 'HEALTHY' } } client = cloud_tpu_client.CloudTPUClient( service=self.mock_service_client(tpu_map=tpu_map)) self.assertClientContains(client)
def testInitializeIpAddress(self): client = cloud_tpu_client.CloudTPUClient(tpu='grpc://1.2.3.4:8470') self.assertEqual('grpc://1.2.3.4:8470', client._tpu) self.assertEqual(False, client._use_api) self.assertEqual(None, client._service) self.assertEqual(None, client._credentials) self.assertEqual(None, client._project) self.assertEqual(None, client._zone) self.assertEqual(None, client._discovery_url) self.assertEqual([{ 'ipAddress': '1.2.3.4', 'port': '8470' }], client.network_endpoints())
def testRecoverableNoApiAccess(self): client = cloud_tpu_client.CloudTPUClient(tpu='grpc://1.2.3.4:8470') self.assertEqual(True, client.recoverable())
def testInitializeMultiElementTpuArray(self): with self.assertRaisesRegex( NotImplementedError, 'Using multiple TPUs in a single session is not yet implemented' ): cloud_tpu_client.CloudTPUClient(tpu=['multiple', 'elements'])
def testInitializeNoArguments(self): with self.assertRaisesRegex( ValueError, 'Please provide a TPU Name to connect to.'): cloud_tpu_client.CloudTPUClient()