def test_straight_flush_hands(self): state = GameState() card_piles = [[[Card("K", "d")], [Card("Q", "d")], [Card("A", "s")]], [[Card("2", "c")], [Card("A", "d")], [Card("J", "d")]], [[Card("3", "s")], [Card("T", "d")], [Card("9", "d")]]] state.start_new_game(lucky_card=Card("7", "h"), card_piles=card_piles) straight_flush_hands = state._get_straight_flush_hands() assert len(straight_flush_hands) == 2 assert set([(0, 0), (0, 1), (1, 1), (1, 2), (2, 1)]) in straight_flush_hands assert set([(0, 0), (0, 1), (1, 2), (2, 1), (2, 2)]) in straight_flush_hands # The hand must all be the same suit assert set([(0, 0), (0, 1), (0, 2), (1, 2), (2, 1)]) not in straight_flush_hands