
'''simple docstring'''






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

import pytest

from .sola import PokerHand

UpperCAmelCase_       :   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)
)

UpperCAmelCase_       :   int   =       (
    ('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'),
)

UpperCAmelCase_       :   Dict   =       (
    ('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),
)

UpperCAmelCase_       :   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),
)

UpperCAmelCase_       :   List[Any]   =       (
    ('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]),
)

UpperCAmelCase_       :   Optional[Any]   =       (
    ('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),
)

UpperCAmelCase_       :   Optional[int]   =       (
    ('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"""




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







def 		A_       (	_lowerCAmelCase   :    int = 100							):



      """simple docstring"""




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







@pytest.mark.parametrize("hand, expected"				,	_lowerCAmelCase							)
def 		A_       (	_lowerCAmelCase   :    Optional[Any]				,	_lowerCAmelCase   :    Optional[Any]							):



      """simple docstring"""




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







@pytest.mark.parametrize("hand, expected"				,	_lowerCAmelCase							)
def 		A_       (	_lowerCAmelCase   :    Optional[Any]				,	_lowerCAmelCase   :    List[str]							):



      """simple docstring"""




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







@pytest.mark.parametrize("hand, expected, card_values"				,	_lowerCAmelCase							)
def 		A_       (	_lowerCAmelCase   :    Optional[int]				,	_lowerCAmelCase   :    Any				,	_lowerCAmelCase   :    Tuple							):



      """simple docstring"""




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







@pytest.mark.parametrize("hand, expected"				,	_lowerCAmelCase							)
def 		A_       (	_lowerCAmelCase   :    Any				,	_lowerCAmelCase   :    int							):



      """simple docstring"""




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







@pytest.mark.parametrize("hand, expected"				,	_lowerCAmelCase							)
def 		A_       (	_lowerCAmelCase   :    Tuple				,	_lowerCAmelCase   :    Optional[int]							):



      """simple docstring"""




      assert PokerHand(_lowerCAmelCase							)._hand_type == expected







@pytest.mark.parametrize("hand, other, expected"				,	_lowerCAmelCase							)
def 		A_       (	_lowerCAmelCase   :    Optional[int]				,	_lowerCAmelCase   :    Tuple				,	_lowerCAmelCase   :    List[str]							):



      """simple docstring"""




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







@pytest.mark.parametrize("hand, other, expected"				,	generate_random_hands()							)
def 		A_       (	_lowerCAmelCase   :    Optional[int]				,	_lowerCAmelCase   :    Optional[Any]				,	_lowerCAmelCase   :    int							):



      """simple docstring"""




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







def 		A_       (	):



      """simple docstring"""




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







def 		A_       (	):



      """simple docstring"""




      _lowerCamelCase				:   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"""




      _lowerCamelCase				:   List[Any]								=   PokerHand("2C 4S AS 3D 5C"							)
      _lowerCamelCase				:   Optional[int]								=   True
      _lowerCamelCase				:   Dict								=   [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"""




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

'''simple docstring'''






import unittest

import numpy as np

from transformers import AlbertConfig, is_flax_available
from transformers.testing_utils import require_flax, slow

from ...test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, random_attention_mask


if is_flax_available():
 import jax.numpy as jnp

 from transformers.models.albert.modeling_flax_albert import (
     FlaxAlbertForMaskedLM,
     FlaxAlbertForMultipleChoice,
     FlaxAlbertForPreTraining,
     FlaxAlbertForQuestionAnswering,
     FlaxAlbertForSequenceClassification,
     FlaxAlbertForTokenClassification,
     FlaxAlbertModel,
 )

class  UpperCAmelCase__						(       unittest.TestCase ):



    def __init__(	self      :       Union[str, Any],__A      :       Union[str, Any],__A      :       Optional[int]=1_3,__A      :       Optional[Any]=7,__A      :       int=True,__A      :       Any=True,__A      :       Dict=True,__A      :       List[Any]=True,__A      :       Optional[Any]=9_9,__A      :       List[str]=3_2,__A      :       int=5,__A      :       str=4,__A      :       int=3_7,__A      :       Optional[Any]="gelu",__A      :       Union[str, Any]=0.1,__A      :       str=0.1,__A      :       Any=5_1_2,__A      :       Any=1_6,__A      :       Optional[int]=2,__A      :       int=0.02,__A      :       Optional[int]=4,):
          _lowerCamelCase				:   Union[str, Any]								=   parent
          _lowerCamelCase				:   Tuple								=   batch_size
          _lowerCamelCase				:   List[str]								=   seq_length
          _lowerCamelCase				:   Optional[int]								=   is_training
          _lowerCamelCase				:   str								=   use_attention_mask
          _lowerCamelCase				:   str								=   use_token_type_ids
          _lowerCamelCase				:   List[str]								=   use_labels
          _lowerCamelCase				:   Tuple								=   vocab_size
          _lowerCamelCase				:   Dict								=   hidden_size
          _lowerCamelCase				:   str								=   num_hidden_layers
          _lowerCamelCase				:   List[str]								=   num_attention_heads
          _lowerCamelCase				:   Optional[Any]								=   intermediate_size
          _lowerCamelCase				:   Optional[int]								=   hidden_act
          _lowerCamelCase				:   Any								=   hidden_dropout_prob
          _lowerCamelCase				:   List[Any]								=   attention_probs_dropout_prob
          _lowerCamelCase				:   Tuple								=   max_position_embeddings
          _lowerCamelCase				:   str								=   type_vocab_size
          _lowerCamelCase				:   str								=   type_sequence_label_size
          _lowerCamelCase				:   int								=   initializer_range
          _lowerCamelCase				:   List[Any]								=   num_choices



    def   lowerCamelCase_  (	self      :       Optional[int]     ):
          _lowerCamelCase				:   List[Any]								=   ids_tensor([self.batch_size, self.seq_length],self.vocab_size     )

          _lowerCamelCase				:   Union[str, Any]								=   None
          if self.use_attention_mask:
                _lowerCamelCase				:   Tuple								=   random_attention_mask([self.batch_size, self.seq_length]     )

          _lowerCamelCase				:   Any								=   None
          if self.use_token_type_ids:
                _lowerCamelCase				:   int								=   ids_tensor([self.batch_size, self.seq_length],self.type_vocab_size     )

          _lowerCamelCase				:   int								=   AlbertConfig(
              vocab_size=self.vocab_size,hidden_size=self.hidden_size,num_hidden_layers=self.num_hidden_layers,num_attention_heads=self.num_attention_heads,intermediate_size=self.intermediate_size,hidden_act=self.hidden_act,hidden_dropout_prob=self.hidden_dropout_prob,attention_probs_dropout_prob=self.attention_probs_dropout_prob,max_position_embeddings=self.max_position_embeddings,type_vocab_size=self.type_vocab_size,is_decoder=__A,initializer_range=self.initializer_range,)

          return config, input_ids, token_type_ids, attention_mask



    def   lowerCamelCase_  (	self      :       Dict     ):
          _lowerCamelCase				:   str								=   self.prepare_config_and_inputs()
          _lowerCamelCase						,		_lowerCamelCase						,		_lowerCamelCase						,		_lowerCamelCase				:   Union[str, Any]								=   config_and_inputs
          _lowerCamelCase				:   Any								=   {"input_ids": input_ids, "token_type_ids": token_type_ids, "attention_mask": attention_mask}
          return config, inputs_dict


@require_flax
class  UpperCAmelCase__						(       A						,	unittest.TestCase ):
    lowerCAmelCase_				    =		(
        (
            FlaxAlbertModel,
            FlaxAlbertForPreTraining,
            FlaxAlbertForMaskedLM,
            FlaxAlbertForMultipleChoice,
            FlaxAlbertForQuestionAnswering,
            FlaxAlbertForSequenceClassification,
            FlaxAlbertForTokenClassification,
            FlaxAlbertForQuestionAnswering,
        )
        if is_flax_available()
        else ()
    )



    def   lowerCamelCase_  (	self      :       Any     ):
          _lowerCamelCase				:   List[str]								=   FlaxAlbertModelTester(self     )



    @slow
    def   lowerCamelCase_  (	self      :       Union[str, Any]     ):
          for model_class_name in self.all_model_classes:
                _lowerCamelCase				:   str								=   model_class_name.from_pretrained("albert-base-v2"     )
                _lowerCamelCase				:   Any								=   model(np.ones((1, 1)     )     )
                self.assertIsNotNone(__A     )


@require_flax
class  UpperCAmelCase__						(       unittest.TestCase ):



    @slow
    def   lowerCamelCase_  (	self      :       Tuple     ):
          _lowerCamelCase				:   Optional[int]								=   FlaxAlbertModel.from_pretrained("albert-base-v2"     )
          _lowerCamelCase				:   int								=   np.array([[0, 3_4_5, 2_3_2, 3_2_8, 7_4_0, 1_4_0, 1_6_9_5, 6_9, 6_0_7_8, 1_5_8_8, 2]]     )
          _lowerCamelCase				:   Union[str, Any]								=   np.array([[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]     )
          _lowerCamelCase				:   Optional[Any]								=   model(__A,attention_mask=__A     )[0]
          _lowerCamelCase				:   Dict								=   (1, 1_1, 7_6_8)
          self.assertEqual(output.shape,__A     )
          _lowerCamelCase				:   Tuple								=   np.array(
              [[[-0.6513, 1.5035, -0.2766], [-0.6515, 1.5046, -0.2780], [-0.6512, 1.5049, -0.2784]]]     )

          self.assertTrue(jnp.allclose(output[:, 1:4, 1:4],__A,atol=1e-4     )     )