From c3e46017ebacc2a6f2f482789b212cd553930319 Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Thu, 30 Jan 2025 10:28:41 +0100 Subject: [PATCH] now the legal moves function actually returns a list of moves --- src/logic/pieces/bishop.py | 4 +++- src/logic/pieces/pawn.py | 4 ++-- src/logic/pieces/piece.py | 3 ++- 3 files changed, 7 insertions(+), 4 deletions(-) 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")