Linting

board.py

@@ -1,11 +1,14 @@
 """Contains board class"""
 import enum
-from typing import Union, List, Dict, Optional, NewType
+from typing import Union, List, Dict, Optional, Set, Tuple
+import dataclasses
 from dataclasses import dataclass
+import itertools
 
 
 class SpecialCard(enum.Enum):
     """Different types of special cards"""
+
     Zhong = enum.auto()
     Bai = enum.auto()
     Fa = enum.auto()
@@ -15,11 +18,14 @@ class SpecialCard(enum.Enum):
 @dataclass(frozen=True)
 class NumberCard:
     """Different number cards"""
+
     class Suit(enum.Enum):
         """Different colors number cards can have"""
+
         Red = enum.auto()
         Green = enum.auto()
         Black = enum.auto()
+
     suit: Suit
     number: int
 
@@ -29,19 +35,65 @@ Card = Union[NumberCard, SpecialCard]
 
 class Position(enum.Enum):
     """Possible Board positions"""
+
     Field = enum.auto()
     Bunker = enum.auto()
     Goal = enum.auto()
 
 
-KilledDragon = NewType('KilledDragon', SpecialCard)
-
-@dataclass
 class Board:
     """Solitaire board"""
-    field: List[List[Card]] = [[]] * 8
-    bunker: List[Union[KilledDragon, Optional[Card]]] = [None] * 3
-    goal: Dict[NumberCard.Suit, int] = {NumberCard.Suit.Red: 0,
-                                        NumberCard.Suit.Green: 0,
-                                        NumberCard.Suit.Black: 0}
+
+    def __init__(self) -> None:
+        self.field: List[List[Card]] = [[]] * 8
+        self.bunker: List[Union[Tuple[SpecialCard, int], Optional[Card]]] = [None] * 3
+        self.goal: Dict[NumberCard.Suit, int] = {
+            NumberCard.Suit.Red: 0,
+            NumberCard.Suit.Green: 0,
+            NumberCard.Suit.Black: 0,
+        }
+
     flowerGone: bool = False
+
+    def check_correct(self) -> bool:
+        """Returns true, if the board is in a valid state"""
+        number_cards: Dict[NumberCard.Suit, Set[int]] = {
+            NumberCard.Suit.Red: set(),
+            NumberCard.Suit.Green: set(),
+            NumberCard.Suit.Black: set(),
+        }
+        special_cards: Dict[SpecialCard, int] = {
+            SpecialCard.Zhong: 0,
+            SpecialCard.Bai: 0,
+            SpecialCard.Fa: 0,
+            SpecialCard.Hua: 0,
+        }
+
+        if self.flowerGone:
+            special_cards[SpecialCard.Hua] += 1
+
+        for card in itertools.chain(
+            self.bunker,
+            itertools.chain.from_iterable(stack for stack in self.field if stack),
+        ):
+            if isinstance(card, tuple):
+                special_cards[card[0]] += 4  # pylint: disable=E1136
+            elif isinstance(card, SpecialCard):
+                special_cards[card] += 1
+            elif isinstance(card, NumberCard):
+                if card.number in number_cards[card.suit]:
+                    return False
+                number_cards[card.suit].add(card.number)
+
+        for _, numbers in number_cards.items():
+            if set(range(1, 10)) != numbers:
+                return False
+
+        for cardtype, count in special_cards.items():
+            if cardtype == SpecialCard.Hua:
+                if count != 1:
+                    return False
+            else:
+                if count != 4:
+                    return False
+        return True