


'''simple docstring'''
import re
import tempfile
from pathlib import Path

import pytest
import yaml

from datasets.utils.readme import ReadMe


# @pytest.fixture
# def example_yaml_structure():

UpperCAmelCase_ :     Optional[int]								=      yaml.safe_load(
    '\\nname: ""\nallow_empty: false\nallow_empty_text: true\nsubsections:\n  - name: "Dataset Card for X" # First-level markdown heading\n    allow_empty: false\n    allow_empty_text: true\n    subsections:\n      - name: "Table of Contents"\n        allow_empty: false\n        allow_empty_text: false\n        subsections: null\n      - name: "Dataset Description"\n        allow_empty: false\n        allow_empty_text: false\n        subsections:\n          - name: "Dataset Summary"\n            allow_empty: false\n            allow_empty_text: false\n            subsections: null\n          - name: "Supported Tasks and Leaderboards"\n            allow_empty: true\n            allow_empty_text: true\n            subsections: null\n          - name: Languages\n            allow_empty: false\n            allow_empty_text: true\n            subsections: null\n'
)


UpperCAmelCase_ :     List[str]								=      {
    'name': 'root',
    'text': '',
    'is_empty_text': True,
    'subsections': [
        {
            'name': 'Dataset Card for My Dataset',
            'text': '',
            'is_empty_text': True,
            'subsections': [
                {'name': 'Table of Contents', 'text': 'Some text here.', 'is_empty_text': False, 'subsections': []},
                {
                    'name': 'Dataset Description',
                    'text': 'Some text here.',
                    'is_empty_text': False,
                    'subsections': [
                        {
                            'name': 'Dataset Summary',
                            'text': 'Some text here.',
                            'is_empty_text': False,
                            'subsections': [],
                        },
                        {
                            'name': 'Supported Tasks and Leaderboards',
                            'text': '',
                            'is_empty_text': True,
                            'subsections': [],
                        },
                        {'name': 'Languages', 'text': 'Language Text', 'is_empty_text': False, 'subsections': []},
                    ],
                },
            ],
        }
    ],
}


UpperCAmelCase_ :     int								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'


UpperCAmelCase_ :     str								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n#### Extra Ignored Subsection\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     int								=      {
    'name': 'root',
    'text': '',
    'is_empty_text': True,
    'subsections': [
        {
            'name': 'Dataset Card for My Dataset',
            'text': '',
            'is_empty_text': True,
            'subsections': [
                {'name': 'Table of Contents', 'text': 'Some text here.', 'is_empty_text': False, 'subsections': []},
                {
                    'name': 'Dataset Description',
                    'text': 'Some text here.',
                    'is_empty_text': False,
                    'subsections': [
                        {
                            'name': 'Dataset Summary',
                            'text': 'Some text here.',
                            'is_empty_text': False,
                            'subsections': [
                                {
                                    'name': 'Extra Ignored Subsection',
                                    'text': '',
                                    'is_empty_text': True,
                                    'subsections': [],
                                }
                            ],
                        },
                        {
                            'name': 'Supported Tasks and Leaderboards',
                            'text': '',
                            'is_empty_text': True,
                            'subsections': [],
                        },
                        {'name': 'Languages', 'text': 'Language Text', 'is_empty_text': False, 'subsections': []},
                    ],
                },
            ],
        }
    ],
}

UpperCAmelCase_ :     Dict								=      '\\n---\n---\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     List[str]								=      (
    'The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README.'
)

UpperCAmelCase_ :     Tuple								=      '\\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     Optional[int]								=      (
    'The following issues were found for the README at `{path}`:\n-\tNo YAML markers are present in the README.'
)

UpperCAmelCase_ :     Optional[Any]								=      '\\n---\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     Dict								=      'The following issues were found for the README at `{path}`:\n-\tOnly the start of YAML tags present in the README.'

UpperCAmelCase_ :     Dict								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'
UpperCAmelCase_ :     Union[str, Any]								=      'The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Summary` but it is empty.\n-\tExpected some text in section `Dataset Summary` but it is empty (text in subsections are ignored).'


UpperCAmelCase_ :     Union[str, Any]								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n'
UpperCAmelCase_ :     Optional[int]								=      'The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Card for My Dataset` but it is empty.\n-\tSection `Dataset Card for My Dataset` expected the following subsections: `Table of Contents`, `Dataset Description`. Found \'None\'.'

UpperCAmelCase_ :     List[Any]								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     Union[str, Any]								=      'The following issues were found for the README at `{path}`:\n-\tSection `Dataset Description` is missing subsection: `Supported Tasks and Leaderboards`.'


UpperCAmelCase_ :     List[str]								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\n'

UpperCAmelCase_ :     Dict								=      'The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Languages` but it is empty.'

UpperCAmelCase_ :     Optional[Any]								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'
UpperCAmelCase_ :     List[Any]								=      'The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README.'

UpperCAmelCase_ :     Any								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n# Dataset Card My Dataset\n'

UpperCAmelCase_ :     int								=      'The following issues were found for the README at `{path}`:\n-\tThe README has several first-level headings: `Dataset Card for My Dataset`, `Dataset Card My Dataset`. Only one heading is expected. Skipping further validation for this README.'

UpperCAmelCase_ :     int								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     List[str]								=      'The following issues were found for the README at `{path}`:\n-\tNo first-level heading starting with `Dataset Card for` found in README. Skipping further validation for this README.'

UpperCAmelCase_ :     Dict								=      ''

UpperCAmelCase_ :     Optional[int]								=      'The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README.\n-\tNo YAML markers are present in the README.'

UpperCAmelCase_ :     Union[str, Any]								=      '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_ :     Dict								=      'The following issues were found while parsing the README at `{path}`:\n-\tMultiple sections with the same heading `Dataset Card for My Dataset` have been found. Please keep only one of these sections.'


@pytest.mark.parametrize(
    'readme_md, expected_dict',				[
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					Optional[Any],				UpperCamelCase__       :					List[Any]			):
 assert ReadMe.from_string(UpperCamelCase__,				UpperCamelCase__			).to_dict() == expected_dict


@pytest.mark.parametrize(
    'readme_md, expected_error',				[
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					Any,				UpperCamelCase__       :					Union[str, Any]			):
 with pytest.raises(UpperCamelCase__,				match=re.escape(expected_error.format(path='root'			)			)			):
  __A  :		Any       =			ReadMe.from_string(UpperCamelCase__,				UpperCamelCase__			)
  readme.validate()


@pytest.mark.parametrize(
    'readme_md, expected_error',				[
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					str,				UpperCamelCase__       :					Tuple			):
 with pytest.raises(UpperCamelCase__,				match=re.escape(expected_error.format(path='root'			)			)			):
  ReadMe.from_string(UpperCamelCase__,				UpperCamelCase__			)


@pytest.mark.parametrize(
    'readme_md,',				[
        (README_MULTIPLE_SAME_HEADING_1),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					str			):
 ReadMe.from_string(UpperCamelCase__,				UpperCamelCase__,				suppress_parsing_errors=UpperCamelCase__			)


@pytest.mark.parametrize(
    'readme_md, expected_dict',				[
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					int,				UpperCamelCase__       :					str			):
 with tempfile.TemporaryDirectory() as tmp_dir:
  __A  :		Union[str, Any]       =			Path(UpperCamelCase__			) / 'README.md'
  with open(UpperCamelCase__,				'w+'			) as readme_file:
   readme_file.write(UpperCamelCase__			)
  __A  :		Optional[int]       =			ReadMe.from_readme(UpperCamelCase__,				UpperCamelCase__			).to_dict()
  assert out["name"] == path
  assert out["text"] == ""
  assert out["is_empty_text"]
  assert out["subsections"] == expected_dict["subsections"]


@pytest.mark.parametrize(
    'readme_md, expected_error',				[
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					List[Any],				UpperCamelCase__       :					str			):
 with tempfile.TemporaryDirectory() as tmp_dir:
  __A  :		Dict       =			Path(UpperCamelCase__			) / 'README.md'
  with open(UpperCamelCase__,				'w+'			) as readme_file:
   readme_file.write(UpperCamelCase__			)
  __A  :		str       =			expected_error.format(path=UpperCamelCase__			)
  with pytest.raises(UpperCamelCase__,				match=re.escape(UpperCamelCase__			)			):
   __A  :		List[str]       =			ReadMe.from_readme(UpperCamelCase__,				UpperCamelCase__			)
   readme.validate()


@pytest.mark.parametrize(
    'readme_md, expected_error',				[
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					List[str],				UpperCamelCase__       :					Union[str, Any]			):
 with tempfile.TemporaryDirectory() as tmp_dir:
  __A  :		str       =			Path(UpperCamelCase__			) / 'README.md'
  with open(UpperCamelCase__,				'w+'			) as readme_file:
   readme_file.write(UpperCamelCase__			)
  __A  :		int       =			expected_error.format(path=UpperCamelCase__			)
  with pytest.raises(UpperCamelCase__,				match=re.escape(UpperCamelCase__			)			):
   ReadMe.from_readme(UpperCamelCase__,				UpperCamelCase__			)


@pytest.mark.parametrize(
    'readme_md,',				[
        (README_MULTIPLE_SAME_HEADING_1),
    ],				)
def 			_lowercase       (				UpperCamelCase__       :					Optional[Any]			):
 with tempfile.TemporaryDirectory() as tmp_dir:
  __A  :		Optional[Any]       =			Path(UpperCamelCase__			) / 'README.md'
  with open(UpperCamelCase__,				'w+'			) as readme_file:
   readme_file.write(UpperCamelCase__			)
  ReadMe.from_readme(UpperCamelCase__,				UpperCamelCase__,				suppress_parsing_errors=UpperCamelCase__			)










'''simple docstring'''
import unittest

from datasets import load_dataset

from transformers.pipelines import pipeline
from transformers.testing_utils import is_pipeline_test, nested_simplify, require_torch, slow

@is_pipeline_test
@require_torch
class 						_lowerCamelCase					(    unittest.TestCase				):




 '''simple docstring'''


 @require_torch
 def  snake_case__   (      self						):

  """simple docstring"""







  __A  :		Optional[int]       =			pipeline(
      task='zero-shot-audio-classification'      ,      model='hf-internal-testing/tiny-clap-htsat-unfused'						)
  __A  :		str       =			load_dataset('ashraq/esc50'						)
  __A  :		Union[str, Any]       =			dataset['train']['audio'][-1]['array']
  __A  :		List[str]       =			audio_classifier(__lowercase      ,      candidate_labels=['Sound of a dog', 'Sound of vaccum cleaner']						)
  self.assertEqual(
      nested_simplify(__lowercase						)      ,      [{'score': 0.5_0_1, 'label': 'Sound of a dog'}, {'score': 0.4_9_9, 'label': 'Sound of vaccum cleaner'}]      ,      )


 @unittest.skip('No models are available in TF'						)
 def  snake_case__   (      self						):

  """simple docstring"""







  pass


 @slow
 @require_torch
 def  snake_case__   (      self						):

  """simple docstring"""







  __A  :		Dict       =			pipeline(
      task='zero-shot-audio-classification'      ,      model='laion/clap-htsat-unfused'      ,      )
  # This is an audio of a dog
  __A  :		Optional[Any]       =			load_dataset('ashraq/esc50'						)
  __A  :		List[str]       =			dataset['train']['audio'][-1]['array']
  __A  :		Union[str, Any]       =			audio_classifier(__lowercase      ,      candidate_labels=['Sound of a dog', 'Sound of vaccum cleaner']						)

  self.assertEqual(
      nested_simplify(__lowercase						)      ,      [
          {'score': 0.9_9_9, 'label': 'Sound of a dog'},
          {'score': 0.0_0_1, 'label': 'Sound of vaccum cleaner'},
      ]      ,      )

  __A  :		Any       =			audio_classifier([audio] * 5      ,      candidate_labels=['Sound of a dog', 'Sound of vaccum cleaner']						)
  self.assertEqual(
      nested_simplify(__lowercase						)      ,      [
          [
              {'score': 0.9_9_9, 'label': 'Sound of a dog'},
              {'score': 0.0_0_1, 'label': 'Sound of vaccum cleaner'},
          ],
      ]
      * 5      ,      )
  __A  :		str       =			audio_classifier(
      [audio] * 5      ,      candidate_labels=['Sound of a dog', 'Sound of vaccum cleaner']      ,      batch_size=5						)
  self.assertEqual(
      nested_simplify(__lowercase						)      ,      [
          [
              {'score': 0.9_9_9, 'label': 'Sound of a dog'},
              {'score': 0.0_0_1, 'label': 'Sound of vaccum cleaner'},
          ],
      ]
      * 5      ,      )







 @unittest.skip('No models are available in TF'						)
 def  snake_case__   (      self						):

  """simple docstring"""







  pass






