def METHOD_NAME(shape):
    d = NormalBomb(0, 1).mask(False)
    if shape is not None:
        d = d.expand(shape)
    x = d.sample()
    actual = d.log_prob(x)
    assert_equal(actual, torch.zeros(shape if shape else ()))
    actual = d.score_parts(x)
    assert_equal(actual.log_prob, torch.zeros(shape if shape else ()))