more work

import deepchem.models.layers as layers

from tensorflow.python.framework import test_util

def test_cosine_dist():

  """Test invoking _cosine_dist."""

  x = tf.ones((5, 4), dtype=tf.dtypes.float32, name=None)

  assert tf.reduce_sum(cos_sim_orth) == 0  # True

  assert all([cos_sim_orth.shape[dim] == 256 for dim in range(2)])  # True

def test_highway():

  """Test invoking Highway."""

  width = 5