Add ColorName literal type

niklasf · Aug 2, 2023 · 48bd065 · 48bd065
1 parent 60d38d5
commit 48bd065
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/chess/__init__.py b/chess/__init__.py
@@ -29,7 +29,8 @@
 
 Color = bool
 COLORS = [WHITE, BLACK] = [True, False]
-COLOR_NAMES = ["black", "white"]
+ColorName = Literal["white", "black"]
+COLOR_NAMES: List[ColorName] = ["black", "white"]
 
 PieceType = int
 PIECE_TYPES = [PAWN, KNIGHT, BISHOP, ROOK, QUEEN, KING] = range(1, 7)