code: def test_rejection_standard_gamma_sample_shape(
      sample_shape, batch_shape
  ):
      alphas = torch.ones(batch_shape)
      dist = RejectionStandardGamma(alphas)
      x = dist.rsample(sample_shape)
      assert x.shape == sample_shape + batch_shape
  
file_path: pyro/tests/distributions/test_rejector.py
imports: import pytest
  import torch
  from torch.autograd import grad
  from pyro.distributions import Exponential, Gamma
  from pyro.distributions.testing.rejection_exponential import RejectionExponential
  from pyro.distributions.testing.rejection_gamma import RejectionGamma, RejectionStandardGamma, ShapeAugmentedBeta, ShapeAugmentedGamma
  from tests.common import assert_equal
label_desc: Use image node_id 1 for calling a global function with example usage: test_rejection_standard_gamma_sample_shape(sample_shape, batch_shape) without return types