def test_add_footprint(self):
team = 'team'
footprint = FootPrint(Token.ATTACKING, Direction.N, team)
tile = Tile()
self.assertEqual(tile.footprint_vectors(), {})
tile.add_footprint(footprint)
self.assertEqual(tile.footprint_vectors(),
{'team': footprint.vectorize()})
def test_add_footprint_multiple_footprints_single_team(self):
team = 'team'
footprint = FootPrint(Token.ATTACKING, Direction.N, team)
footprint2 = FootPrint(Token.DANGER, Direction.E, team)
tile = Tile()
self.assertEqual(tile.footprint_vectors(), {})
tile.add_footprint(footprint)
tile.add_footprint(footprint2)
self.assertEqual(
tile.footprint_vectors(),
{'team': footprint.vectorize().add(footprint2.vectorize())})
def test_add_footprint_multiple_footprints_multiple_teams(self):
team_1 = 'team_1'
footprint = FootPrint(Token.ATTACKING, Direction.N, team_1)
footprint2 = FootPrint(Token.DANGER, Direction.E, team_1)
team_2 = 'team_2'
footprint3 = FootPrint(Token.DEAD, Direction.W, team_2)
footprint4 = FootPrint(Token.NEUTRAL, Direction.S, team_2)
tile = Tile()
self.assertEqual(tile.footprint_vectors(), {})
tile.add_footprint(footprint)
tile.add_footprint(footprint4)
tile.add_footprint(footprint3)
tile.add_footprint(footprint2)
expected = {
team_1: footprint.vectorize().add(footprint2.vectorize()),
team_2: footprint3.vectorize().add(footprint4.vectorize())
}
self.assertEqual(tile.footprint_vectors(), expected)
