code: def test_constraint(value_shape):
      value = torch.randn(value_shape).clamp(-2, 2).tril()
      value.diagonal(dim1=-2, dim2=-1).exp_()
      value = value / value.norm(2, dim=-1, keepdim=True)
  
      assert (constraints.corr_cholesky.check(value) == 1).all()
  
file_path: pyro/tests/distributions/test_lkj.py
imports: import math
  import pytest
  import torch
  from torch.distributions import AffineTransform, Beta, TransformedDistribution, biject_to, transform_to
  from pyro.distributions import constraints, transforms
  from pyro.distributions.torch import LKJCholesky
  from tests.common import assert_equal, assert_tensors_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_constraint(value_shape) without return types