class_docstr: Test for ContextConditionedPolicy.
class_docstr_tok: ["Test","for","ContextConditionedPolicy","."]
docstr: Test get_action.
docstr_tok: ["Test","get_action","."]
code: def test_get_action(self):
      """Test get_action."""
      obs = np.random.rand(self.obs_dim)
      action, _ = self.module.get_action(obs)
      assert len(action) == self.action_dim
  
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 7 for calling the TestContextConditionedPolicy obj's underlying member method code with example usage: obj.test_get_action() without return types