def test_get_action(action_handler: ActionHandler):
    action_ind = action_handler.get_action([1, 0, 0, 0], random=False)
    assert isinstance(action_ind, np.integer), "expected int got {}".format(
        type(action_ind))
    assert action_ind == 0

    action_handler.get_action([1, 0, 0,
                               0])  # just make sure random doesn't fail
def test_rand_vals():
    # just test to make sure rand vals doesn't fail
    action_handler = ActionHandler(ActionPolicy.randVals, [1, 0.1, 2],
                                   [0, 2, 4, 6])
    action_handler.get_action([0, 0, 0, 0])
def test_rand_vals():
    # just test to make sure rand vals doesn't fail
    action_handler = ActionHandler(ActionPolicy.randVals, [1, 0.1, 2], [0, 2, 4, 6])
    action_handler.get_action([0, 0, 0, 0])
def test_get_action(action_handler: ActionHandler):
    action_ind = action_handler.get_action([1, 0, 0, 0], random=False)
    assert isinstance(action_ind, np.integer), "expected int got {}".format(type(action_ind))
    assert action_ind == 0

    action_handler.get_action([1, 0, 0, 0])  # just make sure random doesn't fail