code: def test_DAG_gather():
      """Test invoking DAGGather."""
      # TODO(rbharath): We need more documentation about why
      # these numbers work.
      batch_size = 10
      n_graph_feat = 30
      n_atom_feat = 30
      n_outputs = 75
      max_atoms = 50
      layer_sizes = [100]
      layer = layers.DAGGather(
          n_graph_feat=n_graph_feat,
          n_outputs=n_outputs,
          max_atoms=max_atoms,
          layer_sizes=layer_sizes,
      )
      atom_features = np.random.rand(batch_size, n_atom_feat)
      membership = np.sort(
          np.random.randint(0, batch_size, size=(batch_size))
      )
      outputs = layer([atom_features, membership])
  
imports: import deepchem
  import numpy
  import pytest
  import os
label_desc: Use image node_id 27 for calling a global function with example usage: test_DAG_gather() without return types