yapf

import deepchem as dc

import scipy.ndimage

class TestTransformers(unittest.TestCase):

  """

  Test top-level API for transformer objects.

  def test_data_transformers(self):

    mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

    # extracting validation set of MNIST for testing the DataTransforms

      valid = dc.data.NumpyDataset(mnist.validation.images, mnist.validation.labels)

    valid = dc.data.NumpyDataset(mnist.validation.images,

                                 mnist.validation.labels)

    # extract only the images (no need of the labels)

    data = (valid.X)[0]

    # reshaping the vector to image