def METHOD_NAME(self) -> None:
    num = torch.tensor(4.0)
    assert safe_div(num, 2) == 2.0
    assert safe_div(num, 0, 2) == 2.0
    assert safe_div(num, 2.0) == 2.0
    assert safe_div(num, 0.0, 2.0) == 2.0