def METHOD_NAME(self, batch_shape):
    likelihood = self.create_likelihood()
    likelihood.max_plate_nesting += len(batch_shape)
    input = self._create_conditional_input(batch_shape)
    output = likelihood(input)
    self.assertTrue(isinstance(output, Distribution))
    self.assertEqual(output.sample().shape, input.shape)