class_docstr: Test Container usage.
docstr: Test that tensorflow can compute grid cells.
docstr_tok: ["Test","that","tensorflow","can","compute","grid","cells","."]
code: def test_get_cells(self):
      """Test that tensorflow can compute grid cells."""
      N = 10
      start = 0
      stop = 4
      nbr_cutoff = 1
      with self.session() as sess:
          ndim = 3
          cells = get_cells(start, stop, nbr_cutoff, ndim=ndim).eval()
          assert len(cells.shape) == 2
          assert cells.shape[0] == 4**ndim
  
          ndim = 2
          cells = get_cells(start, stop, nbr_cutoff, ndim=ndim).eval()
          assert len(cells.shape) == 2
          assert cells.shape[0] == 4**ndim
  
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 3 for calling the TestVinaModel obj's underlying member method code with example usage: obj.test_get_cells() without return types