code: def test_energy_conservation(example):
      model, args = example
      q_f, p_f, _, _ = velocity_verlet(
          args.q_i,
          args.p_i,
          model.potential_fn,
          model.kinetic_grad,
          args.step_size,
          args.num_steps,
      )
      energy_initial = model.energy(args.q_i, args.p_i)
      energy_final = model.energy(q_f, p_f)
      logger.info("initial energy: {}".format(energy_initial.item()))
      logger.info("final energy: {}".format(energy_final.item()))
      assert_equal(energy_final, energy_initial)
  
imports: import logging
  from collections import namedtuple
  import pytest
  import torch
  from pyro.ops.integrator import velocity_verlet
  from tests.common import assert_equal
label_desc: Use image node_id 3 for calling a global function with example usage: test_energy_conservation(example) without return types