def METHOD_NAME(self):
    self._test_conditional(batch_shape=torch.Size([2, 3]))
    self._test_log_marginal(batch_shape=torch.Size([2, 3]))
    self._test_log_prob(batch_shape=torch.Size([2, 3]))
    self._test_marginal(batch_shape=torch.Size([2, 3]))