class TestGame(TestCase):
def setUp(self):
self.btree = BehaviourTree(YAML_MODEL_PATH)
self.btree.load()
def test_can_eat_no_retry_needed(self):
character_data = {
"coordinates": (2, 2),
"food": ["apple", "orange"],
}
self.btree.execute(character_data)
self.assertListEqual(
self.btree.execution_path,
[
("am_i_hungry", True),
("have_i_food", True),
("NOT", ("enemies_nearby", False), True),
("eat", True)
]
)
@patch("models.game.tasks.DB.query", side_effect=[["Bowser", "Revolver Ocelot"], ["Marlene"], []])
def test_can_eat_on_third_retry(self, mock_db_query):
"""Mock the db call to check for enemies around to pass on the 3rd attempt."""
character_data = {
"coordinates": (2, 2),
"food": ["apple", "orange"],
}
self.btree.execute(character_data)
self.assertListEqual(
self.btree.execution_path,
[
("am_i_hungry", True),
("have_i_food", True),
("NOT", ("enemies_nearby", True), False),
("NOT", ("enemies_nearby", True), False),
("NOT", ("enemies_nearby", False), True),
("eat", True)
]
)
class TestFootball(TestCase):
def setUp(self):
self.btree = BehaviourTree(JSON_MODEL_PATH)
self.btree.load()
def test_attacker_can_shoot(self):
game_data = {
"attacker": {
"name": "Mane",
"coordinates": (0, 1)
},
"opposition": [{
"name": "Jones",
"coordinates": (0, 2)
}],
"teammates": []
}
self.btree.execute(game_data)
self.assertListEqual(self.btree.execution_path,
[("check_have_space", True),
("check_close_to_goal", True), ("shoot", True)])
def test_attacker_cannot_shoot_but_can_pass(self):
game_data = {
"attacker": {
"name": "Mane",
"coordinates": (1, 0) # too far out to shoot
},
"opposition": [{
"name": "Jones",
"coordinates": (0, 1)
}],
"teammates": [{
"name": "Robertson",
"coordinates": (1, 0)
}]
}
self.btree.execute(game_data)
self.assertListEqual(self.btree.execution_path,
[("check_have_space", True),
("check_close_to_goal", False),
("check_have_space", True),
("check_teammate_nearby", True),
("NOT",
("check_nearby_teammates_marked", False), True),
("pass_ball", True)])
def test_attacker_cannot_shoot_cannot_pass_but_can_cross(self):
game_data = {
"attacker": {
"name": "Mane",
"coordinates": (0, 0) # too wide to shoot
},
"opposition": [{
"name": "Jones",
"coordinates": (0, 1)
}],
"teammates": [{
"name": "Firmino",
"coordinates": (0, 1) # too far away to pass
}]
}
self.btree.execute(game_data)
self.assertListEqual(self.btree.execution_path,
[("check_have_space", True),
("check_close_to_goal", False),
("check_have_space", True),
("check_teammate_nearby", False),
("check_have_space", True),
("check_in_crossing_position", True),
("cross", True)])
def test_attacker_can_only_go_backwards(self):
game_data = {
"attacker": {
"name": "Mane",
"coordinates": (2, 2) # too far out to shoot, cross
},
"opposition": [{
"name": "Jones",
"coordinates": (1, 1)
}],
"teammates": [{
"name": "Firmino",
"coordinates": (0, 1) # too far away to pass
}]
}
self.btree.execute(game_data)
self.assertListEqual(self.btree.execution_path,
[("check_have_space", True),
("check_close_to_goal", False),
("check_have_space", True),
("check_teammate_nearby", False),
("check_have_space", True),
("check_in_crossing_position", False),
("go_backwards", True)])