


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_      : Any  					=       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_      : Optional[Any]  					=       {
    '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_      : 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\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'


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\nSome text here.\n#### Extra Ignored Subsection\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n'

UpperCAmelCase_      : Union[str, Any]  					=       {
    '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_      : int  					=       '\\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_      : Dict  					=       (
    'The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README.'
)

UpperCAmelCase_      : Dict  					=       '\\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_      : int  					=       '\\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[Any]  					=       'The following issues were found for the README at `{path}`:\n-\tOnly the start of YAML tags present in the README.'

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\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_      : List[str]  					=       '\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n'
UpperCAmelCase_      : Union[str, Any]  					=       '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[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### Languages\nLanguage Text\n'

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


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\n'

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

UpperCAmelCase_      : Dict  					=       '\\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_      : 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.'

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\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n# Dataset Card My Dataset\n'

UpperCAmelCase_      : Any  					=       '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_      : Any  					=       '\\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_      : Optional[Any]  					=       '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_      : Optional[int]  					=       ''

UpperCAmelCase_      : Union[str, 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.\n-\tNo YAML markers are present in the README.'

UpperCAmelCase_      : 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_      : List[Any]  					=       '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        SCREAMING_SNAKE_CASE_						(			__A :			int  ,    __A :			Optional[int]				)	->		Any:

       """simple docstring"""

       assert ReadMe.from_string(__A  ,    __A				).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        SCREAMING_SNAKE_CASE_						(			__A :			Optional[int]  ,    __A :			Union[str, Any]				)	->		int:

       """simple docstring"""

       with pytest.raises(__A  ,    match=re.escape(expected_error.format(path='root'				)				)				):
              a_					:		List[str]							=  ReadMe.from_string(__A  ,    __A				)
              readme.validate()







@pytest.mark.parametrize(
    'readme_md, expected_error'  ,    [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]  ,    )
def        SCREAMING_SNAKE_CASE_						(			__A :			Any  ,    __A :			Optional[int]				)	->		Dict:

       """simple docstring"""

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







@pytest.mark.parametrize(
    'readme_md,'  ,    [
        (README_MULTIPLE_SAME_HEADING_1),
    ]  ,    )
def        SCREAMING_SNAKE_CASE_						(			__A :			str				)	->		Optional[Any]:

       """simple docstring"""

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







@pytest.mark.parametrize(
    'readme_md, expected_dict'  ,    [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ]  ,    )
def        SCREAMING_SNAKE_CASE_						(			__A :			int  ,    __A :			List[str]				)	->		Any:

       """simple docstring"""

       with tempfile.TemporaryDirectory() as tmp_dir:
              a_					:		Any							=  Path(__A				) / 'README.md'
              with open(__A  ,    'w+'				) as readme_file:
                     readme_file.write(__A				)
              a_					:		List[str]							=  ReadMe.from_readme(__A  ,    __A				).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        SCREAMING_SNAKE_CASE_						(			__A :			Dict  ,    __A :			str				)	->		Optional[Any]:

       """simple docstring"""

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







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

       """simple docstring"""

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







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

       """simple docstring"""

       with tempfile.TemporaryDirectory() as tmp_dir:
              a_					:		Tuple							=  Path(__A				) / 'README.md'
              with open(__A  ,    'w+'				) as readme_file:
                     readme_file.write(__A				)
              ReadMe.from_readme(__A  ,    __A  ,    suppress_parsing_errors=__A				)









"""simple docstring"""




import heapq as hq
import math
from collections.abc import Iterator


class        __UpperCamelCase				:






							def __init__(   self      ,					lowerCAmelCase__      )	-> Optional[int]:
								a      :		List[str]	=				str(id_      )
								a      :		Optional[Any]	=				None
								a      :		Tuple	=				None
								a      :		str	=				[]
								a      :		Any	=				{}  # {vertex:distance}






							def __lt__(   self      ,					lowerCAmelCase__      )	-> Any:
								return self.key < other.key






							def __repr__(   self      )	-> Optional[Any]:
								return self.id






							def     __a   (   self      ,					lowerCAmelCase__      )	-> Any:
								self.neighbors.append(lowerCAmelCase__      )






							def     __a   (   self      ,					lowerCAmelCase__      ,					lowerCAmelCase__      )	-> Tuple:
								a      :		Optional[Any]	=				weight






def       _SCREAMING_SNAKE_CASE		(	_lowercase						:    int    ,       _lowercase						:    int    ,       _lowercase						:    int    ,       _lowercase						:    Union[str, Any]					) ->str:






	'''simple docstring'''






	graph[a - 1].add_neighbor(graph[b - 1]					)
	graph[b - 1].add_neighbor(graph[a - 1]					)
	# add the edges:
	graph[a - 1].add_edge(graph[b - 1]    ,       _lowercase					)
	graph[b - 1].add_edge(graph[a - 1]    ,       _lowercase					)






def       _SCREAMING_SNAKE_CASE		(	_lowercase						:    list    ,       _lowercase						:    Vertex					) ->list:






	'''simple docstring'''






	a      :		int	=				[]
	for u in graph:
		a      :		List[str]	=				math.inf
		a      :		int	=				None
	a      :		str	=				0
	a      :		Union[str, Any]	=				graph[:]
	while q:
		a      :		List[Any]	=				min(_lowercase					)
		q.remove(_lowercase					)
		for v in u.neighbors:
			if (v in q) and (u.edges[v.id] < v.key):
				a      :		List[Any]	=				u
				a      :		Optional[int]	=				u.edges[v.id]
	for i in range(1    ,       len(_lowercase					)					):
		a.append((int(graph[i].id					) + 1, int(graph[i].pi.id					) + 1)					)
	return a






def       _SCREAMING_SNAKE_CASE		(	_lowercase						:    list    ,       _lowercase						:    Vertex					) ->Iterator[tuple]:






	'''simple docstring'''






	for u in graph:
		a      :		str	=				math.inf
		a      :		Dict	=				None
	a      :		Dict	=				0

	a      :		List[Any]	=				list(_lowercase					)
	hq.heapify(_lowercase					)

	while h:
		a      :		Dict	=				hq.heappop(_lowercase					)
		for v in u.neighbors:
			if (v in h) and (u.edges[v.id] < v.key):
				a      :		Dict	=				u
				a      :		Optional[Any]	=				u.edges[v.id]
				hq.heapify(_lowercase					)

	for i in range(1    ,       len(_lowercase					)					):
		yield (int(graph[i].id					) + 1, int(graph[i].pi.id					) + 1)






def       _SCREAMING_SNAKE_CASE		(	) ->None:






	'''simple docstring'''








if __name__ == "__main__":
		import doctest

		doctest.testmod()


