test_funcs.py

from PokerPy import Card, Hand, calculate_hand_frequency, get_best_hand

def test_calculate_frec():
    test_cards = [[Card(12, 2),Card(8, 1)],[Card(12, 3),Card(13, 3)]]
    frecs = calculate_hand_frequency(test_cards)
    test_frecs = [
        {'Double Pairs': 355878, 'Draw': 20234, 'Flush': 38643, 'Full House': 28846, 'High Card': 358523, 'Pairs': 784240, 'Poker': 1430, 'Royal Flush': 46, 'Straight': 77914, 'Straight Flush': 284, 'Total Cases': 1712305, 'Triples': 66500, 'Win': 380882},
        {'Double Pairs': 347144, 'Draw': 20234, 'Flush': 124371, 'Full House': 28846, 'High Card': 337972, 'Pairs': 747226, 'Poker': 1430, 'Royal Flush': 992, 'Straight': 59303, 'Straight Flush': 70, 'Total Cases': 1712305, 'Triples': 64950, 'Win': 1311188}
    ]
    assert frecs == test_frecs

def test_best_hand():
    test_cards = [Card(12, 2),Card(9, 1),Card(12, 3),Card(13, 3),Card(3, 3),Card(2, 2),Card(1, 2)]
    best_hand = get_best_hand(test_cards)
    test_hand = Hand(2, [Card(12, 2),Card(12, 3),Card(13, 3),Card(9, 1),Card(3, 3)])
    assert test_hand.hand_type == best_hand.hand_type
    assert test_hand.Cards == best_hand.Cards