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

import pytest

from .sola import PokerHand

__lowerCamelCase							=       (
    '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							=       (
    ('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							=       (
    ('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							=       (
    ('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							=       (
    ('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							=       (
    ('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							=       (
    ('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 			UpperCamelCase					(    ):
       snake_case       :   Dict							=      randrange(len(_lowerCamelCase    )    ), randrange(len(_lowerCamelCase    )    )
       snake_case       :   Optional[int]							=      ["Loss", "Tie", "Win"][(play >= oppo) + (play > oppo)]
       snake_case       :   Union[str, Any]							=      SORTED_HANDS[play], SORTED_HANDS[oppo]
       return hand, other, expected


def 			UpperCamelCase					(    __lowerCamelCase     : int = 100    ):
       return (generate_random_hand() for _ in range(_lowerCamelCase    ))


@pytest.mark.parametrize("hand, expected"				,					_lowerCamelCase    )
def 			UpperCamelCase					(    __lowerCamelCase     : List[str]				,					__lowerCamelCase     : Any    ):
       assert PokerHand(_lowerCamelCase    )._is_flush() == expected


@pytest.mark.parametrize("hand, expected"				,					_lowerCamelCase    )
def 			UpperCamelCase					(    __lowerCamelCase     : Dict				,					__lowerCamelCase     : Tuple    ):
       assert PokerHand(_lowerCamelCase    )._is_straight() == expected


@pytest.mark.parametrize("hand, expected, card_values"				,					_lowerCamelCase    )
def 			UpperCamelCase					(    __lowerCamelCase     : Dict				,					__lowerCamelCase     : int				,					__lowerCamelCase     : str    ):
       snake_case       :   List[str]							=      PokerHand(_lowerCamelCase    )
       assert player._is_five_high_straight() == expected
       assert player._card_values == card_values


@pytest.mark.parametrize("hand, expected"				,					_lowerCamelCase    )
def 			UpperCamelCase					(    __lowerCamelCase     : str				,					__lowerCamelCase     : Any    ):
       assert PokerHand(_lowerCamelCase    )._is_same_kind() == expected


@pytest.mark.parametrize("hand, expected"				,					_lowerCamelCase    )
def 			UpperCamelCase					(    __lowerCamelCase     : Tuple				,					__lowerCamelCase     : Dict    ):
       assert PokerHand(_lowerCamelCase    )._hand_type == expected


@pytest.mark.parametrize("hand, other, expected"				,					_lowerCamelCase    )
def 			UpperCamelCase					(    __lowerCamelCase     : Any				,					__lowerCamelCase     : List[Any]				,					__lowerCamelCase     : Tuple    ):
       assert PokerHand(_lowerCamelCase    ).compare_with(PokerHand(_lowerCamelCase    )    ) == expected


@pytest.mark.parametrize("hand, other, expected"				,					generate_random_hands()    )
def 			UpperCamelCase					(    __lowerCamelCase     : int				,					__lowerCamelCase     : Optional[Any]				,					__lowerCamelCase     : List[Any]    ):
       assert PokerHand(_lowerCamelCase    ).compare_with(PokerHand(_lowerCamelCase    )    ) == expected


def 			UpperCamelCase					(    ):
       snake_case       :   int							=      [PokerHand(_lowerCamelCase    ) for hand in SORTED_HANDS]
       snake_case       :   Dict							=      poker_hands.copy()
       shuffle(_lowerCamelCase    )
       snake_case       :   List[Any]							=      chain(sorted(_lowerCamelCase    )    )
       for index, hand in enumerate(_lowerCamelCase    ):
              assert hand == poker_hands[index]


def 			UpperCamelCase					(    ):
       snake_case       :   Optional[int]							=      [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 			UpperCamelCase					(    ):
       snake_case       :   Tuple							=      PokerHand("2C 4S AS 3D 5C"    )
       snake_case       :   int							=      True
       snake_case       :   Tuple							=      [5, 4, 3, 2, 14]
       for _ in range(10    ):
              assert pokerhand._is_five_high_straight() == expected
              assert pokerhand._card_values == expected_card_values




def 			UpperCamelCase					(    ):
       snake_case       :   List[str]							=      0
       snake_case       :   int							=      os.path.abspath(os.path.dirname(_lowerCamelCase    )    )
       snake_case       :   str							=      os.path.join(_lowerCamelCase				,					"poker_hands.txt"    )
       with open(_lowerCamelCase    ) as file_hand:
              for line in file_hand:
                     snake_case       :   Tuple							=      line[:14].strip()
                     snake_case       :   Union[str, Any]							=      line[15:].strip()
                     snake_case       :   str							=      PokerHand(_lowerCamelCase    ), PokerHand(_lowerCamelCase    )
                     snake_case       :   str							=      player.compare_with(_lowerCamelCase    )
                     if output == "Win":
                            answer += 1
       assert answer == 376







# XXX: we want transformers master here - in the absense of conftest manipulating sys.path:
# hack it in for now:
import sys
from pathlib import Path


lowercase  :					Optional[int]													=       Path(__file__).resolve().parents[3] / 'src'
sys.path.insert(1, str(git_repo_path))

import dataclasses  # noqa
import io  # noqa
import itertools  # noqa
import json  # noqa
import os  # noqa
import unittest  # noqa
from copy import deepcopy  # noqa

from parameterized import parameterized  # noqa
from transformers import TrainingArguments, is_torch_available  # noqa
from transformers.deepspeed import is_deepspeed_available  # noqa
from transformers.file_utils import WEIGHTS_NAME  # noqa
from transformers.testing_utils import (  # noqa
    CaptureLogger,
    ExtendSysPath,
    TestCasePlus,
    execute_subprocess_async,
    get_gpu_count,
    mockenv_context,
    require_deepspeed,
    require_torch_gpu,
    require_torch_multi_gpu,
    slow,
)
from transformers.trainer_utils import set_seed  # noqa


set_seed(42)

lowercase  :					Optional[Any]													=       {'base': 'patrickvonplaten/wav2vec2_tiny_random', 'robust': 'patrickvonplaten/wav2vec2_tiny_random_robust'}

lowercase  :					str													=       'zero2'
lowercase  :					Optional[int]													=       'zero3'
lowercase  :					Optional[Any]													=       [ZEROa, ZEROa]




def 	_SCREAMING_SNAKE_CASE					(		_lowerCamelCase :			str						,  _lowerCamelCase :			Union[str, Any]						,  _lowerCamelCase :			List[str])					->    int:






							'''simple docstring'''





							__UpperCamelCase	:   Union[str, Any]			   =					parameterized.to_safe_name("_".join(str(_lowerCamelCase) for x in param.args))
							return F'{func.__name__}_{param_based_name}'


# Cartesian-product of zero stages with models to test
lowercase  :					List[str]													=       list(itertools.product(stages, models.keys()))




@slow
@require_deepspeed
@require_torch_gpu
class  lowerCamelCase__							(				__lowercase):




	'''simple docstring'''




	@parameterized.expand(a     ,   name_func=a						)
	def    _lowerCamelCase				(					self		:Dict     ,   a		:Optional[Any]     ,   a		:str						)		-> Optional[int]:
								self.run_and_check(
								    stage=a     ,   model=a     ,   distributed=a     ,   fpaa=a     ,   )







	@require_torch_multi_gpu
	@parameterized.expand(a     ,   name_func=a						)
	def    _lowerCamelCase				(					self		:List[str]     ,   a		:str     ,   a		:str						)		-> List[Any]:
								self.run_and_check(
								    stage=a     ,   model=a     ,   distributed=a     ,   fpaa=a     ,   )







	@parameterized.expand(a     ,   name_func=a						)
	def    _lowerCamelCase				(					self		:List[Any]     ,   a		:List[str]     ,   a		:int						)		-> Optional[int]:
								self.run_and_check(
								    stage=a     ,   model=a     ,   distributed=a     ,   fpaa=a     ,   )







	@require_torch_multi_gpu
	@parameterized.expand(a     ,   name_func=a						)
	def    _lowerCamelCase				(					self		:List[str]     ,   a		:List[Any]     ,   a		:Dict						)		-> int:
								self.run_and_check(
								    stage=a     ,   model=a     ,   distributed=a     ,   fpaa=a     ,   )







	def    _lowerCamelCase				(					self		:Any     ,   a		:List[str]						)		-> Optional[Any]:
								# XXX: run_asr is premature and doesn't save any results
								# so all we check for now is that the process didn't fail
								pass







	def    _lowerCamelCase				(					self		:Optional[Any]     ,   a		:str     ,   a		:str     ,   a		:int = 1_0     ,   a		:bool = True     ,   a		:bool = True     ,   a		:bool = True     ,   )		-> Any:
								__UpperCamelCase	:   Optional[Any]			   =					models[model]

								__UpperCamelCase	:   List[Any]			   =					self.run_trainer(
								    stage=a     ,   model_name=a     ,   eval_steps=a     ,   num_train_epochs=1     ,   distributed=a     ,   fpaa=a     ,   )

								self.do_checks(a						)

								return output_dir







	def    _lowerCamelCase				(					self		:List[str]     ,   a		:str     ,   a		:str     ,   a		:int = 1_0     ,   a		:int = 1     ,   a		:bool = True     ,   a		:bool = True     ,   )		-> Dict:
								__UpperCamelCase	:   int			   =					self.get_auto_remove_tmp_dir("./xxx"     ,   after=a						)
								__UpperCamelCase	:   int			   =					f'\n            --model_name_or_path {model_name}\n            --dataset_name hf-internal-testing/librispeech_asr_dummy\n            --dataset_config_name clean\n            --train_split_name validation\n            --validation_split_name validation\n            --output_dir {output_dir}\n            --num_train_epochs {str(a						)}\n            --per_device_train_batch_size 2\n            --per_device_eval_batch_size 2\n            --evaluation_strategy steps\n            --learning_rate 5e-4\n            --warmup_steps 8\n            --orthography timit\n            --preprocessing_num_workers 1\n            --group_by_length\n            --freeze_feature_extractor\n            --report_to none\n            --save_steps 0\n            --eval_steps {eval_steps}\n            --report_to none\n        '.split()

								if fpaa:
															args.extend(["--fp16"]						)

								# currently ds_config_wav2vec2_zero.json requires "zero_optimization.find_unused_parameters": true,
								# hence the separate config files
								__UpperCamelCase	:   Dict			   =					f'--deepspeed {self.test_file_dir_str}/ds_config_wav2vec2_{stage}.json'.split()
								__UpperCamelCase	:   int			   =					[f'{self.examples_dir_str}/research_projects/wav2vec2/run_asr.py']
								__UpperCamelCase	:   Optional[Any]			   =					self.get_launcher(a						)

								__UpperCamelCase	:   Optional[int]			   =					launcher + script + args + ds_args
								# keep for quick debug
								# print(" ".join([f"\nPYTHONPATH={self.src_dir_str}"] +cmd)); die
								execute_subprocess_async(a     ,   env=self.get_env()						)

								return output_dir







	def    _lowerCamelCase				(					self		:Any     ,   a		:List[Any]=False						)		-> List[Any]:
								# 1. explicitly set --num_nodes=1 just in case these tests end up run on a multi-node setup
								# - it won't be able to handle that
								# 2. for now testing with just 2 gpus max (since some quality tests may give different
								# results with mode gpus because we use very little data)
								__UpperCamelCase	:   List[Any]			   =					min(2     ,   get_gpu_count()						) if distributed else 1
								return f'deepspeed --num_nodes 1 --num_gpus {num_gpus}'.split()