'''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,
)


_UpperCAmelCase    :     List[str]         =       pytest.mark.integration





@pytest.mark.parametrize('path'   ,		['paws', 'csv']				)
def  _SCREAMING_SNAKE_CASE		(      __snake_case							:			List[Any]   ,		__snake_case							:			Optional[Any]				):
     inspect_dataset(__snake_case   ,		__snake_case				)
     _A    		=  path + '.py'
     assert script_name in os.listdir(__snake_case				)
     assert "__pycache__" not in os.listdir(__snake_case				)





@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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			Dict   ,		__snake_case							:			Dict				):
     inspect_metric(__snake_case   ,		__snake_case				)
     _A    		=  path + '.py'
     assert script_name in os.listdir(__snake_case				)
     assert "__pycache__" not in os.listdir(__snake_case				)





@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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			Dict   ,		__snake_case							:			Any   ,		__snake_case							:			Union[str, Any]				):
     _A    		=  get_dataset_config_info(__snake_case   ,		config_name=__snake_case				)
     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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			Union[str, Any]   ,		__snake_case							:			List[Any]   ,		__snake_case							:			List[str]				):
     with pytest.raises(__snake_case				):
          get_dataset_config_info(__snake_case   ,		config_name=__snake_case				)





@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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			int   ,		__snake_case							:			Optional[Any]				):
     _A    		=  get_dataset_config_names(__snake_case				)
     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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			Any   ,		__snake_case							:			Optional[Any]   ,		__snake_case							:			Optional[Any]				):
     _A    		=  get_dataset_infos(__snake_case				)
     assert list(infos.keys()				) == expected_configs
     _A    		=  expected_configs[0]
     assert expected_config in infos
     _A    		=  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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			str   ,		__snake_case							:			Optional[int]   ,		__snake_case							:			List[Any]				):
     _A    		=  get_dataset_infos(__snake_case				)
     assert expected_config in infos
     _A    		=  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  _SCREAMING_SNAKE_CASE		(      __snake_case							:			List[Any]   ,		__snake_case							:			Optional[Any]   ,		__snake_case							:			Any				):
     with pytest.raises(__snake_case				):
          get_dataset_split_names(__snake_case   ,		config_name=__snake_case				)




'''simple docstring'''






import pytest
import requests

from datasets.utils.file_utils import http_head

from .utils import OfflineSimulationMode, RequestWouldHangIndefinitelyError, offline





@pytest.mark.integration
def  _SCREAMING_SNAKE_CASE		(      ):
     with offline(OfflineSimulationMode.CONNECTION_TIMES_OUT				):
          with pytest.raises(__snake_case				):
               requests.request('GET'   ,		'https://huggingface.co'				)
          with pytest.raises(requests.exceptions.ConnectTimeout				):
               requests.request('GET'   ,		'https://huggingface.co'   ,		timeout=1.0				)





@pytest.mark.integration
def  _SCREAMING_SNAKE_CASE		(      ):
     with offline(OfflineSimulationMode.CONNECTION_FAILS				):
          with pytest.raises(requests.exceptions.ConnectionError				):
               requests.request('GET'   ,		'https://huggingface.co'				)





def  _SCREAMING_SNAKE_CASE		(      ):
     with offline(OfflineSimulationMode.HF_DATASETS_OFFLINE_SET_TO_1				):
          with pytest.raises(__snake_case				):
               http_head('https://huggingface.co'				)



