

"""simple docstring"""






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, 1_4]),
    ("""2H 5D 3C AS 5S""", False, [1_4, 5, 5, 3, 2]),
    ("""JH QD KC AS TS""", False, [1_4, 1_3, 1_2, 1_1, 1_0]),
    ("""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""", 2_3),
    ("""JH 9H TH KH QH""", 2_2),
    ("""JC KH JS JD JH""", 2_1),
    ("""KH KC 3S 3H 3D""", 2_0),
    ("""8C 9C 5C 3C TC""", 1_9),
    ("""JS QS 9H TS KH""", 1_8),
    ("""7C 7S KH 2H 7H""", 1_7),
    ("""3C KH 5D 5S KH""", 1_6),
    ("""QH 8H KD JH 8S""", 1_5),
    ("""2D 6D 9D TH 7D""", 1_4),
)

def 		lowerCamelCase__		(					):

    '''simple docstring'''



    _lowerCAmelCase,					_lowerCAmelCase       :   Optional[Any]  =    randrange(len(_lowerCamelCase   )   ), randrange(len(_lowerCamelCase   )   )
    _lowerCAmelCase       :   List[Any]  =    ['Loss', 'Tie', 'Win'][(play >= oppo) + (play > oppo)]
    _lowerCAmelCase,					_lowerCAmelCase       :   int  =    SORTED_HANDS[play], SORTED_HANDS[oppo]
    return hand, other, expected

def 		lowerCamelCase__		(					_lowerCamelCase = 100   ):

    '''simple docstring'''



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

@pytest.mark.parametrize('hand, expected' ,   _lowerCamelCase   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



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

@pytest.mark.parametrize('hand, expected' ,   _lowerCamelCase   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



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

@pytest.mark.parametrize('hand, expected, card_values' ,   _lowerCamelCase   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



    _lowerCAmelCase       :   int  =    PokerHand(_lowerCamelCase   )
    assert player._is_five_high_straight() == expected
    assert player._card_values == card_values

@pytest.mark.parametrize('hand, expected' ,   _lowerCamelCase   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



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

@pytest.mark.parametrize('hand, expected' ,   _lowerCamelCase   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



    assert PokerHand(_lowerCamelCase   )._hand_type == expected

@pytest.mark.parametrize('hand, other, expected' ,   _lowerCamelCase   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



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

@pytest.mark.parametrize('hand, other, expected' ,   generate_random_hands()   )
def 		lowerCamelCase__		(					_lowerCamelCase ,   _lowerCamelCase ,   _lowerCamelCase   ):

    '''simple docstring'''



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

def 		lowerCamelCase__		(					):

    '''simple docstring'''



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

def 		lowerCamelCase__		(					):

    '''simple docstring'''



    _lowerCAmelCase       :   Tuple  =    [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 		lowerCamelCase__		(					):

    '''simple docstring'''



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

def 		lowerCamelCase__		(					):

    '''simple docstring'''



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








"""simple docstring"""






import inspect
import os
import unittest

import torch

import accelerate
from accelerate import Accelerator
from accelerate.test_utils import execute_subprocess_async, require_multi_gpu
from accelerate.utils import patch_environment







class 	__UpperCamelCase    (    unittest.TestCase						):

  def __lowerCamelCase		(		self      ):
      '''simple docstring'''

      _lowerCAmelCase       :   Dict  =    inspect.getfile(accelerate.test_utils      )
      _lowerCAmelCase       :   Union[str, Any]  =    os.path.sep.join(mod_file.split(os.path.sep      )[:-1] + ['scripts', 'test_script.py']      )
      _lowerCAmelCase       :   int  =    os.path.sep.join(
          mod_file.split(os.path.sep      )[:-1] + ['scripts', 'test_distributed_data_loop.py']      )
      _lowerCAmelCase       :   Optional[Any]  =    os.path.sep.join(mod_file.split(os.path.sep      )[:-1] + ['scripts', 'test_ops.py']      )

  @require_multi_gpu
  def __lowerCamelCase		(		self      ):
      '''simple docstring'''

      print(F"""Found {torch.cuda.device_count()} devices."""      )
      _lowerCAmelCase       :   int  =    ['torchrun', F"""--nproc_per_node={torch.cuda.device_count()}""", self.test_file_path]
      with patch_environment(omp_num_threads=1      ):
          execute_subprocess_async(_A    ,env=os.environ.copy()      )

  @require_multi_gpu
  def __lowerCamelCase		(		self      ):
      '''simple docstring'''

      print(F"""Found {torch.cuda.device_count()} devices."""      )
      _lowerCAmelCase       :   str  =    ['torchrun', F"""--nproc_per_node={torch.cuda.device_count()}""", self.operation_file_path]
      print(F"""Command: {cmd}"""      )
      with patch_environment(omp_num_threads=1      ):
          execute_subprocess_async(_A    ,env=os.environ.copy()      )

  @require_multi_gpu
  def __lowerCamelCase		(		self      ):
      '''simple docstring'''

      _lowerCAmelCase       :   List[Any]  =    ['torchrun', F"""--nproc_per_node={torch.cuda.device_count()}""", inspect.getfile(self.__class__      )]
      with patch_environment(omp_num_threads=1      ):
          execute_subprocess_async(_A    ,env=os.environ.copy()      )







  @require_multi_gpu
  def __lowerCamelCase		(		self      ):
      '''simple docstring'''

      print(F"""Found {torch.cuda.device_count()} devices, using 2 devices only"""      )
      _lowerCAmelCase       :   Tuple  =    ['torchrun', F"""--nproc_per_node={torch.cuda.device_count()}""", self.data_loop_file_path]
      with patch_environment(omp_num_threads=1    ,cuda_visible_devices='0,1'      ):
          execute_subprocess_async(_A    ,env=os.environ.copy()      )


if __name__ == "__main__":
 _lowerCAmelCase						=		Accelerator()
 _lowerCAmelCase						=		(accelerator.state.process_index + 2, 1_0)
 _lowerCAmelCase						=		torch.randint(0, 1_0, shape).to(accelerator.device)

 _lowerCAmelCase						=		""""""

 _lowerCAmelCase						=		accelerator.pad_across_processes(tensor)
 if tensora.shape[0] != accelerator.state.num_processes + 1:
  error_msg += F"Found shape {tensora.shape} but should have {accelerator.state.num_processes + 1} at dim 0."
 if not torch.equal(tensora[: accelerator.state.process_index + 2], tensor):
  error_msg += "Tensors have different values."
 if not torch.all(tensora[accelerator.state.process_index + 2 :] == 0):
  error_msg += "Padding was not done with the right value (0)."

 _lowerCAmelCase						=		accelerator.pad_across_processes(tensor, pad_first=True)
 if tensora.shape[0] != accelerator.state.num_processes + 1:
  error_msg += F"Found shape {tensora.shape} but should have {accelerator.state.num_processes + 1} at dim 0."
 _lowerCAmelCase						=		accelerator.state.num_processes - accelerator.state.process_index - 1
 if not torch.equal(tensora[index:], tensor):
  error_msg += "Tensors have different values."
 if not torch.all(tensora[:index] == 0):
  error_msg += "Padding was not done with the right value (0)."

 # Raise error at the end to make sure we don't stop at the first failure.
 if len(error_msg) > 0:
  raise ValueError(error_msg)





