class_docstr: Test for ContextConditionedPolicy.
class_docstr_tok: ["Test","for","ContextConditionedPolicy","."]
docstr: Test sample_from_belief.
docstr_tok: ["Test","sample_from_belief","."]
code: def test_sample_from_belief(self):
      """Test sample_from_belief."""
      self.module.sample_from_belief()
      expected_shape = [1, self.latent_dim]
      assert all(
          [a == b for a, b in zip(self.module.z.shape, expected_shape)]
      )
  
file_path: garage/tests/garage/torch/policies/test_context_conditioned_policy.py
imports: import akro
  import numpy
  import torch
  import torch.nn
  from torch.nn import functional
  from garage import EnvSpec, StepType, TimeStep
  from garage.envs import GymEnv
  from garage.torch.embeddings import MLPEncoder
  from garage.torch.policies import ContextConditionedPolicy, TanhGaussianMLPPolicy
  from tests.fixtures.envs.dummy import DummyBoxEnv
label_desc: Use image node_id 3 for calling the TestContextConditionedPolicy obj's underlying member method code with example usage: obj.test_sample_from_belief() without return types