add-poker-kata-with-one-started-branch

Author: hnomichith

src/poker/__init__.py

@@ -0,0 +1,4 @@
+# Texas Holdem
+
+https://codingdojo.org/kata/TexasHoldEm/
+

src/poker/readme.md

@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from enum import Enum, auto
+
+class Suit(Enum):
+    SPADE = "s"
+    HEART = "h"
+    DIAMOND = "d"
+    CLUB = "c"
+
+class Rank(Enum):
+    ONE = "1"
+    TWO = "2"
+    THREE = "3"
+    FOUR = "4"
+    FIVE = "5"
+    SIX = "6"
+    SEVEN = "7"
+    EIGHT = "8"
+    NINE = "9"
+    TEN = "T"
+    JACK = "J"
+    QUEEN = "Q"
+    KING = "K"
+    ACE = "A"
+
+class HandRankType(Enum):
+    HIGH_CARD = auto()
+    PAIR = auto()
+
+RANK_ORDER = tuple(r for r in Rank)
+
+
+@dataclass
+class Card:
+    rank: Rank
+    suit: Suit
+    
+    @classmethod
+    def from_str(cls, card_string: str) -> Card:
+        rank, suit = card_string
+        return Card(rank=Rank(rank), suit=Suit(suit))
+
+    def __gt__(self, other):
+        return RANK_ORDER.index(self.rank) > RANK_ORDER.index(other.rank)
+
+@dataclass
+class HandRank:
+    type: HandRankType
+    values: list[Rank]
+
+
+def read_input(input_str: str) -> list[list[Card]]:
+    hands = input_str.split('\n')
+    return [
+        [Card.from_str(card) for card in hand.strip().split(' ')] for hand in hands
+    ]
+
+
+def rank_hand(hand: list[Card]) -> HandRank:
+    values = [card.rank for card in reversed(sorted(hand))]
+    if (values[0] == values[1]):
+        return HandRank(
+            type=HandRankType.PAIR,
+            values=values,
+        )
+    return HandRank(
+        type=HandRankType.HIGH_CARD,
+        values=values,
+    )

src/poker/with_read_input/__init__.py

@@ -0,0 +1,32 @@
+from poker.with_read_input.poker import *
+import pytest
+
+king_of_spade = Card(rank=Rank.KING, suit=Suit.SPADE)
+nine_of_diamond = Card(rank=Rank.NINE, suit=Suit.DIAMOND)
+nine_of_spade = Card(rank=Rank.NINE, suit=Suit.SPADE)
+ace_of_club = Card(rank=Rank.ACE, suit=Suit.CLUB)
+
+@pytest.mark.parametrize(
+    "input,expected",
+    [
+        ('Ks', [[king_of_spade]]),
+        ('Ks\n9d', [[king_of_spade], [nine_of_diamond]]),
+        ('Ks 9d\nAc 9d', [[king_of_spade,nine_of_diamond], [ace_of_club,nine_of_diamond]]),
+        ('  Ks  ', [[king_of_spade]]),
+    ]
+)
+def test_read_input(input: str, expected: list[list[Card]]) -> None:
+    result = read_input(input)
+    assert result == expected
+
+@pytest.mark.parametrize(
+    "hand,expected",
+    [
+        ([king_of_spade, nine_of_diamond], HandRank(type=HandRankType.HIGH_CARD, values=[Rank.KING, Rank.NINE])),
+        ([nine_of_diamond, king_of_spade], HandRank(type=HandRankType.HIGH_CARD, values=[Rank.KING, Rank.NINE])),
+        ([nine_of_diamond, nine_of_spade], HandRank(type=HandRankType.PAIR, values=[Rank.NINE, Rank.NINE])),
+    ]
+)
+def test_rank_hand(hand: list[Card], expected: HandRank):
+    result = rank_hand(hand)
+    assert result == expected