class_docstr: Test Container usage.
docstr: Test that indices of neighboring cells can be computed.
docstr_tok: ["Test","that","indices","of","neighboring","cells","can","be","computed","."]
code: def test_compute_neighbor_cells(self):
      """Test that indices of neighboring cells can be computed."""
      N = 10
      start = 0
      stop = 4
      nbr_cutoff = 1
      ndim = 3
      # The number of cells which we should theoretically have
      n_cells = ((stop - start) / nbr_cutoff) ** ndim
  
      # TODO(rbharath): The test below only checks that shapes work out.
      # Need to do a correctness implementation vs. a simple CPU impl.
  
      with self.session() as sess:
          cells = get_cells(start, stop, nbr_cutoff, ndim=ndim)
          nbr_cells = compute_neighbor_cells(cells, ndim, n_cells)
          nbr_cells = nbr_cells.eval()
          assert len(nbr_cells) == n_cells
          nbr_cells = [nbr_cell for nbr_cell in nbr_cells]
          for nbr_cell in nbr_cells:
              assert nbr_cell.shape == (26,)
  
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 6 for calling the TestVinaModel obj's underlying member method code with example usage: obj.test_compute_neighbor_cells() without return types