|
4 | 4 | import networkx
|
5 | 5 |
|
6 | 6 | from pelita.layout import parse_layout, get_random_layout, initial_positions
|
7 |
| -from pelita.game import run_game, setup_game, play_turn |
| 7 | +from pelita.game import run_game, setup_game, play_turn, SHADOW_DISTANCE |
| 8 | +from pelita.gamestate_filters import manhattan_dist |
8 | 9 |
|
9 | 10 | def stopping(bot, state):
|
10 | 11 | return bot.position
|
@@ -395,12 +396,14 @@ def asserting_team(bot, state):
|
395 | 396 |
|
396 | 397 | assert bot.walls == tuple(sorted(bot.walls))
|
397 | 398 | assert bot.homezone == tuple(sorted(bot.homezone))
|
398 |
| - |
399 | 399 | if bot.is_blue:
|
400 | 400 | assert set(bot.homezone) == set(homezones[0])
|
401 | 401 | assert set(bot.enemy[0].homezone) == set(homezones[1])
|
402 |
| - assert set(bot.shaded_food) == set([(1, 1), (3, 1), (4, 1), (5, 1)]) |
403 |
| - assert set(bot.other.shaded_food) == set([(1, 1), (3, 1), (4, 1), (5, 1)]) |
| 402 | + shaded_food = set(food for food in bot.food if |
| 403 | + manhattan_dist(bot.position, food) <= SHADOW_DISTANCE or |
| 404 | + manhattan_dist(bot.other.position, food) <= SHADOW_DISTANCE) |
| 405 | + assert set(bot.shaded_food) == shaded_food |
| 406 | + assert set(bot.other.shaded_food) == shaded_food |
404 | 407 | assert set(bot.enemy[0].shaded_food) == set()
|
405 | 408 | assert set(bot.enemy[1].shaded_food) == set()
|
406 | 409 | else:
|
|
0 commit comments