code: def test_expand(batch_shape):
      d1 = dist.Delta(torch.tensor(1.234))
      d2 = d1.expand(batch_shape)
      assert d2.batch_shape == torch.Size(batch_shape)
  
file_path: pyro/tests/distributions/test_delta.py
imports: from unittest import TestCase
  import numpy
  import pytest
  import torch
  import pyro.distributions
  from tests.common import assert_equal
label_desc: Use image node_id 2 for calling a global function with example usage: test_expand(batch_shape) without return types