Beispiel #1
0
 def test_brew_middle_recursive(self):
     witch = Witch((3, 2, 2, 1), 0)
     brew_action, score, brew_id = witch.recursive_find_best_action(
         ALL_ACTIONS, 1)
     self.assertEqual(69, brew_action.action_id)
     self.assertEqual(13, score)
     self.assertEqual(69, brew_id)
Beispiel #2
0
 def test_find_cast_deep_4(self):
     witch = Witch((3, 0, 0, 0), 0)
     all_actions = {
         "BREW": POTIONS + [BrewAction((0, 0, 0, -1), 100, 1)],
         "CAST": SORTS,
     }
     cast_action, score, brew_id = witch.recursive_find_best_action(
         all_actions, 4)
     self.assertEqual(79, cast_action.action_id)
     self.assertEqual(1, brew_id)
Beispiel #3
0
    def test_choose_sort_3_avoid_too_much_of_1_ingredient(self):
        witch_state = Witch((3, 0, 0, 0), 0)
        sorts = [
            CastAction((-1, 4, 0, 0), 0, 1, 0, 0, True, True),
            CastAction((-1, 1, 1, 1), 0, 2, 0, 0, True, True),
        ]
        potions = [
            BrewAction((1, 1, 1, 1), 0, 5, 0, 0, True, True),
        ]
        sort = witch_state.find_best_cast(sorts, potions)
        self.assertEqual(2, sort.action_id)

        sort = witch_state.find_best_cast(sorts[::-1], potions)
        self.assertEqual(2, sort.action_id)
Beispiel #4
0
    def test_choose_sort_max_10(self):
        witch_state = Witch((3, 0, 0, 0), 0)
        sorts = [
            CastAction((-1, 10, 0, 0), 0, 1, 0, 0, True, True),
            CastAction((-1, 2, 0, 0), 0, 2, 0, 0, True, True),
        ]
        potions = [
            BrewAction((1, 1, 1, 1), 0, 5, 0, 0, True, True),
        ]
        sort = witch_state.find_best_cast(sorts, potions)
        self.assertEqual(2, sort.action_id)

        sort = witch_state.find_best_cast(sorts[::-1], potions)
        self.assertEqual(2, sort.action_id)
Beispiel #5
0
def main():
    while True:
        start = time()
        all_actions = read_actions()

        me = Witch.read_input()
        other = Witch.read_input()

        best_action, score, brew_id = me.recursive_find_best_action(
            all_actions, 6)
        if best_action:
            best_action.doit(round(time() - start, 2), brew_id)
        else:
            best_action = me.heuristique_find_best_action(all_actions)
            if best_action is None:
                print("REST", time() - start)
            else:
                best_action.doit(round(time() - start, 2), "heuristic")
Beispiel #6
0
 def test_brew_middle_heuristic(self):
     witch = Witch((3, 2, 2, 1), 0)
     brew_action = witch.heuristique_find_best_action(ALL_ACTIONS)
     self.assertEqual(69, brew_action.action_id)