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) }