def METHOD_NAME(
    self,
    model: Module,
    attr_method: Union[DeepLift, DeepLiftShap],
    input: Tensor,
    baselines,
    target: TargetType,
) -> None:
    # TODO add test cases for multiple different layers
    model.zero_grad()
    attributions, delta = attr_method.attribute(
        input, baselines=baselines, target=target, return_convergence_delta=True
    )
    self._assert_attributions(model, attributions, input, baselines, delta, target)
    target2 = torch.tensor(1)
    attributions, delta = attr_method.attribute(
        input, baselines=baselines, target=target2, return_convergence_delta=True
    )
    self._assert_attributions(model, attributions, input, baselines, delta, target2)