def METHOD_NAME():
    # Regression test for https://github.com/pyro-ppl/pyro/issues/2615
    dist = LKJCholesky(3, concentration=torch.ones(())).expand([12])
    # batch shape and event shape are as you'd expect
    assert dist.batch_shape == torch.Size([12])
    assert dist.event_shape == torch.Size([3, 3])
    # samples have correct shape when sample_shape=()
    assert dist.shape(()) == torch.Size([12, 3, 3])
    assert dist.sample().shape == torch.Size([12, 3, 3])
    # samples had the wrong shape when sample_shape is non-unit
    assert dist.shape((4,)) == torch.Size([4, 12, 3, 3])
    assert dist.sample((4,)).shape == torch.Size([4, 12, 3, 3])