def METHOD_NAME(dist, values):
    if values is None:
        values = dist.enumerate_support()
    log_probs = dist.log_prob(values)
    assert_close(log_probs.logsumexp(0), torch.tensor(0.0), atol=0.01)