def test_should_be_constructed_from_config_with_default_fallbacks(self):
config = BanditConfig(bandit_id="test_bandit",
arm_ids=["arm1", "arm2", "arm3"])
egreedy: EGreedy = EGreedy.make_from_bandit_config(config)
assert egreedy.bandit_id == "test_bandit"
assert egreedy.epsilon == 0.90
assert egreedy.arms_dict == {
"arm1": EpsilonGreedyArm("arm1", 1, 0),
"arm2": EpsilonGreedyArm("arm2", 1, 0),
"arm3": EpsilonGreedyArm("arm3", 1, 0)
}
def test_should_be_constructed_from_config(self):
config = BanditConfig(bandit_id="test_bandit",
arm_ids=["arm1", "arm2", "arm3"],
priors={
'n': 2,
'reward_sum': 2
},
parameters={'epsilon': 0.80})
egreedy: EGreedy = EGreedy.make_from_bandit_config(config)
assert egreedy.bandit_id == "test_bandit"
assert egreedy.epsilon == 0.80
assert egreedy.arms_dict == {
"arm1": EpsilonGreedyArm("arm1", 2, 2),
"arm2": EpsilonGreedyArm("arm2", 2, 2),
"arm3": EpsilonGreedyArm("arm3", 2, 2)
}