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