From c900ebcfa01f94420dff9952e921841f1f23451c Mon Sep 17 00:00:00 2001 From: Karma Riuk Date: Thu, 30 Jan 2025 10:45:12 +0100 Subject: [PATCH] extracted the colour to an enum --- src/logic/pieces/piece.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/logic/pieces/piece.py b/src/logic/pieces/piece.py index 9698660..cd44a3b 100644 --- a/src/logic/pieces/piece.py +++ b/src/logic/pieces/piece.py @@ -1,14 +1,17 @@ from logic.position import Position from logic.move import Move +from enum import Enum -class Piece: +class Colour(Enum): WHITE = "white" BLACK = "black" - def __init__(self, pos, colour) -> None: +class Piece: + + def __init__(self, pos: Position, colour: Colour) -> None: self.pos = pos - assert colour == self.WHITE or colour == self.BLACK, "The colour of the piece must be either Piece.WHITE or Piece.BLACK" + assert colour == Colour.WHITE or colour == Colour.BLACK, "The colour of the piece must be either Piece.WHITE or Piece.BLACK" self.colour = colour def position(self) -> Position: