code: def test_log_prob_phylo(num_leaves, dtype, bp_iters):
      logits, times = random_phylo_logits(num_leaves, dtype)
      d = dist.OneTwoMatching(logits, bp_iters=bp_iters)
      values = d.enumerate_support()
      log_total = d.log_prob(values).logsumexp(0).item()
      logging.info(
          f"log_total = {log_total:0.3g}, "
          + f"log_Z = {d.log_partition_function:0.3g}"
      )
      assert_close(log_total, 0.0, atol=1.0)
  
imports: import logging
  import math
  import pytest
  import torch
  import pyro.distributions
  from tests.common import assert_close, assert_equal, xfail_if_not_implemented
label_desc: Use image node_id 7 for calling a global function with example usage: test_log_prob_phylo(num_leaves, dtype, bp_iters) without return types