class_docstr: Test Container usage.
docstr: Test that neighbor list can be computed with tensorflow
docstr_tok: ["Test","that","neighbor","list","can","be","computed","with","tensorflow"]
code: def test_compute_neighbor_list(self):
      """Test that neighbor list can be computed with tensorflow"""
      N = 10
      start = 0
      stop = 12
      nbr_cutoff = 3
      ndim = 3
      M = 6
      k = 5
      # The number of cells which we should theoretically have
      n_cells = int(((stop - start) / nbr_cutoff) ** ndim)
  
      with self.session() as sess:
          coords = start + np.random.rand(N, ndim) * (stop - start)
          coords = tf.stack(coords)
          nbr_list = compute_neighbor_list(
              coords, nbr_cutoff, N, M, n_cells, ndim=ndim, k=k
          )
          nbr_list = nbr_list.eval()
          assert nbr_list.shape == (N, M)
  
imports: from __future__ import print_function
  from __future__ import division
  from __future__ import unicode_literals
  import os
  import unittest
  import tensorflow
  import deepchem
  import numpy
  from tensorflow.python.framework import test_util
  from deepchem.models.tensorflow_models.vina_model import VinaModel
  from deepchem.models.tensorflow_models.vina_model import get_cells
  from deepchem.models.tensorflow_models.vina_model import put_atoms_in_cells
  from deepchem.models.tensorflow_models.vina_model import compute_neighbor_cells
  from deepchem.models.tensorflow_models.vina_model import compute_closest_neighbors
  from deepchem.models.tensorflow_models.vina_model import get_cells_for_atoms
  from deepchem.models.tensorflow_models.vina_model import compute_neighbor_list
  import deepchem.utils.rdkit_util
  from deepchem.utils import pad_array
label_desc: Use image node_id 4 for calling the TestVinaModel obj's underlying member method code with example usage: obj.test_compute_neighbor_list() without return types