diff --git a/src/logic/pieces/bishop.py b/src/logic/pieces/bishop.py index 5171ca6..3e9e386 100644 --- a/src/logic/pieces/bishop.py +++ b/src/logic/pieces/bishop.py @@ -1,4 +1,6 @@ +from logic.move import Move from .piece import Piece class Bishop(Piece): - pass + def legal_moves(self, board) -> list[Move]: + return super().legal_moves(board) diff --git a/src/logic/pieces/pawn.py b/src/logic/pieces/pawn.py index 76e7171..8de40b7 100644 --- a/src/logic/pieces/pawn.py +++ b/src/logic/pieces/pawn.py @@ -1,8 +1,8 @@ -from logic.position import Position +from logic.move import Move from logic.pieces.piece import Piece class Pawn(Piece): - def legal_moves(self, board) -> list[Position]: + def legal_moves(self, board) -> list[Move]: ret = [] # can we capture to the left? diff --git a/src/logic/pieces/piece.py b/src/logic/pieces/piece.py index c4a7afa..9698660 100644 --- a/src/logic/pieces/piece.py +++ b/src/logic/pieces/piece.py @@ -1,4 +1,5 @@ from logic.position import Position +from logic.move import Move class Piece: @@ -13,5 +14,5 @@ class Piece: def position(self) -> Position: return self.pos - def legal_moves(self, board) -> list[Position]: + def legal_moves(self, board) -> list[Move]: raise NotImplementedError(f"Can't say what the legal moves are for {type(self).__name__}, the method hasn't been implemented yet")