PyPoker/tests.py at main · DELIRIOUSDONUT/PyPoker · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
from unittest import TestCase
from scoring import Scorer

#imports from scoring
import pyCardDeck
# noinspection PyCompatibility
from typing import List
from pyCardDeck.cards import PokerCard
from collections import OrderedDict


class TestScoring(TestCase):
    def testHighCard(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Hearts", "10", "Ten")]
        field = [
            PokerCard("Spades", "9", "Nine"), PokerCard("Spades", "K", "King"),
            PokerCard("Spades", "A", "Ace"), PokerCard("Spades", "2", "Two"),
            PokerCard("Spades", "10", "Ten")
        ]
        scorer = Scorer(hand, field)
        result = scorer.high_card_search()
        self.assertEqual(result['cards'], PokerCard("Spades", "A", "Ace"))

    def testOnePair(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Diamonds", "9", "Nine")]
        field = [
            PokerCard("Spades", "K", "King"), PokerCard("Clubs", "A", "Ace"),
            PokerCard("Hearts", "2", "Two"), PokerCard("Diamonds", "3", "Three"),
            PokerCard("Clubs", "4", "Four")
        ]
        scorer = Scorer(hand, field)
        result = scorer.one_pair()
        self.assertEqual(result['name'], "Pair")

    def testTwoPair(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Diamonds", "9", "Nine")]
        field = [
            PokerCard("Spades", "K", "King"), PokerCard("Clubs", "K", "King"),
            PokerCard("Hearts", "2", "Two"), PokerCard("Diamonds", "3", "Three"),
            PokerCard("Clubs", "4", "Four")
        ]
        scorer = Scorer(hand, field)
        result = scorer.two_pair()
        self.assertEqual(result['name'], "Two Pair")

    def testThreeOfAKind(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Diamonds", "9", "Nine")]
        field = [
            PokerCard("Spades", "9", "Nine"), PokerCard("Clubs", "K", "King"),
            PokerCard("Hearts", "2", "Two"), PokerCard("Diamonds", "3", "Three"),
            PokerCard("Clubs", "4", "Four")
        ]
        scorer = Scorer(hand, field)
        result = scorer.three_of_a_kind()
        self.assertEqual(result['name'], "Three of a kind")

    def testFourOfAKind(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Diamonds", "9", "Nine")]
        field = [
            PokerCard("Spades", "9", "Nine"), PokerCard("Clubs", "9", "Nine"),
            PokerCard("Hearts", "2", "Two"), PokerCard("Diamonds", "3", "Three"),
            PokerCard("Clubs", "4", "Four")
        ]
        scorer = Scorer(hand, field)
        result = scorer.four_of_a_kind()
        self.assertEqual(result['name'], "Four of a kind")

    def testFullHouse(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Diamonds", "9", "Nine")]
        field = [
            PokerCard("Spades", "K", "King"), PokerCard("Clubs", "K", "King"),
            PokerCard("Diamonds", "K", "King"), PokerCard("Hearts", "2", "Two"),
            PokerCard("Diamonds", "3", "Three")
        ]
        scorer = Scorer(hand, field)
        result = scorer.full_house()
        self.assertEqual(result['name'], "Full House")

    def testFlush(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Hearts", "K", "King")]
        field = [
            PokerCard("Hearts", "A", "Ace"), PokerCard("Hearts", "2", "Two"),
            PokerCard("Hearts", "3", "Three"), PokerCard("Diamonds", "4", "Four"),
            PokerCard("Clubs", "5", "Five")
        ]
        scorer = Scorer(hand, field)
        result = scorer.flush()
        self.assertEqual(result['name'], "Flush")

    def testStraight(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Diamonds", "10", "Ten")]
        field = [
            PokerCard("Spades", "J", "Jack"), PokerCard("Clubs", "Q", "Queen"),
            PokerCard("Diamonds", "K", "King"), PokerCard("Hearts", "2", "Two"),
            PokerCard("Diamonds", "3", "Three")
        ]
        scorer = Scorer(hand, field)
        result = scorer.straight()
        self.assertEqual(result['name'], "Straight")

    def testStraightFlush(self):
        hand = [PokerCard("Hearts", "9", "Nine"), PokerCard("Hearts", "10", "Ten")]
        field = [
            PokerCard("Hearts", "J", "Jack"), PokerCard("Hearts", "Q", "Queen"),
            PokerCard("Hearts", "K", "King"), PokerCard("Diamonds", "2", "Two"),
            PokerCard("Clubs", "3", "Three")
        ]
        scorer = Scorer(hand, field)
        result = scorer.straight_flush()
        self.assertEqual(result['name'], "Straight Flush")

    def testRoyalFlush(self):
        hand = [PokerCard("Hearts", "10", "Ten"), PokerCard("Hearts", "J", "Jack")]
        field = [
            PokerCard("Hearts", "Q", "Queen"), PokerCard("Hearts", "K", "King"),
            PokerCard("Hearts", "A", "Ace"), PokerCard("Diamonds", "2", "Two"),
            PokerCard("Clubs", "3", "Three")
        ]
        scorer = Scorer(hand, field)
        result = scorer.royal_flush()
        self.assertEqual(result['name'], "Royal Flush")