def test_attacked_locations_black_none(self):
pawn = Pawn(Location(5, 'd'), Player.BLACK, self.board)
results = set()
self.assertEquals(set(pawn.attacked_locations()), results)
def test_attacked_locations_white(self):
pawn = Pawn(Location(5, 'd'), Player.WHITE, self.board)
queen = Queen(Location(6, 'c'), Player.BLACK, self.board)
knight = Knight(Location(6, 'e'), Player.BLACK, self.board)
results = {Location(6, 'c'), Location(6, 'e')}
self.assertEquals(set(pawn.attacked_locations()), results)
