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)