class_docstr: Test Container usage.
docstr: Test that atoms are placed in the correct cells.
docstr_tok: ["Test","that","atoms","are","placed","in","the","correct","cells","."]
code: def test_get_cells_for_atoms(self):
      """Test that atoms are placed in the correct cells."""
      N = 10
      start = 0
      stop = 4
      nbr_cutoff = 1
      ndim = 3
      k = 5
      # 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)
          coords = np.random.rand(N, ndim)
          cells_for_atoms = get_cells_for_atoms(
              coords, cells, N, n_cells, ndim
          )
          cells_for_atoms = cells_for_atoms.eval()
          assert cells_for_atoms.shape == (N, 1)
  
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 8 for calling the TestVinaModel obj's underlying member method code with example usage: obj.test_get_cells_for_atoms() without return types