bc08dbfbe77554bf3458529285003f0bf92eecd3,sonnet/python/modules/nets/vqvae_test.py,VqvaeTest,testShapeChecking,#VqvaeTest#Any#Any#,77

Before Change

    output = vqvae(tf.constant(wrong_shape_input.astype(np.float32)),
                   is_training=False)

    init_op = tf.global_variables_initializer()
    with self.test_session() as session:
      session.run(init_op)
      with self.assertRaisesRegexp(tf.errors.InvalidArgumentError,
                                   "assertion failed"):
        session.run(output)

After Change

  def testShapeChecking(self, constructor, kwargs):
    vqvae = constructor(**kwargs)
    wrong_shape_input = np.random.randn(100, kwargs["embedding_dim"] * 2)
    with self.assertRaisesRegexp(ValueError, "Cannot reshape a tensor"):
      vqvae(tf.constant(wrong_shape_input.astype(np.float32)),
            is_training=False)

Instances