


'''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():

__snake_case											=					yaml.safe_load(
    '''\
name: \"\"
allow_empty: false
allow_empty_text: true
subsections:
  - name: \"Dataset Card for X\" # First-level markdown heading
    allow_empty: false
    allow_empty_text: true
    subsections:
      - name: \"Table of Contents\"
        allow_empty: false
        allow_empty_text: false
        subsections: null
      - name: \"Dataset Description\"
        allow_empty: false
        allow_empty_text: false
        subsections:
          - name: \"Dataset Summary\"
            allow_empty: false
            allow_empty_text: false
            subsections: null
          - name: \"Supported Tasks and Leaderboards\"
            allow_empty: true
            allow_empty_text: true
            subsections: null
          - name: Languages
            allow_empty: false
            allow_empty_text: true
            subsections: null
'''
)


__snake_case											=					{
    """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""": []},
                    ],
                },
            ],
        }
    ],
}


__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""


__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
#### Extra Ignored Subsection
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

__snake_case											=					{
    """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""": []},
                    ],
                },
            ],
        }
    ],
}

__snake_case											=					"""\
---
---
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

__snake_case											=					(
    """The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README."""
)

__snake_case											=					"""\
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

__snake_case											=					(
    """The following issues were found for the README at `{path}`:\n-\tNo YAML markers are present in the README."""
)

__snake_case											=					"""\
---
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

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

__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
### Supported Tasks and Leaderboards
### Languages
Language Text
"""
__snake_case											=					"""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)."""


__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
"""
__snake_case											=					"""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'."""

__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Languages
Language Text
"""

__snake_case											=					"""The following issues were found for the README at `{path}`:\n-\tSection `Dataset Description` is missing subsection: `Supported Tasks and Leaderboards`."""


__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
"""

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

__snake_case											=					"""\
---
language:
- zh
- en
---

## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""
__snake_case											=					"""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."""

__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
# Dataset Card My Dataset
"""

__snake_case											=					"""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."""

__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

__snake_case											=					"""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."""

__snake_case											=					""""""

__snake_case											=					"""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."""

__snake_case											=					"""\
---
language:
- zh
- en
---

# Dataset Card for My Dataset
# Dataset Card for My Dataset
## Table of Contents
Some text here.
## Dataset Description
Some text here.
### Dataset Summary
Some text here.
### Supported Tasks and Leaderboards
### Languages
Language Text
"""

__snake_case											=					"""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       a				(      __a			, __a       )  ->     int:



					'''simple docstring'''






					assert ReadMe.from_string(lowerCAmelCase__			, lowerCAmelCase__       ).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       a				(      __a			, __a       )  ->     Union[str, Any]:



					'''simple docstring'''






					with pytest.raises(lowerCAmelCase__			, match=re.escape(expected_error.format(path='''root'''       )       )       ):
										UpperCamelCase__       :Dict							 =			ReadMe.from_string(lowerCAmelCase__			, lowerCAmelCase__       )
										readme.validate()


@pytest.mark.parametrize(
    '''readme_md, expected_error'''			, [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]			, )
def       a				(      __a			, __a       )  ->     List[Any]:



					'''simple docstring'''






					with pytest.raises(lowerCAmelCase__			, match=re.escape(expected_error.format(path='''root'''       )       )       ):
										ReadMe.from_string(lowerCAmelCase__			, lowerCAmelCase__       )


@pytest.mark.parametrize(
    '''readme_md,'''			, [
        (README_MULTIPLE_SAME_HEADING_1),
    ]			, )
def       a				(      __a       )  ->     Optional[Any]:



					'''simple docstring'''






					ReadMe.from_string(lowerCAmelCase__			, lowerCAmelCase__			, suppress_parsing_errors=lowerCAmelCase__       )


@pytest.mark.parametrize(
    '''readme_md, expected_dict'''			, [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ]			, )
def       a				(      __a			, __a       )  ->     Dict:



					'''simple docstring'''






					with tempfile.TemporaryDirectory() as tmp_dir:
										UpperCamelCase__       :Optional[Any]							 =			Path(lowerCAmelCase__       ) / '''README.md'''
										with open(lowerCAmelCase__			, '''w+'''       ) as readme_file:
															readme_file.write(lowerCAmelCase__       )
										UpperCamelCase__       :Tuple							 =			ReadMe.from_readme(lowerCAmelCase__			, lowerCAmelCase__       ).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       a				(      __a			, __a       )  ->     Optional[Any]:



					'''simple docstring'''






					with tempfile.TemporaryDirectory() as tmp_dir:
										UpperCamelCase__       :Optional[int]							 =			Path(lowerCAmelCase__       ) / '''README.md'''
										with open(lowerCAmelCase__			, '''w+'''       ) as readme_file:
															readme_file.write(lowerCAmelCase__       )
										UpperCamelCase__       :List[Any]							 =			expected_error.format(path=lowerCAmelCase__       )
										with pytest.raises(lowerCAmelCase__			, match=re.escape(lowerCAmelCase__       )       ):
															UpperCamelCase__       :int							 =			ReadMe.from_readme(lowerCAmelCase__			, lowerCAmelCase__       )
															readme.validate()


@pytest.mark.parametrize(
    '''readme_md, expected_error'''			, [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]			, )
def       a				(      __a			, __a       )  ->     List[str]:



					'''simple docstring'''






					with tempfile.TemporaryDirectory() as tmp_dir:
										UpperCamelCase__       :Optional[Any]							 =			Path(lowerCAmelCase__       ) / '''README.md'''
										with open(lowerCAmelCase__			, '''w+'''       ) as readme_file:
															readme_file.write(lowerCAmelCase__       )
										UpperCamelCase__       :Optional[Any]							 =			expected_error.format(path=lowerCAmelCase__       )
										with pytest.raises(lowerCAmelCase__			, match=re.escape(lowerCAmelCase__       )       ):
															ReadMe.from_readme(lowerCAmelCase__			, lowerCAmelCase__       )







@pytest.mark.parametrize(
    '''readme_md,'''			, [
        (README_MULTIPLE_SAME_HEADING_1),
    ]			, )
def       a				(      __a       )  ->     Tuple:



					'''simple docstring'''






					with tempfile.TemporaryDirectory() as tmp_dir:
										UpperCamelCase__       :Optional[int]							 =			Path(lowerCAmelCase__       ) / '''README.md'''
										with open(lowerCAmelCase__			, '''w+'''       ) as readme_file:
															readme_file.write(lowerCAmelCase__       )
										ReadMe.from_readme(lowerCAmelCase__			, lowerCAmelCase__			, suppress_parsing_errors=lowerCAmelCase__       )



'''simple docstring'''
# tests directory-specific settings - this file is run automatically
# by pytest before any tests are run

import sys
import warnings
from os.path import abspath, dirname, join


# allow having multiple repository checkouts and not needing to remember to rerun
# 'pip install -e .[dev]' when switching between checkouts and running tests.
__snake_case											=					abspath(join(dirname(dirname(__file__)), '''src'''))
sys.path.insert(1, git_repo_path)

# silence FutureWarning warnings in tests since often we can't act on them until
# they become normal warnings - i.e. the tests still need to test the current functionality
warnings.simplefilter(action='''ignore''', category=FutureWarning)


def       a				(      __a       )  ->     Optional[int]:



					'''simple docstring'''






					from diffusers.utils.testing_utils import pytest_addoption_shared

					pytest_addoption_shared(__a       )







def       a				(      __a       )  ->     str:



					'''simple docstring'''






					from diffusers.utils.testing_utils import pytest_terminal_summary_main

					UpperCamelCase__       :Union[str, Any]							 =			terminalreporter.config.getoption('''--make-reports'''       )
					if make_reports:
										pytest_terminal_summary_main(__a			, id=__a       )