code: def test_cholesky_uncholesky(seeded_rng):
      size = 3
      flat_chol_length = (size * (size + 1)) // 2
      flatchol = np.random.normal(size=[flat_chol_length])
      unflatchol = unflatten_cholesky_unique(flatchol)
      npt.assert_equal(unflatchol.shape, [3, 3])
      reflatchol = flatten_cholesky_unique(unflatchol)
      npt.assert_allclose(flatchol, reflatchol)
  
imports: from brainiak.matnormal.utils import pack_trainable_vars, unpack_trainable_vars, flatten_cholesky_unique, unflatten_cholesky_unique
  import tensorflow
  import numpy
  import numpy.testing
label_desc: Use image node_id 2 for calling a global function with example usage: test_cholesky_uncholesky(seeded_rng) without return types