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)