
"""simple docstring"""

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

import pytest

from .sola import PokerHand

UpperCamelCase     :    Any       =			(
    "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     :    str       =			(
    ("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     :    List[str]       =			(
    ("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[Any]       =			(
    ("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[str]       =			(
    ("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]),
)

UpperCamelCase     :    Tuple       =			(
    ("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     :    List[str]       =			(
    ("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 							A						(    )				->     Optional[int]:
 __UpperCamelCase  ,     __UpperCamelCase      =	randrange(len(snake_case		)		), randrange(len(snake_case		)		)
 __UpperCamelCase      =	['Loss', 'Tie', 'Win'][(play >= oppo) + (play > oppo)]
 __UpperCamelCase  ,     __UpperCamelCase      =	SORTED_HANDS[play], SORTED_HANDS[oppo]
 return hand, other, expected






def 							A						(    snake_case				:int = 1_0_0		)				->     Optional[int]:
 return (generate_random_hand() for _ in range(snake_case		))






@pytest.mark.parametrize('hand, expected'    , snake_case		)
def 							A						(    snake_case				:Union[str, Any]    , snake_case				:Optional[int]		)				->     Optional[Any]:
 assert PokerHand(snake_case		)._is_flush() == expected






@pytest.mark.parametrize('hand, expected'    , snake_case		)
def 							A						(    snake_case				:List[str]    , snake_case				:str		)				->     int:
 assert PokerHand(snake_case		)._is_straight() == expected






@pytest.mark.parametrize('hand, expected, card_values'    , snake_case		)
def 							A						(    snake_case				:str    , snake_case				:Dict    , snake_case				:str		)				->     Optional[Any]:
 __UpperCamelCase      =	PokerHand(snake_case		)
 assert player._is_five_high_straight() == expected
 assert player._card_values == card_values






@pytest.mark.parametrize('hand, expected'    , snake_case		)
def 							A						(    snake_case				:Optional[int]    , snake_case				:int		)				->     Dict:
 assert PokerHand(snake_case		)._is_same_kind() == expected






@pytest.mark.parametrize('hand, expected'    , snake_case		)
def 							A						(    snake_case				:int    , snake_case				:Optional[Any]		)				->     Dict:
 assert PokerHand(snake_case		)._hand_type == expected






@pytest.mark.parametrize('hand, other, expected'    , snake_case		)
def 							A						(    snake_case				:List[str]    , snake_case				:str    , snake_case				:Any		)				->     Optional[int]:
 assert PokerHand(snake_case		).compare_with(PokerHand(snake_case		)		) == expected






@pytest.mark.parametrize('hand, other, expected'    , generate_random_hands()		)
def 							A						(    snake_case				:int    , snake_case				:List[Any]    , snake_case				:List[str]		)				->     List[Any]:
 assert PokerHand(snake_case		).compare_with(PokerHand(snake_case		)		) == expected






def 							A						(    )				->     Dict:
 __UpperCamelCase      =	[PokerHand(snake_case		) for hand in SORTED_HANDS]
 __UpperCamelCase      =	poker_hands.copy()
 shuffle(snake_case		)
 __UpperCamelCase      =	chain(sorted(snake_case		)		)
 for index, hand in enumerate(snake_case		):
  assert hand == poker_hands[index]






def 							A						(    )				->     Optional[int]:
 # Test that five high straights are compared correctly.
 __UpperCamelCase      =	[PokerHand('2D AC 3H 4H 5S'		), PokerHand('2S 3H 4H 5S 6C'		)]
 pokerhands.sort(reverse=snake_case		)
 assert pokerhands[0].__str__() == "2S 3H 4H 5S 6C"






def 							A						(    )				->     str:
 # 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, 1_4]
 for _ in range(1_0		):
  assert pokerhand._is_five_high_straight() == expected
  assert pokerhand._card_values == expected_card_values






def 							A						(    )				->     List[Any]:
 # Problem number 54 from Project Euler
 # Testing from poker_hands.txt file
 __UpperCamelCase      =	0
 __UpperCamelCase      =	os.path.abspath(os.path.dirname(snake_case		)		)
 __UpperCamelCase      =	os.path.join(snake_case    , 'poker_hands.txt'		)
 with open(snake_case		) as file_hand:
  for line in file_hand:
   __UpperCamelCase      =	line[:1_4].strip()
   __UpperCamelCase      =	line[1_5:].strip()
   __UpperCamelCase  ,     __UpperCamelCase      =	PokerHand(snake_case		), PokerHand(snake_case		)
   __UpperCamelCase      =	player.compare_with(snake_case		)
   if output == "Win":
    answer += 1
 assert answer == 3_7_6





"""simple docstring"""

import multiprocessing
import os
from typing import BinaryIO, Optional, Union

import fsspec

from .. import Dataset, Features, NamedSplit, config
from ..formatting import query_table
from ..packaged_modules.json.json import Json
from ..utils import logging
from ..utils.typing import NestedDataStructureLike, PathLike
from .abc import AbstractDatasetReader

class __lowerCAmelCase		(				__SCREAMING_SNAKE_CASE      ):





     def __init__(     self     ,					__UpperCAmelCase     ,					__UpperCAmelCase = None     ,					__UpperCAmelCase = None     ,					__UpperCAmelCase = None     ,					__UpperCAmelCase = False     ,					__UpperCAmelCase = False     ,					__UpperCAmelCase = None     ,					__UpperCAmelCase = None     ,					**__UpperCAmelCase     ,					):

      '''simple docstring'''




      super().__init__(
          __UpperCAmelCase     ,					split=__UpperCAmelCase     ,					features=__UpperCAmelCase     ,					cache_dir=__UpperCAmelCase     ,					keep_in_memory=__UpperCAmelCase     ,					streaming=__UpperCAmelCase     ,					num_proc=__UpperCAmelCase     ,					**__UpperCAmelCase     ,					)
      __UpperCamelCase      =	field
      __UpperCamelCase      =	path_or_paths if isinstance(__UpperCAmelCase     ,					__UpperCAmelCase				) else {self.split: path_or_paths}
      __UpperCamelCase      =	Json(
          cache_dir=__UpperCAmelCase     ,					data_files=__UpperCAmelCase     ,					features=__UpperCAmelCase     ,					field=__UpperCAmelCase     ,					**__UpperCAmelCase     ,					)







     def 						UpperCAmelCase (     self				):

      '''simple docstring'''




      if self.streaming:
       __UpperCamelCase      =	self.builder.as_streaming_dataset(split=self.split				)
      # Build regular (map-style) dataset
      else:
       __UpperCamelCase      =	None
       __UpperCamelCase      =	None
       __UpperCamelCase      =	None
       __UpperCamelCase      =	None

       self.builder.download_and_prepare(
           download_config=__UpperCAmelCase     ,					download_mode=__UpperCAmelCase     ,					verification_mode=__UpperCAmelCase     ,					base_path=__UpperCAmelCase     ,					num_proc=self.num_proc     ,					)
       __UpperCamelCase      =	self.builder.as_dataset(
           split=self.split     ,					verification_mode=__UpperCAmelCase     ,					in_memory=self.keep_in_memory				)
      return dataset


class __lowerCAmelCase		:





     def __init__(     self     ,					__UpperCAmelCase     ,					__UpperCAmelCase     ,					__UpperCAmelCase = None     ,					__UpperCAmelCase = None     ,					**__UpperCAmelCase     ,					):

      '''simple docstring'''




      if num_proc is not None and num_proc <= 0:
       raise ValueError(F'num_proc {num_proc} must be an integer > 0.'				)

      __UpperCamelCase      =	dataset
      __UpperCamelCase      =	path_or_buf
      __UpperCamelCase      =	batch_size if batch_size else config.DEFAULT_MAX_BATCH_SIZE
      __UpperCamelCase      =	num_proc
      __UpperCamelCase      =	'utf-8'
      __UpperCamelCase      =	to_json_kwargs





     def 						UpperCAmelCase (     self				):

      '''simple docstring'''




      __UpperCamelCase      =	self.to_json_kwargs.pop('path_or_buf'     ,					__UpperCAmelCase				)
      __UpperCamelCase      =	self.to_json_kwargs.pop('orient'     ,					'records'				)
      __UpperCamelCase      =	self.to_json_kwargs.pop('lines'     ,					True if orient == 'records' else False				)
      __UpperCamelCase      =	self.to_json_kwargs.pop('index'     ,					False if orient in ['split', 'table'] else True				)
      __UpperCamelCase      =	self.to_json_kwargs.pop('compression'     ,					__UpperCAmelCase				)

      if compression not in [None, "infer", "gzip", "bz2", "xz"]:
       raise NotImplementedError(F'`datasets` currently does not support {compression} compression'				)

      if isinstance(self.path_or_buf     ,					(str, bytes, os.PathLike)				):
       with fsspec.open(self.path_or_buf     ,					'wb'     ,					compression=__UpperCAmelCase				) as buffer:
        __UpperCamelCase      =	self._write(file_obj=__UpperCAmelCase     ,					orient=__UpperCAmelCase     ,					lines=__UpperCAmelCase     ,					index=__UpperCAmelCase     ,					**self.to_json_kwargs				)
      else:
       if compression:
        raise NotImplementedError(
            F'The compression parameter is not supported when writing to a buffer, but compression={compression}'
            ' was passed. Please provide a local path instead.'				)
       __UpperCamelCase      =	self._write(
           file_obj=self.path_or_buf     ,					orient=__UpperCAmelCase     ,					lines=__UpperCAmelCase     ,					index=__UpperCAmelCase     ,					**self.to_json_kwargs				)
      return written





     def 						UpperCAmelCase (     self     ,					__UpperCAmelCase				):

      '''simple docstring'''




      __UpperCamelCase  ,     __UpperCamelCase  ,     __UpperCamelCase  ,     __UpperCamelCase  ,     __UpperCamelCase      =	args

      __UpperCamelCase      =	query_table(
          table=self.dataset.data     ,					key=slice(__UpperCAmelCase     ,					offset + self.batch_size				)     ,					indices=self.dataset._indices     ,					)
      __UpperCamelCase      =	batch.to_pandas().to_json(
          path_or_buf=__UpperCAmelCase     ,					orient=__UpperCAmelCase     ,					lines=__UpperCAmelCase     ,					index=__UpperCAmelCase     ,					**__UpperCAmelCase				)
      if not json_str.endswith('\n'				):
       json_str += "\n"
      return json_str.encode(self.encoding				)







     def 						UpperCAmelCase (     self     ,					__UpperCAmelCase     ,					__UpperCAmelCase     ,					__UpperCAmelCase     ,					__UpperCAmelCase     ,					**__UpperCAmelCase     ,					):

      '''simple docstring'''




      __UpperCamelCase      =	0

      if self.num_proc is None or self.num_proc == 1:
       for offset in logging.tqdm(
           range(0     ,					len(self.dataset				)     ,					self.batch_size				)     ,					unit='ba'     ,					disable=not logging.is_progress_bar_enabled()     ,					desc='Creating json from Arrow format'     ,					):
        __UpperCamelCase      =	self._batch_json((offset, orient, lines, index, to_json_kwargs)				)
        written += file_obj.write(__UpperCAmelCase				)
      else:
       __UpperCamelCase  ,     __UpperCamelCase      =	len(self.dataset				), self.batch_size
       with multiprocessing.Pool(self.num_proc				) as pool:
        for json_str in logging.tqdm(
            pool.imap(
                self._batch_json     ,					[(offset, orient, lines, index, to_json_kwargs) for offset in range(0     ,					__UpperCAmelCase     ,					__UpperCAmelCase				)]     ,					)     ,					total=(num_rows // batch_size) + 1 if num_rows % batch_size else num_rows // batch_size     ,					unit='ba'     ,					disable=not logging.is_progress_bar_enabled()     ,					desc='Creating json from Arrow format'     ,					):
         written += file_obj.write(__UpperCAmelCase				)

      return written



