def test_sample_agent(self):
    agent = SampleAgent([RandomAgent(all_actions=False), RandomAgent(all_actions=True)])
    probs = agent.action(None)[1]['action_probs']
    expected_probs = np.array([0.18333333, 0.18333333, 0.18333333, 0.18333333, 0.18333333, 0.08333333])
    self.assertTrue(np.allclose(probs, expected_probs))
def test_sample_agent(self):
    agent = SampleAgent([RandomAgent(all_actions=False), RandomAgent(all_actions=True)])
    probs = agent.action(None)[1]['action_probs']
    expected_probs = np.array([0.18333333, 0.18333333, 0.18333333, 0.18333333, 0.18333333, 0.08333333])
    self.assertTrue(np.allclose(probs, expected_probs))