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