def METHOD_NAME(self) -> None:
    model = BasicModel()
    input = torch.tensor([[2.0, -9.0, 9.0, 1.0, -3.0]])
    mask = torch.tensor([[1, 1, 0, 0, 0]])
    adv = PGD(model)
    perturbed_input = adv.perturb(input, 0.25, 0.1, 2, 4, mask=mask)
    assertTensorAlmostEqual(
        self,
        perturbed_input,
        [[2.0, -9.0, 9.0, 1.0, -3.0]],
        delta=0.01,
        mode="max",
    )