code: def test_get_action():
      # discrete control
      action_space = DiscreteActionSpace(3)
      policy = Greedy(action_space)
  
      best_action, _ = policy.get_action(np.array([10, 20, 30]))
      assert best_action == 2
  
      # continuous control
      action_space = BoxActionSpace(np.array([10]))
      policy = Greedy(action_space)
  
      best_action = policy.get_action(np.array([1, 1, 1]))
      assert np.all(best_action == np.array([1, 1, 1]))
  
file_path: coach/rl_coach/tests/exploration_policies/test_greedy.py
imports: import os
  import sys
  import pytest
  from rl_coach.spaces import DiscreteActionSpace, BoxActionSpace
  from rl_coach.exploration_policies.greedy import Greedy
  import numpy
label_desc: Use image node_id 1 for calling a global function with example usage: test_get_action() without return types