code: def test_shape(expression, expected_shape):
      result = eval(expression)
      assert result.shape == expected_shape
  
file_path: pyro/tests/ops/test_indexing.py
imports: import itertools
  import pytest
  import torch
  import pyro.distributions
  from pyro.distributions.util import broadcast_shape
  from pyro.ops.indexing import Index, Vindex
  from tests.common import assert_equal
label_desc: Use image node_id 2 for calling a global function with example usage: test_shape(expression, expected_shape) without return types