docstr: Returns a nest given an array.
docstr_tok: ["Returns","a","nest","given","an","array","."]
code: def _make_nest(self, array):
      """Returns a nest given an array."""
      return {
          "a": array,
          "b": [jnp.ones_like(array), {"c": jnp.zeros_like(array)}],
      }
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import jax
  import jax.numpy
  import jax.tree_util
  from jraph._src import graph
  from jraph._src import models
  from jraph._src import utils
  import numpy
label_desc: Use image node_id 1 for calling the ModelsTest obj's underlying member method code with example usage: obj._make_nest(array) and returns: dict