Searched refs:mock_service_client (Results 1 – 1 of 1) sorted by relevance
118 def mock_service_client(self, tpu_map=None): member in TPUClusterResolverTest170 service=self.mock_service_client(tpu_map=tpu_map),205 service=self.mock_service_client(tpu_map=tpu_map))232 service=self.mock_service_client(tpu_map=tpu_map))253 service=self.mock_service_client(tpu_map=tpu_map))281 service=self.mock_service_client(tpu_map=tpu_map))323 service=self.mock_service_client(tpu_map=tpu_map),375 service=self.mock_service_client(tpu_map=tpu_map))400 service=self.mock_service_client(tpu_map=tpu_map))410 service=self.mock_service_client(tpu_map={}))[all …]