code: def _test_conditional(self, batch_shape):
      likelihood = self.create_likelihood()
      input = self._create_conditional_input(batch_shape)
      output = likelihood(input)
  
      self.assertIsInstance(output, Distribution)
      self.assertEqual(
          output.sample().shape, torch.Size([*batch_shape, 5])
      )
  
file_path: gpytorch/test/likelihoods/test_softmax_likelihood.py
imports: import unittest
  import torch
  from torch.distributions import Distribution
  from gpytorch.distributions import MultitaskMultivariateNormal, MultivariateNormal
  from gpytorch.likelihoods import SoftmaxLikelihood
  from gpytorch.test.base_likelihood_test_case import BaseLikelihoodTestCase
label_desc: Use image node_id 5 for calling the TestSoftmaxLikelihood obj's underlying member method code with example usage: obj._test_conditional(batch_shape) without return types