import tensorflow as tf

    hparams = get_default_hparams()
    energy_fn, _, _ = l2hmc.get_scg_energy_fn()
    dynamics = l2hmc.Dynamics(
        x_dim=hparams.x_dim,
        minus_loglikelihood_fn=energy_fn,
        n_steps=hparams.n_steps,
        eps=hparams.eps)
    samples = tf.random_normal(shape=[hparams.n_samples, hparams.x_dim])
    x_, v_, x_accept_prob, x_out = dynamics.apply_transition(samples)

    self.assertEqual(x_.shape, v_.shape)
    self.assertEqual(x_out.shape, samples.shape)
    self.assertEqual(x_.shape, x_out.shape)
    self.assertEqual(x_accept_prob.shape, (hparams.n_samples,))
