def test_full_house(self):
poker = Poker()
poker.new_game(['S2', 'H2', 'S5', 'H3', 'C3'])
assert poker.is_full_house() == False
def test_is_straight(self):
poker = Poker()
poker.new_game(['S6', 'D2', 'H3', 'C4', 'C5'])
assert poker.is_straight() == True
poker.new_game(['SA', 'D2', 'H3', 'C4', 'C5'])
assert poker.is_straight() == True
poker.new_game(['SA', 'DJ', 'HQ', 'CK', 'CT'])
assert poker.is_straight() == True
poker.new_game(['SA', 'DJ', 'HQ', 'CK', 'C9'])
assert poker.is_straight() == False
poker.new_game(('S2', 'D2', 'H3', 'C3', 'C5'))
assert poker.is_straight() == False
def test_flush(self):
poker = Poker()
poker.new_game(['S6', 'S2', 'S3', 'S4', 'S5'])
assert poker.is_flush() == True
poker.new_game(['S6', 'S2', 'S3', 'S4', 'H5'])
assert poker.is_flush() == False
def test_4_kind(self):
poker = Poker()
poker.new_game(['S5', 'D5', 'H5', 'C5', 'C3'])
assert poker.kind(4) == 5
poker.new_game(['S5', 'D5', 'CK', 'C5', 'C3'])
assert poker.kind(4) is None
def test_4_kind_suits(self):
poker = Poker()
poker.new_game(['S5', 'S9', 'SA', 'S2', 'C3'])
assert poker.kind(4, 'suits') == 'S'
def test_index_all(self):
poker = Poker()
poker.new_game(('S2', 'D2', 'H3', 'C3', 'C5'))
assert poker.index_all(2) == [0, 1]
poker.new_game(('S3', 'D3', 'H3', 'C3', 'C5'))
assert poker.index_all(3) == [0, 1, 2, 3]
poker.new_game(('SJ', 'D7', 'S8', 'JOKER', 'HJ'))
assert poker.index_all('J') == [0, 4]
def test_3_kind(self):
poker = Poker()
poker.new_game(["S5", "D5", "H5", "C2", "C3"])
assert poker.kind(3) == 5
poker.new_game(["SA", "D5", "H5", "C2", "C3"])
assert poker.kind(3) is None
def test_is_straight_flush(self):
poker = Poker()
poker.new_game(["SA", "S2", "S5", "S3", "S4"])
assert poker.is_straight_flush() == True
poker.new_game(["SA", "S2", "S5", "S3", "H4"])
assert poker.is_straight_flush() == False
def test_play(self):
# two pair
poker = Poker()
poker.new_game(("S2", "D2", "H3", "C3", "C5"))
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 3]
# 3 kind
poker.new_game(("S2", "D2", "H2", "D4", "C5"))
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2]
poker.new_game(("S2", "D2", "JOKER", "D4", "C5"))
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2]
poker.new_game(("SJ", "D7", "S8", "JOKER", "HJ"))
poker.calculate()
assert poker.hold_cards_index == [0, 3, 4]
# straight
poker.new_game(("S2", "D3", "H6", "D4", "C5"))
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(("S2", "D3", "HA", "D4", "C5"))
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(("ST", "DQ", "HA", "DK", "CJ"))
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(["S6", "D2", "JOKER", "C4", "C5"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(["SA", "H3", "JOKER", "C4", "C5"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(["SA", "DJ", "JOKER", "CK", "CT"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# flush
poker.new_game(["S6", "S2", "S8", "S4", "S5"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(["S6", "S2", "JOKER", "S4", "S5"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# full house
poker.new_game(["S2", "H2", "JOKER", "S3", "H3"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(["D3", "H2", "JOKER", "S3", "D2"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(["S2", "H2", "S2", "H3", "C3"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# 4 kind
poker.new_game(["S5", "D5", "H5", "JOKER", "C3"])
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 3]
poker.new_game(["S2", "H2", "JOKER", "S3", "D2"])
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 4]
# straight flush
poker.new_game(["SA", "S2", "JOKER", "S3", "S4"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# 5 kind
poker.new_game(["S5", "D5", "H5", "C5", "JOKER"])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# joker
poker.new_game(["S5", "D6", "HA", "C8", "JOKER"])
poker.calculate()
assert poker.hold_cards_index == [4]
# one pair
poker.new_game(["S5", "D6", "HA", "C8", "D5"])
poker.calculate()
assert poker.hold_cards_index == [0, 4]
poker.new_game(("H5", "D6", "C7", "C5", "S3"))
poker.calculate()
assert poker.hold_cards_index == [0, 3]
# 4 kind suits
poker.new_game(["S5", "S6", "SA", "S8", "D9"])
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 3]
# no pair
poker.new_game(["S5", "D6", "HA", "C8", "CK"])
poker.calculate()
n = len(poker.hold_cards_index)
assert n == 1 or n == 0
def test_flush(self):
poker = Poker()
poker.new_game(["S6", "S2", "S3", "S4", "S5"])
assert poker.is_flush() == True
poker.new_game(["S6", "S2", "S3", "S4", "H5"])
assert poker.is_flush() == False
def test_full_house(self):
poker = Poker()
poker.new_game(["S2", "H2", "S5", "H3", "C3"])
assert poker.is_full_house() == False
def test_is_straight(self):
poker = Poker()
poker.new_game(["S6", "D2", "H3", "C4", "C5"])
assert poker.is_straight() == True
poker.new_game(["SA", "D2", "H3", "C4", "C5"])
assert poker.is_straight() == True
poker.new_game(["SA", "DJ", "HQ", "CK", "CT"])
assert poker.is_straight() == True
poker.new_game(["SA", "DJ", "HQ", "CK", "C9"])
assert poker.is_straight() == False
poker.new_game(("S2", "D2", "H3", "C3", "C5"))
assert poker.is_straight() == False
def test_4_kind_suits(self):
poker = Poker()
poker.new_game(["S5", "S9", "SA", "S2", "C3"])
assert poker.kind(4, "suits") == "S"
def test_4_kind(self):
poker = Poker()
poker.new_game(["S5", "D5", "H5", "C5", "C3"])
assert poker.kind(4) == 5
poker.new_game(["S5", "D5", "CK", "C5", "C3"])
assert poker.kind(4) is None
def test_is_straight_flush(self):
poker = Poker()
poker.new_game(['SA', 'S2', 'S5', 'S3', 'S4'])
assert poker.is_straight_flush() == True
poker.new_game(['SA', 'S2', 'S5', 'S3', 'H4'])
assert poker.is_straight_flush() == False
def test_index_all(self):
poker = Poker()
poker.new_game(("S2", "D2", "H3", "C3", "C5"))
assert poker.index_all(2) == [0, 1]
poker.new_game(("S3", "D3", "H3", "C3", "C5"))
assert poker.index_all(3) == [0, 1, 2, 3]
poker.new_game(("SJ", "D7", "S8", "JOKER", "HJ"))
assert poker.index_all("J") == [0, 4]
def test_play(self):
# two pair
poker = Poker()
poker.new_game(('S2', 'D2', 'H3', 'C3', 'C5'))
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 3]
# 3 kind
poker.new_game(('S2', 'D2', 'H2', 'D4', 'C5'))
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2]
poker.new_game(('S2', 'D2', 'JOKER', 'D4', 'C5'))
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2]
poker.new_game(('SJ', 'D7', 'S8', 'JOKER', 'HJ'))
poker.calculate()
assert poker.hold_cards_index == [0, 3, 4]
# straight
poker.new_game(('S2', 'D3', 'H6', 'D4', 'C5'))
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(('S2', 'D3', 'HA', 'D4', 'C5'))
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(('ST', 'DQ', 'HA', 'DK', 'CJ'))
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(['S6', 'D2', 'JOKER', 'C4', 'C5'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(['SA', 'H3', 'JOKER', 'C4', 'C5'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(['SA', 'DJ', 'JOKER', 'CK', 'CT'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# flush
poker.new_game(['S2', 'H2', 'S3', 'H3', 'C3'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(['S6', 'S2', 'JOKER', 'S4', 'S5'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# full house
poker.new_game(['S2', 'H2', 'JOKER', 'S3', 'H3'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
poker.new_game(['D3', 'H2', 'JOKER', 'S3', 'D2'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# 4 kind
poker.new_game(['S5', 'D5', 'H5', 'JOKER', 'C3'])
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 3]
poker.new_game(['S2', 'H2', 'JOKER', 'S3', 'D2'])
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 4]
# straight flush
poker.new_game(['SA', 'S2', 'JOKER', 'S3', 'S4'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# 5 kind
poker.new_game(['S5', 'D5', 'H5', 'C5', 'JOKER'])
poker.calculate()
assert poker.hold_cards_index == list(range(5))
# joker
poker.new_game(['S5', 'D6', 'HA', 'C8', 'JOKER'])
poker.calculate()
assert poker.hold_cards_index == [4]
# one pair
poker.new_game(['S5', 'D6', 'HA', 'C8', 'D5'])
poker.calculate()
assert poker.hold_cards_index == [0, 4]
poker.new_game(('H5', 'D6', 'C7', 'C5', 'S3'))
poker.calculate()
assert poker.hold_cards_index == [0, 3]
# 4 kind suits
poker.new_game(['S5', 'S6', 'SA', 'S8', 'D9'])
poker.calculate()
assert poker.hold_cards_index == [0, 1, 2, 3]
# no pair
poker.new_game(['S5', 'D6', 'HA', 'C8', 'CK'])
poker.calculate()
n = len(poker.hold_cards_index)
assert n == 1 or n == 0
def test_3_kind(self):
poker = Poker()
poker.new_game(['S5', 'D5', 'H5', 'C2', 'C3'])
assert poker.kind(3) == 5
poker.new_game(['SA', 'D5', 'H5', 'C2', 'C3'])
assert poker.kind(3) is None
def test_two_pair(self):
poker = Poker()
poker.new_game(('S2', 'D2', 'H3', 'C3', 'C5'))
assert poker.two_pair() == (2, 3)
poker.new_game(('SA', 'D2', 'H3', 'C3', 'C5'))
assert poker.two_pair() is None
def test_two_pair(self):
poker = Poker()
poker.new_game(("S2", "D2", "H3", "C3", "C5"))
assert poker.two_pair() == (2, 3)
poker.new_game(("SA", "D2", "H3", "C3", "C5"))
assert poker.two_pair() is None