tests for Avg Pool

from deepchem.feat.graph_features import ConvMolFeaturizer

from deepchem.feat.mol_graphs import ConvMol

from deepchem.models.tensorgraph.layers import Add, MaxPool2D, MaxPool3D, GraphCNN, GraphEmbedPoolLayer, Cast

from deepchem.models.tensorgraph.layers import Add, MaxPool2D, MaxPool3D, AvgPool2D, GraphCNN, GraphEmbedPoolLayer, Cast

from deepchem.models.tensorgraph.layers import AlphaShareLayer

from deepchem.models.tensorgraph.layers import AttnLSTMEmbedding

from deepchem.models.tensorgraph.layers import BatchNorm

      out_tensor = out_tensor.eval()

      assert out_tensor.shape == (batch_size, 1, 1, 1, in_channels)

  def test_avgpool2D(self):

    """Test that AvgPool2D can be invoked."""

    length = 4

    width = 4

    in_channels = 5

    batch_size = 10

    in_tensor = np.random.rand(batch_size, length, width, in_channels)

    with self.test_session() as sess:

      in_tensor = tf.convert_to_tensor(in_tensor, dtype=tf.float32)

      out_tensor = AvgPool2D()(in_tensor)

      sess.run(tf.global_variables_initializer())

      out_tensor = out_tensor.eval()

      assert out_tensor.shape == (batch_size, 2, 2, in_channels)

  def test_input_fifo_queue(self):

    """Test InputFifoQueue can be invoked."""

    batch_size = 10