

import unittest

import numpy as np

from transformers import DistilBertConfig, 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.distilbert.modeling_flax_distilbert import (
					    FlaxDistilBertForMaskedLM,
					    FlaxDistilBertForMultipleChoice,
					    FlaxDistilBertForQuestionAnswering,
					    FlaxDistilBertForSequenceClassification,
					    FlaxDistilBertForTokenClassification,
					    FlaxDistilBertModel,
					)






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

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

								__lowerCamelCase				:				str					       = DistilBertConfig(
								    vocab_size=self.vocab_size   ,dim=self.hidden_size   ,n_layers=self.num_hidden_layers   ,n_heads=self.num_attention_heads   ,hidden_dim=self.intermediate_size   ,hidden_act=self.hidden_act   ,dropout=self.hidden_dropout_prob   ,attention_dropout=self.attention_probs_dropout_prob   ,max_position_embeddings=self.max_position_embeddings   ,initializer_range=self.initializer_range   ,tie_weights_=SCREAMING_SNAKE_CASE__   ,)

								return config, input_ids, attention_mask


							def     lowerCAmelCase     (		self		:			List[Any]):
								__lowerCamelCase				:				List[str]					       = self.prepare_config_and_inputs()
								__lowerCamelCase       , __lowerCamelCase       , __lowerCamelCase				:				Dict					       = config_and_inputs
								__lowerCamelCase				:				Any					       = {'input_ids': input_ids, 'attention_mask': attention_mask}
								return config, inputs_dict






@require_flax
class        A_		(       SCREAMING_SNAKE_CASE			,							unittest.TestCase				):
							_UpperCAmelCase    :  Dict						      =      (
							    (
							        FlaxDistilBertModel,
							        FlaxDistilBertForMaskedLM,
							        FlaxDistilBertForMultipleChoice,
							        FlaxDistilBertForQuestionAnswering,
							        FlaxDistilBertForSequenceClassification,
							        FlaxDistilBertForTokenClassification,
							        FlaxDistilBertForQuestionAnswering,
							    )
							    if is_flax_available()
							    else ()
							)
							def     lowerCAmelCase     (		self		:			Optional[Any]):
								__lowerCamelCase				:				Tuple					       = FlaxDistilBertModelTester(self)


							@slow
							def     lowerCAmelCase     (		self		:			int):
								for model_class_name in self.all_model_classes:
									__lowerCamelCase				:				List[Any]					       = model_class_name.from_pretrained('distilbert-base-uncased')
									__lowerCamelCase				:				List[str]					       = model(np.ones((1, 1)))
									self.assertIsNotNone(SCREAMING_SNAKE_CASE__)






@require_flax
class        A_		(       unittest.TestCase				):
							@slow
							def     lowerCAmelCase     (		self		:			str):
								__lowerCamelCase				:				Union[str, Any]					       = FlaxDistilBertModel.from_pretrained('distilbert-base-uncased')
								__lowerCamelCase				:				str					       = 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				:				List[Any]					       = np.array([[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]])
								__lowerCamelCase				:				Union[str, Any]					       = model(SCREAMING_SNAKE_CASE__   ,attention_mask=SCREAMING_SNAKE_CASE__)[0]
								__lowerCamelCase				:				Optional[int]					       = (1, 1_1, 7_6_8)
								self.assertEqual(output.shape   ,SCREAMING_SNAKE_CASE__)
								__lowerCamelCase				:				Union[str, Any]					       = np.array([[[-0.1639, 0.3299, 0.1648], [-0.1746, 0.3289, 0.1710], [-0.1884, 0.3357, 0.1810]]])

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









'''simple docstring'''


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

import pytest

from .sola import PokerHand

_lowerCamelCase	:       Optional[int]      	=	(
    "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	:       Union[str, 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	:       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),
)

_lowerCamelCase	:       Dict      	=	(
    ("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	:       Optional[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]),
)

_lowerCamelCase	:       List[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),
)

_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 				__lowerCamelCase				(   )       ->	Optional[Any]:






   """simple docstring"""
   UpperCamelCase				,      UpperCamelCase					=  randrange(len(A__    )    ), randrange(len(A__    )    )
   UpperCamelCase					=  ['Loss', 'Tie', 'Win'][(play >= oppo) + (play > oppo)]
   UpperCamelCase				,      UpperCamelCase					=  SORTED_HANDS[play], SORTED_HANDS[oppo]
   return hand, other, expected







def 				__lowerCamelCase				(   A__ = 100    )       ->	Optional[Any]:






   """simple docstring"""
   return (generate_random_hand() for _ in range(A__    ))







@pytest.mark.parametrize('hand, expected'    ,						A__    )
def 				__lowerCamelCase				(   A__    ,						A__    )       ->	Any:






   """simple docstring"""
   assert PokerHand(A__    )._is_flush() == expected







@pytest.mark.parametrize('hand, expected'    ,						A__    )
def 				__lowerCamelCase				(   A__    ,						A__    )       ->	Any:






   """simple docstring"""
   assert PokerHand(A__    )._is_straight() == expected







@pytest.mark.parametrize('hand, expected, card_values'    ,						A__    )
def 				__lowerCamelCase				(   A__    ,						A__    ,						A__    )       ->	str:






   """simple docstring"""
   UpperCamelCase					=  PokerHand(A__    )
   assert player._is_five_high_straight() == expected
   assert player._card_values == card_values







@pytest.mark.parametrize('hand, expected'    ,						A__    )
def 				__lowerCamelCase				(   A__    ,						A__    )       ->	Dict:






   """simple docstring"""
   assert PokerHand(A__    )._is_same_kind() == expected







@pytest.mark.parametrize('hand, expected'    ,						A__    )
def 				__lowerCamelCase				(   A__    ,						A__    )       ->	str:






   """simple docstring"""
   assert PokerHand(A__    )._hand_type == expected







@pytest.mark.parametrize('hand, other, expected'    ,						A__    )
def 				__lowerCamelCase				(   A__    ,						A__    ,						A__    )       ->	Tuple:






   """simple docstring"""
   assert PokerHand(A__    ).compare_with(PokerHand(A__    )    ) == expected







@pytest.mark.parametrize('hand, other, expected'    ,						generate_random_hands()    )
def 				__lowerCamelCase				(   A__    ,						A__    ,						A__    )       ->	List[str]:






   """simple docstring"""
   assert PokerHand(A__    ).compare_with(PokerHand(A__    )    ) == expected







def 				__lowerCamelCase				(   )       ->	str:






   """simple docstring"""
   UpperCamelCase					=  [PokerHand(A__    ) for hand in SORTED_HANDS]
   UpperCamelCase					=  poker_hands.copy()
   shuffle(A__    )
   UpperCamelCase					=  chain(sorted(A__    )    )
   for index, hand in enumerate(A__    ):
      assert hand == poker_hands[index]







def 				__lowerCamelCase				(   )       ->	Optional[int]:






   """simple docstring"""
   # Test that five high straights are compared correctly.
   UpperCamelCase					=  [PokerHand('2D AC 3H 4H 5S'    ), PokerHand('2S 3H 4H 5S 6C'    )]
   pokerhands.sort(reverse=A__    )
   assert pokerhands[0].__str__() == "2S 3H 4H 5S 6C"







def 				__lowerCamelCase				(   )       ->	str:






   """simple docstring"""
   # Multiple calls to five_high_straight function should still return True
   # and shouldn't mutate the list in every call other than the first.
   UpperCamelCase					=  PokerHand('2C 4S AS 3D 5C'    )
   UpperCamelCase					=  True
   UpperCamelCase					=  [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				(   )       ->	List[str]:






   """simple docstring"""
   # Problem number 54 from Project Euler
   # Testing from poker_hands.txt file
   UpperCamelCase					=  0
   UpperCamelCase					=  os.path.abspath(os.path.dirname(A__    )    )
   UpperCamelCase					=  os.path.join(A__    ,						'poker_hands.txt'    )
   with open(A__    ) as file_hand:
      for line in file_hand:
         UpperCamelCase					=  line[:14].strip()
         UpperCamelCase					=  line[15:].strip()
         UpperCamelCase				,      UpperCamelCase					=  PokerHand(A__    ), PokerHand(A__    )
         UpperCamelCase					=  player.compare_with(A__    )
         if output == "Win":
            answer += 1
   assert answer == 376

