'''simple docstring'''





import os

import pytest

from datasets import (
    get_dataset_config_info,
    get_dataset_config_names,
    get_dataset_infos,
    get_dataset_split_names,
    inspect_dataset,
    inspect_metric,
)


_lowerCAmelCase  :Optional[int]  				=    pytest.mark.integration


@pytest.mark.parametrize('path'       ,      ['paws', 'csv']		)
def     __lowerCAmelCase						(       a_       ,      a_		)      ->	List[Any]:





					'''simple docstring'''





					inspect_dataset(a_       ,      a_		)
					SCREAMING_SNAKE_CASE					:			str    	=       path + '.py'
					assert script_name in os.listdir(a_		)
					assert "__pycache__" not in os.listdir(a_		)


@pytest.mark.filterwarnings('ignore:inspect_metric is deprecated:FutureWarning'		)
@pytest.mark.filterwarnings('ignore:metric_module_factory is deprecated:FutureWarning'		)
@pytest.mark.parametrize('path'       ,      ['accuracy']		)
def     __lowerCAmelCase						(       a_       ,      a_		)      ->	Tuple:





					'''simple docstring'''





					inspect_metric(a_       ,      a_		)
					SCREAMING_SNAKE_CASE					:			List[str]    	=       path + '.py'
					assert script_name in os.listdir(a_		)
					assert "__pycache__" not in os.listdir(a_		)


@pytest.mark.parametrize(
    'path, config_name, expected_splits'       ,      [
        ('squad', 'plain_text', ['train', 'validation']),
        ('dalle-mini/wit', 'dalle-mini--wit', ['train']),
        ('paws', 'labeled_final', ['train', 'test', 'validation']),
    ]       ,      )
def     __lowerCAmelCase						(       a_       ,      a_       ,      a_		)      ->	Optional[Any]:





					'''simple docstring'''





					SCREAMING_SNAKE_CASE					:			Tuple    	=       get_dataset_config_info(a_       ,      config_name=a_		)
					assert info.config_name == config_name
					assert list(info.splits.keys()		) == expected_splits


@pytest.mark.parametrize(
    'path, config_name, expected_exception'       ,      [
        ('paws', None, ValueError),
    ]       ,      )
def     __lowerCAmelCase						(       a_       ,      a_       ,      a_		)      ->	str:





					'''simple docstring'''





					with pytest.raises(a_		):
										get_dataset_config_info(a_       ,      config_name=a_		)


@pytest.mark.parametrize(
    'path, expected'       ,      [
        ('squad', 'plain_text'),
        ('acronym_identification', 'default'),
        ('lhoestq/squad', 'plain_text'),
        ('lhoestq/test', 'default'),
        ('lhoestq/demo1', 'lhoestq--demo1'),
        ('dalle-mini/wit', 'dalle-mini--wit'),
    ]       ,      )
def     __lowerCAmelCase						(       a_       ,      a_		)      ->	int:





					'''simple docstring'''





					SCREAMING_SNAKE_CASE					:			Tuple    	=       get_dataset_config_names(a_		)
					assert expected in config_names


@pytest.mark.parametrize(
    'path, expected_configs, expected_splits_in_first_config'       ,      [
        ('squad', ['plain_text'], ['train', 'validation']),
        ('dalle-mini/wit', ['dalle-mini--wit'], ['train']),
        ('paws', ['labeled_final', 'labeled_swap', 'unlabeled_final'], ['train', 'test', 'validation']),
    ]       ,      )
def     __lowerCAmelCase						(       a_       ,      a_       ,      a_		)      ->	Any:





					'''simple docstring'''





					SCREAMING_SNAKE_CASE					:			int    	=       get_dataset_infos(a_		)
					assert list(infos.keys()		) == expected_configs
					SCREAMING_SNAKE_CASE					:			Tuple    	=       expected_configs[0]
					assert expected_config in infos
					SCREAMING_SNAKE_CASE					:			List[Any]    	=       infos[expected_config]
					assert info.config_name == expected_config
					assert list(info.splits.keys()		) == expected_splits_in_first_config


@pytest.mark.parametrize(
    'path, expected_config, expected_splits'       ,      [
        ('squad', 'plain_text', ['train', 'validation']),
        ('dalle-mini/wit', 'dalle-mini--wit', ['train']),
        ('paws', 'labeled_final', ['train', 'test', 'validation']),
    ]       ,      )
def     __lowerCAmelCase						(       a_       ,      a_       ,      a_		)      ->	Optional[int]:





					'''simple docstring'''





					SCREAMING_SNAKE_CASE					:			Optional[int]    	=       get_dataset_infos(a_		)
					assert expected_config in infos
					SCREAMING_SNAKE_CASE					:			Union[str, Any]    	=       infos[expected_config]
					assert info.config_name == expected_config
					assert list(info.splits.keys()		) == expected_splits


@pytest.mark.parametrize(
    'path, config_name, expected_exception'       ,      [
        ('paws', None, ValueError),
    ]       ,      )
def     __lowerCAmelCase						(       a_       ,      a_       ,      a_		)      ->	List[str]:





					'''simple docstring'''





					with pytest.raises(a_		):
										get_dataset_split_names(a_       ,      config_name=a_		)




'''simple docstring'''





import json
import os
import unittest

from transformers import CLIPTokenizer, CLIPTokenizerFast
from transformers.models.clip.tokenization_clip import VOCAB_FILES_NAMES
from transformers.testing_utils import require_ftfy, require_tokenizers

from ...test_tokenization_common import TokenizerTesterMixin






@require_tokenizers
class 				UpperCAmelCase (       _SCREAMING_SNAKE_CASE   ,    unittest.TestCase ):




    '''simple docstring'''







    snake_case__							:   int							=						CLIPTokenizer
    snake_case__							:   Union[str, Any]							=						CLIPTokenizerFast
    snake_case__							:   str							=						True
    snake_case__							:   Optional[int]							=						{}
    snake_case__							:   int							=						False






    def       _UpperCamelCase    (       self )							-> Tuple:
         super().setUp()

         # fmt: off
         SCREAMING_SNAKE_CASE					:			Optional[int]    	=       ['l', 'o', 'w', 'e', 'r', 's', 't', 'i', 'd', 'n', 'lo', 'l</w>', 'w</w>', 'r</w>', 't</w>', 'low</w>', 'er</w>', 'lowest</w>', 'newer</w>', 'wider', '<unk>', '<|startoftext|>', '<|endoftext|>']
         # fmt: on
         SCREAMING_SNAKE_CASE					:			int    	=       dict(zip(lowercase__    ,							range(len(lowercase__ ) ) ) )
         SCREAMING_SNAKE_CASE					:			Dict    	=       ['#version: 0.2', 'l o', 'lo w</w>', 'e r</w>']
         SCREAMING_SNAKE_CASE					:			List[str]    	=       {'unk_token': '<unk>'}

         SCREAMING_SNAKE_CASE					:			Optional[Any]    	=       os.path.join(self.tmpdirname    ,							VOCAB_FILES_NAMES['vocab_file'] )
         SCREAMING_SNAKE_CASE					:			Union[str, Any]    	=       os.path.join(self.tmpdirname    ,							VOCAB_FILES_NAMES['merges_file'] )
         with open(self.vocab_file    ,							'w'    ,							encoding='utf-8' ) as fp:
              fp.write(json.dumps(lowercase__ ) + '\n' )
         with open(self.merges_file    ,							'w'    ,							encoding='utf-8' ) as fp:
              fp.write('\n'.join(lowercase__ ) )






    def       _UpperCamelCase    (       self    ,							**lowercase__ )							-> Dict:
         kwargs.update(self.special_tokens_map )
         return CLIPTokenizer.from_pretrained(self.tmpdirname    ,							**lowercase__ )






    def       _UpperCamelCase    (       self    ,							**lowercase__ )							-> Union[str, Any]:
         kwargs.update(self.special_tokens_map )
         return CLIPTokenizerFast.from_pretrained(self.tmpdirname    ,							**lowercase__ )






    def       _UpperCamelCase    (       self    ,							lowercase__ )							-> List[Any]:
         SCREAMING_SNAKE_CASE					:			Optional[Any]    	=       'lower newer'
         SCREAMING_SNAKE_CASE					:			Dict    	=       'lower newer'
         return input_text, output_text






    def       _UpperCamelCase    (       self )							-> List[Any]:
         SCREAMING_SNAKE_CASE					:			str    	=       CLIPTokenizer(self.vocab_file    ,							self.merges_file    ,							**self.special_tokens_map )
         SCREAMING_SNAKE_CASE					:			Optional[int]    	=       'lower newer'
         SCREAMING_SNAKE_CASE					:			List[Any]    	=       ['lo', 'w', 'er</w>', 'n', 'e', 'w', 'er</w>']
         SCREAMING_SNAKE_CASE					:			Optional[Any]    	=       tokenizer.tokenize(lowercase__ )
         self.assertListEqual(lowercase__    ,							lowercase__ )

         SCREAMING_SNAKE_CASE					:			Any    	=       tokens + [tokenizer.unk_token]
         SCREAMING_SNAKE_CASE					:			int    	=       [10, 2, 16, 9, 3, 2, 16, 20]
         self.assertListEqual(tokenizer.convert_tokens_to_ids(lowercase__ )    ,							lowercase__ )






    @require_ftfy
    def       _UpperCamelCase    (       self )							-> List[str]:
         for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
              with self.subTest(F"""{tokenizer.__class__.__name__} ({pretrained_name})""" ):
                   SCREAMING_SNAKE_CASE					:			Dict    	=       self.tokenizer_class.from_pretrained(lowercase__    ,							**lowercase__ )
                   SCREAMING_SNAKE_CASE					:			List[Any]    	=       self.rust_tokenizer_class.from_pretrained(lowercase__    ,							**lowercase__ )

                   SCREAMING_SNAKE_CASE					:			Tuple    	=       'A\n\'ll 11p223RF☆ho!!to?\'d\'d\'\'d of a cat to-$\'\'d.'
                   SCREAMING_SNAKE_CASE					:			List[str]    	=       tokenizer_s.tokenize(lowercase__ )
                   SCREAMING_SNAKE_CASE					:			List[str]    	=       tokenizer_r.tokenize(lowercase__ )

                   self.assertListEqual(lowercase__    ,							lowercase__ )

                   # Test that the tokenization is identical on an example containing a character (Latin Small Letter A
                   # with Tilde) encoded in 2 different ways
                   SCREAMING_SNAKE_CASE					:			Tuple    	=       'xa\u0303y' + ' ' + 'x\xe3y'
                   SCREAMING_SNAKE_CASE					:			Optional[Any]    	=       tokenizer_s.tokenize(lowercase__ )
                   SCREAMING_SNAKE_CASE					:			Dict    	=       tokenizer_r.tokenize(lowercase__ )

                   self.assertListEqual(lowercase__    ,							lowercase__ )

                   # Test that the tokenization is identical on unicode of space type
                   SCREAMING_SNAKE_CASE					:			Union[str, Any]    	=       [
                       '\u0009',  # (horizontal tab, '\t')
                       '\u000B',  # (vertical tab)
                       '\u000C',  # (form feed)
                       '\u0020',  # (space, ' ')
                       '\u200E',  # (left-to-right mark):w
                       '\u200F',  # (right-to-left mark)
                   ]
                   for unicode_seq in spaces_unicodes:
                        SCREAMING_SNAKE_CASE					:			Tuple    	=       tokenizer_s.tokenize(lowercase__ )
                        SCREAMING_SNAKE_CASE					:			Optional[int]    	=       tokenizer_r.tokenize(lowercase__ )

                        self.assertListEqual(lowercase__    ,							lowercase__ )

                   # Test that the tokenization is identical on unicode of line break type
                   SCREAMING_SNAKE_CASE					:			int    	=       [
                       '\u000A',  # (line feed, '\n')
                       '\r\n',  # (carriage return and line feed, '\r\n')
                       '\u000D',  # (carriage return, '\r')
                       '\r',  # (carriage return, '\r')
                       '\u000D',  # (carriage return, '\r')
                       '\u2028',  # (line separator)
                       '\u2029',  # (paragraph separator)
                       # "\u0085", # (next line)
                   ]

                   # The tokenization is not identical for the character "\u0085" (next line). The slow version using ftfy transforms
                   # it into the Horizontal Ellipsis character "…" ("\u2026") while the fast version transforms it into a
                   # space (and thus into an empty list).

                   for unicode_seq in line_break_unicodes:
                        SCREAMING_SNAKE_CASE					:			List[Any]    	=       tokenizer_s.tokenize(lowercase__ )
                        SCREAMING_SNAKE_CASE					:			int    	=       tokenizer_r.tokenize(lowercase__ )

                        self.assertListEqual(lowercase__    ,							lowercase__ )






    def       _UpperCamelCase    (       self )							-> int:
         # Test which aims to verify that the offsets are well adapted to the argument `add_prefix_space`
         for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
              with self.subTest(F"""{tokenizer.__class__.__name__} ({pretrained_name})""" ):
                   SCREAMING_SNAKE_CASE					:			str    	=       'hello'  # `hello` is a token in the vocabulary of `pretrained_name`
                   SCREAMING_SNAKE_CASE					:			Union[str, Any]    	=       F"""{text_of_1_token} {text_of_1_token}"""

                   SCREAMING_SNAKE_CASE					:			List[str]    	=       self.rust_tokenizer_class.from_pretrained(
                       lowercase__    ,							use_fast=lowercase__    ,							)
                   SCREAMING_SNAKE_CASE					:			Tuple    	=       tokenizer_r(lowercase__    ,							return_offsets_mapping=lowercase__    ,							add_special_tokens=lowercase__ )
                   self.assertEqual(encoding.offset_mapping[0]    ,							(0, len(lowercase__ )) )
                   self.assertEqual(
                       encoding.offset_mapping[1]    ,							(len(lowercase__ ) + 1, len(lowercase__ ) + 1 + len(lowercase__ ))    ,							)

                   SCREAMING_SNAKE_CASE					:			Optional[int]    	=       F""" {text}"""

                   SCREAMING_SNAKE_CASE					:			Any    	=       self.rust_tokenizer_class.from_pretrained(
                       lowercase__    ,							use_fast=lowercase__    ,							)
                   SCREAMING_SNAKE_CASE					:			Optional[Any]    	=       tokenizer_r(lowercase__    ,							return_offsets_mapping=lowercase__    ,							add_special_tokens=lowercase__ )
                   self.assertEqual(encoding.offset_mapping[0]    ,							(1, 1 + len(lowercase__ )) )
                   self.assertEqual(
                       encoding.offset_mapping[1]    ,							(1 + len(lowercase__ ) + 1, 1 + len(lowercase__ ) + 1 + len(lowercase__ ))    ,							)






    def       _UpperCamelCase    (       self )							-> int:
         # Test related to the breaking change introduced in transformers v4.17.0
         # We need to check that an error in raised when the user try to load a previous version of the tokenizer.
         with self.assertRaises(lowercase__ ) as context:
              self.rust_tokenizer_class.from_pretrained('robot-test/old-clip-tokenizer' )

         self.assertTrue(
             context.exception.args[0].startswith(
                 'The `backend_tokenizer` provided does not match the expected format.' ) )






    @require_ftfy
    def       _UpperCamelCase    (       self )							-> Union[str, Any]:
         super().test_tokenization_python_rust_equals()






    def       _UpperCamelCase    (       self )							-> int:
         # CLIP always lower cases letters
         pass



