code: def test_rmv(A, b):
      assert_equal(rmv(A, b), A.mv(b), prec=1e-8)
      batched_A = lexpand(A, 5, 4)
      batched_b = lexpand(b, 5, 4)
      expected_Ab = lexpand(A.mv(b), 5, 4)
      assert_equal(rmv(batched_A, batched_b), expected_Ab, prec=1e-8)
  
file_path: pyro/tests/contrib/test_util.py
imports: from collections import OrderedDict
  import pytest
  import torch
  from pyro.contrib.util import get_indices, lexpand, rdiag, rexpand, rmv, rtril, rvv, tensor_to_dict
  from tests.common import assert_equal
label_desc: Use image node_id 3 for calling a global function with example usage: test_rmv(A, b) without return types