def test_pure2mixed(): num_actions = 3 pure_action = 0 mixed_action = [1.0, 0.0, 0.0] assert_array_equal(pure2mixed(num_actions, pure_action), mixed_action)
def test_pure2mixed(): num_actions = 3 pure_action = 0 mixed_action = [1., 0., 0.] assert_array_equal(pure2mixed(num_actions, pure_action), mixed_action)