def test_update_tensorboard_display_name(self, update_tensorboard_mock):
aiplatform.init(project=_TEST_PROJECT)
my_tensorboard = tensorboard.Tensorboard(tensorboard_name=_TEST_NAME)
my_tensorboard.update(display_name=_TEST_DISPLAY_NAME_UPDATE)
expected_tensorboard = gca_tensorboard.Tensorboard(
name=_TEST_NAME,
display_name=_TEST_DISPLAY_NAME_UPDATE,
)
update_tensorboard_mock.assert_called_once_with(
update_mask=field_mask_pb2.FieldMask(paths=["display_name"]),
tensorboard=expected_tensorboard,
metadata=_TEST_REQUEST_METADATA,
)
def test_update_tensorboard_encryption_spec(self, update_tensorboard_mock):
aiplatform.init(project=_TEST_PROJECT)
my_tensorboard = tensorboard.Tensorboard(tensorboard_name=_TEST_NAME)
my_tensorboard.update(
encryption_spec_key_name=_TEST_ENCRYPTION_KEY_NAME)
expected_tensorboard = gca_tensorboard.Tensorboard(
name=_TEST_NAME,
encryption_spec=_TEST_ENCRYPTION_SPEC,
)
update_tensorboard_mock.assert_called_once_with(
update_mask=field_mask_pb2.FieldMask(paths=["encryption_spec"]),
tensorboard=expected_tensorboard,
metadata=_TEST_REQUEST_METADATA,
)
def test_init_tensorboard_with_invalid_name(self):
with pytest.raises(ValueError):
aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION)
tensorboard.Tensorboard(tensorboard_name=_TEST_INVALID_NAME)
def test_init_tensorboard_with_location_override(self,
get_tensorboard_mock):
aiplatform.init(project=_TEST_PROJECT, location=_TEST_LOCATION)
tensorboard.Tensorboard(tensorboard_name=_TEST_ID,
location=_TEST_ALT_LOCATION)
get_tensorboard_mock.assert_called_once_with(name=_TEST_ALT_NAME)
def test_init_tensorboard_with_id_only_without_project_or_location(self):
with pytest.raises(GoogleAuthError):
tensorboard.Tensorboard(
tensorboard_name=_TEST_ID,
credentials=auth_credentials.AnonymousCredentials(),
)
def test_init_tensorboard(self, get_tensorboard_mock):
aiplatform.init(project=_TEST_PROJECT)
tensorboard.Tensorboard(tensorboard_name=_TEST_NAME)
get_tensorboard_mock.assert_called_once_with(name=_TEST_NAME)
def test_init_tensorboard_with_alt_location(self, get_tensorboard_mock):
aiplatform.init(project=_TEST_PROJECT, location=_TEST_ALT_LOCATION)
tensorboard.Tensorboard(tensorboard_name=_TEST_NAME, )
get_tensorboard_mock.assert_called_once_with(name=_TEST_NAME,
retry=base._DEFAULT_RETRY)
