

'''simple docstring'''


import os
from itertools import chain
from random import randrange, shuffle

import pytest

from .sola import PokerHand

lowerCAmelCase__						:       List[str]										=						(
    """4S 3H 2C 7S 5H""",
    """9D 8H 2C 6S 7H""",
    """2D 6D 9D TH 7D""",
    """TC 8C 2S JH 6C""",
    """JH 8S TH AH QH""",
    """TS KS 5S 9S AC""",
    """KD 6S 9D TH AD""",
    """KS 8D 4D 9S 4S""",  # pair
    """8C 4S KH JS 4D""",  # pair
    """QH 8H KD JH 8S""",  # pair
    """KC 4H KS 2H 8D""",  # pair
    """KD 4S KC 3H 8S""",  # pair
    """AH 8S AS KC JH""",  # pair
    """3H 4C 4H 3S 2H""",  # 2 pairs
    """5S 5D 2C KH KH""",  # 2 pairs
    """3C KH 5D 5S KH""",  # 2 pairs
    """AS 3C KH AD KH""",  # 2 pairs
    """7C 7S 3S 7H 5S""",  # 3 of a kind
    """7C 7S KH 2H 7H""",  # 3 of a kind
    """AC KH QH AH AS""",  # 3 of a kind
    """2H 4D 3C AS 5S""",  # straight (low ace)
    """3C 5C 4C 2C 6H""",  # straight
    """6S 8S 7S 5H 9H""",  # straight
    """JS QS 9H TS KH""",  # straight
    """QC KH TS JS AH""",  # straight (high ace)
    """8C 9C 5C 3C TC""",  # flush
    """3S 8S 9S 5S KS""",  # flush
    """4C 5C 9C 8C KC""",  # flush
    """JH 8H AH KH QH""",  # flush
    """3D 2H 3H 2C 2D""",  # full house
    """2H 2C 3S 3H 3D""",  # full house
    """KH KC 3S 3H 3D""",  # full house
    """JC 6H JS JD JH""",  # 4 of a kind
    """JC 7H JS JD JH""",  # 4 of a kind
    """JC KH JS JD JH""",  # 4 of a kind
    """2S AS 4S 5S 3S""",  # straight flush (low ace)
    """2D 6D 3D 4D 5D""",  # straight flush
    """5C 6C 3C 7C 4C""",  # straight flush
    """JH 9H TH KH QH""",  # straight flush
    """JH AH TH KH QH""",  # royal flush (high ace straight flush)
)

lowerCAmelCase__						:       Any										=						(
    ("""2H 3H 4H 5H 6H""", """KS AS TS QS JS""", """Loss"""),
    ("""2H 3H 4H 5H 6H""", """AS AD AC AH JD""", """Win"""),
    ("""AS AH 2H AD AC""", """JS JD JC JH 3D""", """Win"""),
    ("""2S AH 2H AS AC""", """JS JD JC JH AD""", """Loss"""),
    ("""2S AH 2H AS AC""", """2H 3H 5H 6H 7H""", """Win"""),
    ("""AS 3S 4S 8S 2S""", """2H 3H 5H 6H 7H""", """Win"""),
    ("""2H 3H 5H 6H 7H""", """2S 3H 4H 5S 6C""", """Win"""),
    ("""2S 3H 4H 5S 6C""", """3D 4C 5H 6H 2S""", """Tie"""),
    ("""2S 3H 4H 5S 6C""", """AH AC 5H 6H AS""", """Win"""),
    ("""2S 2H 4H 5S 4C""", """AH AC 5H 6H AS""", """Loss"""),
    ("""2S 2H 4H 5S 4C""", """AH AC 5H 6H 7S""", """Win"""),
    ("""6S AD 7H 4S AS""", """AH AC 5H 6H 7S""", """Loss"""),
    ("""2S AH 4H 5S KC""", """AH AC 5H 6H 7S""", """Loss"""),
    ("""2S 3H 6H 7S 9C""", """7H 3C TH 6H 9S""", """Loss"""),
    ("""4S 5H 6H TS AC""", """3S 5H 6H TS AC""", """Win"""),
    ("""2S AH 4H 5S 6C""", """AD 4C 5H 6H 2C""", """Tie"""),
    ("""AS AH 3H AD AC""", """AS AH 2H AD AC""", """Win"""),
    ("""AH AC 5H 5C QS""", """AH AC 5H 5C KS""", """Loss"""),
    ("""AH AC 5H 5C QS""", """KH KC 5H 5C QS""", """Win"""),
    ("""7C 7S KH 2H 7H""", """3C 3S AH 2H 3H""", """Win"""),
    ("""3C 3S AH 2H 3H""", """7C 7S KH 2H 7H""", """Loss"""),
    ("""6H 5H 4H 3H 2H""", """5H 4H 3H 2H AH""", """Win"""),
    ("""5H 4H 3H 2H AH""", """5H 4H 3H 2H AH""", """Tie"""),
    ("""5H 4H 3H 2H AH""", """6H 5H 4H 3H 2H""", """Loss"""),
    ("""AH AD KS KC AC""", """AH KD KH AC KC""", """Win"""),
    ("""2H 4D 3C AS 5S""", """2H 4D 3C 6S 5S""", """Loss"""),
    ("""2H 3S 3C 3H 2S""", """3S 3C 2S 2H 2D""", """Win"""),
    ("""4D 6D 5D 2D JH""", """3S 8S 3H TC KH""", """Loss"""),
    ("""4S 6C 8S 3S 7S""", """AD KS 2D 7D 7C""", """Loss"""),
    ("""6S 4C 7H 8C 3H""", """5H JC AH 9D 9C""", """Loss"""),
    ("""9D 9H JH TC QH""", """3C 2S JS 5C 7H""", """Win"""),
    ("""2H TC 8S AD 9S""", """4H TS 7H 2C 5C""", """Win"""),
    ("""9D 3S 2C 7S 7C""", """JC TD 3C TC 9H""", """Loss"""),
)

lowerCAmelCase__						:       Union[str, Any]										=						(
    ("""2H 3H 4H 5H 6H""", True),
    ("""AS AH 2H AD AC""", False),
    ("""2H 3H 5H 6H 7H""", True),
    ("""KS AS TS QS JS""", True),
    ("""8H 9H QS JS TH""", False),
    ("""AS 3S 4S 8S 2S""", True),
)

lowerCAmelCase__						:       Optional[int]										=						(
    ("""2H 3H 4H 5H 6H""", True),
    ("""AS AH 2H AD AC""", False),
    ("""2H 3H 5H 6H 7H""", False),
    ("""KS AS TS QS JS""", True),
    ("""8H 9H QS JS TH""", True),
)

lowerCAmelCase__						:       Dict										=						(
    ("""2H 4D 3C AS 5S""", True, [5, 4, 3, 2, 14]),
    ("""2H 5D 3C AS 5S""", False, [14, 5, 5, 3, 2]),
    ("""JH QD KC AS TS""", False, [14, 13, 12, 11, 10]),
    ("""9D 3S 2C 7S 7C""", False, [9, 7, 7, 3, 2]),
)

lowerCAmelCase__						:       str										=						(
    ("""JH AH TH KH QH""", 0),
    ("""JH 9H TH KH QH""", 0),
    ("""JC KH JS JD JH""", 7),
    ("""KH KC 3S 3H 3D""", 6),
    ("""8C 9C 5C 3C TC""", 0),
    ("""JS QS 9H TS KH""", 0),
    ("""7C 7S KH 2H 7H""", 3),
    ("""3C KH 5D 5S KH""", 2),
    ("""QH 8H KD JH 8S""", 1),
    ("""2D 6D 9D TH 7D""", 0),
)

lowerCAmelCase__						:       List[str]										=						(
    ("""JH AH TH KH QH""", 23),
    ("""JH 9H TH KH QH""", 22),
    ("""JC KH JS JD JH""", 21),
    ("""KH KC 3S 3H 3D""", 20),
    ("""8C 9C 5C 3C TC""", 19),
    ("""JS QS 9H TS KH""", 18),
    ("""7C 7S KH 2H 7H""", 17),
    ("""3C KH 5D 5S KH""", 16),
    ("""QH 8H KD JH 8S""", 15),
    ("""2D 6D 9D TH 7D""", 14),
)




def 	_a       (				):

		"""simple docstring"""

		snake_case__						,       snake_case__						:       List[str]       						=  randrange(len(__lowerCAmelCase     )     ), randrange(len(__lowerCAmelCase     )     )
		snake_case__						:       int       						=  ['''Loss''', '''Tie''', '''Win'''][(play >= oppo) + (play > oppo)]
		snake_case__						,       snake_case__						:       Union[str, Any]       						=  SORTED_HANDS[play], SORTED_HANDS[oppo]
		return hand, other, expected




def 	_a       (				__lowerCAmelCase		:      int = 1_00     ):

		"""simple docstring"""

		return (generate_random_hand() for _ in range(__lowerCAmelCase     ))




@pytest.mark.parametrize('''hand, expected'''   ,							__lowerCAmelCase     )
def 	_a       (				__lowerCAmelCase		:      Optional[Any]   ,							__lowerCAmelCase		:      Any     ):

		"""simple docstring"""

		assert PokerHand(__lowerCAmelCase     )._is_flush() == expected




@pytest.mark.parametrize('''hand, expected'''   ,							__lowerCAmelCase     )
def 	_a       (				__lowerCAmelCase		:      List[str]   ,							__lowerCAmelCase		:      Tuple     ):

		"""simple docstring"""

		assert PokerHand(__lowerCAmelCase     )._is_straight() == expected




@pytest.mark.parametrize('''hand, expected, card_values'''   ,							__lowerCAmelCase     )
def 	_a       (				__lowerCAmelCase		:      List[Any]   ,							__lowerCAmelCase		:      Union[str, Any]   ,							__lowerCAmelCase		:      Optional[int]     ):

		"""simple docstring"""

		snake_case__						:       Union[str, Any]       						=  PokerHand(__lowerCAmelCase     )
		assert player._is_five_high_straight() == expected
		assert player._card_values == card_values




@pytest.mark.parametrize('''hand, expected'''   ,							__lowerCAmelCase     )
def 	_a       (				__lowerCAmelCase		:      List[str]   ,							__lowerCAmelCase		:      Tuple     ):

		"""simple docstring"""

		assert PokerHand(__lowerCAmelCase     )._is_same_kind() == expected




@pytest.mark.parametrize('''hand, expected'''   ,							__lowerCAmelCase     )
def 	_a       (				__lowerCAmelCase		:      str   ,							__lowerCAmelCase		:      Any     ):

		"""simple docstring"""

		assert PokerHand(__lowerCAmelCase     )._hand_type == expected




@pytest.mark.parametrize('''hand, other, expected'''   ,							__lowerCAmelCase     )
def 	_a       (				__lowerCAmelCase		:      List[str]   ,							__lowerCAmelCase		:      str   ,							__lowerCAmelCase		:      List[Any]     ):

		"""simple docstring"""

		assert PokerHand(__lowerCAmelCase     ).compare_with(PokerHand(__lowerCAmelCase     )     ) == expected




@pytest.mark.parametrize('''hand, other, expected'''   ,							generate_random_hands()     )
def 	_a       (				__lowerCAmelCase		:      Optional[Any]   ,							__lowerCAmelCase		:      Any   ,							__lowerCAmelCase		:      Any     ):

		"""simple docstring"""

		assert PokerHand(__lowerCAmelCase     ).compare_with(PokerHand(__lowerCAmelCase     )     ) == expected




def 	_a       (				):

		"""simple docstring"""

		snake_case__						:       str       						=  [PokerHand(__lowerCAmelCase     ) for hand in SORTED_HANDS]
		snake_case__						:       Optional[Any]       						=  poker_hands.copy()
		shuffle(__lowerCAmelCase     )
		snake_case__						:       str       						=  chain(sorted(__lowerCAmelCase     )     )
		for index, hand in enumerate(__lowerCAmelCase     ):
				assert hand == poker_hands[index]




def 	_a       (				):

		"""simple docstring"""

		snake_case__						:       str       						=  [PokerHand('''2D AC 3H 4H 5S'''     ), PokerHand('''2S 3H 4H 5S 6C'''     )]
		pokerhands.sort(reverse=__lowerCAmelCase     )
		assert pokerhands[0].__str__() == "2S 3H 4H 5S 6C"




def 	_a       (				):

		"""simple docstring"""

		snake_case__						:       Optional[int]       						=  PokerHand('''2C 4S AS 3D 5C'''     )
		snake_case__						:       int       						=  True
		snake_case__						:       List[Any]       						=  [5, 4, 3, 2, 14]
		for _ in range(10     ):
				assert pokerhand._is_five_high_straight() == expected
				assert pokerhand._card_values == expected_card_values





def 	_a       (				):

		"""simple docstring"""

		snake_case__						:       Optional[int]       						=  0
		snake_case__						:       Union[str, Any]       						=  os.path.abspath(os.path.dirname(__lowerCAmelCase     )     )
		snake_case__						:       Optional[int]       						=  os.path.join(__lowerCAmelCase   ,							'''poker_hands.txt'''     )
		with open(__lowerCAmelCase     ) as file_hand:
				for line in file_hand:
						snake_case__						:       Any       						=  line[:14].strip()
						snake_case__						:       Optional[Any]       						=  line[15:].strip()
						snake_case__						,       snake_case__						:       Optional[Any]       						=  PokerHand(__lowerCAmelCase     ), PokerHand(__lowerCAmelCase     )
						snake_case__						:       Optional[int]       						=  player.compare_with(__lowerCAmelCase     )
						if output == "Win":
								answer += 1
		assert answer == 3_76






'''simple docstring'''


import argparse

from transformers import BigBirdConfig, BigBirdForPreTraining, BigBirdForQuestionAnswering, load_tf_weights_in_big_bird
from transformers.utils import logging


logging.set_verbosity_info()




def 	_a       (				__lowerCAmelCase		:      Union[str, Any]   ,							__lowerCAmelCase		:      Any   ,							__lowerCAmelCase		:      Optional[int]   ,							__lowerCAmelCase		:      List[Any]     ):

		"""simple docstring"""

		snake_case__						:       Any       						=  BigBirdConfig.from_json_file(__lowerCAmelCase     )
		print(F"""Building PyTorch model from configuration: {config}"""     )

		if is_trivia_qa:
				snake_case__						:       str       						=  BigBirdForQuestionAnswering(__lowerCAmelCase     )
		else:
				snake_case__						:       Tuple       						=  BigBirdForPreTraining(__lowerCAmelCase     )

		# Load weights from tf checkpoint
		load_tf_weights_in_big_bird(__lowerCAmelCase   ,							__lowerCAmelCase   ,							is_trivia_qa=__lowerCAmelCase     )

		# Save pytorch-model
		print(F"""Save PyTorch model to {pytorch_dump_path}"""     )
		model.save_pretrained(__lowerCAmelCase     )


if __name__ == "__main__":
					lowerCAmelCase__						:       Union[str, Any]										=						argparse.ArgumentParser()
					# Required parameters
					parser.add_argument(
					    """--tf_checkpoint_path""", default=None, type=str, required=True, help="""Path to the TensorFlow checkpoint path."""
					)
					parser.add_argument(
					    """--big_bird_config_file""",
					    default=None,
					    type=str,
					    required=True,
					    help=(
					        """The config json file corresponding to the pre-trained BERT model. \n"""
					        """This specifies the model architecture."""
					    ),
					)
					parser.add_argument(
					    """--pytorch_dump_path""", default=None, type=str, required=True, help="""Path to the output PyTorch model."""
					)
					parser.add_argument(
					    """--is_trivia_qa""", action="""store_true""", help="""Whether to convert a model with a trivia_qa head."""
					)
					lowerCAmelCase__						:       Optional[int]										=						parser.parse_args()
					convert_tf_checkpoint_to_pytorch(
					    args.tf_checkpoint_path, args.big_bird_config_file, args.pytorch_dump_path, args.is_trivia_qa
					)



