



"""simple docstring"""
import textwrap

import pyarrow as pa
import pytest

from datasets import Features, Value
from datasets.packaged_modules.json.json import Json
@pytest.fixture
def   A_    (     _lowercase      ):







						'''simple docstring'''


						snake_case_					:int		=     tmp_path / """file.jsonl"""
						snake_case_					:Any		=     textwrap.dedent(
						    """\
        {\"col_1\": -1}
        {\"col_1\": 1, \"col_2\": 2}
        {\"col_1\": 10, \"col_2\": 20}
        """      )
						with open(_lowercase,   """w"""      ) as f:
												f.write(_lowercase      )
						return str(_lowercase      )
@pytest.fixture
def   A_    (     _lowercase      ):







						'''simple docstring'''


						snake_case_					:List[Any]		=     tmp_path / """file_utf16_encoded.jsonl"""
						snake_case_					:str		=     textwrap.dedent(
						    """\
        {\"col_1\": -1}
        {\"col_1\": 1, \"col_2\": 2}
        {\"col_1\": 10, \"col_2\": 20}
        """      )
						with open(_lowercase,   """w""",   encoding="""utf-16"""      ) as f:
												f.write(_lowercase      )
						return str(_lowercase      )
@pytest.fixture
def   A_    (     _lowercase      ):







						'''simple docstring'''


						snake_case_					:Union[str, Any]		=     tmp_path / """file_with_list_of_dicts.json"""
						snake_case_					:Dict		=     textwrap.dedent(
						    """\
        [
            {\"col_1\": -1},
            {\"col_1\": 1, \"col_2\": 2},
            {\"col_1\": 10, \"col_2\": 20}
        ]
        """      )
						with open(_lowercase,   """w"""      ) as f:
												f.write(_lowercase      )
						return str(_lowercase      )
@pytest.fixture
def   A_    (     _lowercase      ):







						'''simple docstring'''


						snake_case_					:List[Any]		=     tmp_path / """file_with_list_of_dicts_field.json"""
						snake_case_					:Any		=     textwrap.dedent(
						    """\
        {
            \"field1\": 1,
            \"field2\": \"aabb\",
            \"field3\": [
                {\"col_1\": -1},
                {\"col_1\": 1, \"col_2\": 2},
                {\"col_1\": 10, \"col_2\": 20}
            ]
        }
        """      )
						with open(_lowercase,   """w"""      ) as f:
												f.write(_lowercase      )
						return str(_lowercase      )
@pytest.mark.parametrize(
    """file_fixture, config_kwargs""",   [
        ("""jsonl_file""", {}),
        ("""jsonl_file_utf16_encoded""", {"""encoding""": """utf-16"""}),
        ("""json_file_with_list_of_dicts""", {}),
        ("""json_file_with_list_of_dicts_field""", {"""field""": """field3"""}),
    ],   )
def   A_    (     _lowercase,   _lowercase,   _lowercase      ):







						'''simple docstring'''


						snake_case_					:Optional[Any]		=     Json(**_lowercase      )
						snake_case_					:int		=     json._generate_tables([[request.getfixturevalue(_lowercase      )]]      )
						snake_case_					:Optional[int]		=     pa.concat_tables([table for _, table in generator]      )
						assert pa_table.to_pydict() == {"col_1": [-1, 1, 10], "col_2": [None, 2, 20]}

@pytest.mark.parametrize(
    """file_fixture, config_kwargs""",   [
        (
            """jsonl_file""",
            {"""features""": Features({"""col_1""": Value("""int64"""      ), """col_2""": Value("""int64"""      ), """missing_col""": Value("""string"""      )}      )},
        ),
        (
            """json_file_with_list_of_dicts""",
            {"""features""": Features({"""col_1""": Value("""int64"""      ), """col_2""": Value("""int64"""      ), """missing_col""": Value("""string"""      )}      )},
        ),
        (
            """json_file_with_list_of_dicts_field""",
            {
                """field""": """field3""",
                """features""": Features(
                    {"""col_1""": Value("""int64"""      ), """col_2""": Value("""int64"""      ), """missing_col""": Value("""string"""      )}      ),
            },
        ),
    ],   )
def   A_    (     _lowercase,   _lowercase,   _lowercase      ):







						'''simple docstring'''


						snake_case_					:Dict		=     Json(**_lowercase      )
						snake_case_					:List[Any]		=     json._generate_tables([[request.getfixturevalue(_lowercase      )]]      )
						snake_case_					:Optional[Any]		=     pa.concat_tables([table for _, table in generator]      )
						assert pa_table.to_pydict() == {"col_1": [-1, 1, 10], "col_2": [None, 2, 20], "missing_col": [None, None, None]}



import math
import os

import cva
import numpy as np

lowercase_									=    255.0
def 		_snake_case(       SCREAMING_SNAKE_CASE__     :      float       ,     SCREAMING_SNAKE_CASE__     :      float )				-> float:





    '''simple docstring'''






    A__        =  np.mean((original - contrast) ** 2 )
    if mse == 0:
        return 100

    return 20 * math.logaa(PIXEL_MAX / math.sqrt(SCREAMING_SNAKE_CASE__ ) )
def 		_snake_case(       )				-> None:





    '''simple docstring'''






    A__        =  os.path.dirname(os.path.realpath(SCREAMING_SNAKE_CASE__ ) )
    # Loading images (original image and compressed image)
    A__        =  cva.imread(os.path.join(SCREAMING_SNAKE_CASE__       ,     'image_data/original_image.png' ) )
    A__        =  cva.imread(os.path.join(SCREAMING_SNAKE_CASE__       ,     'image_data/compressed_image.png' )       ,     1 )

    A__        =  cva.imread(os.path.join(SCREAMING_SNAKE_CASE__       ,     'image_data/PSNR-example-base.png' ) )
    A__        =  cva.imread(
        os.path.join(SCREAMING_SNAKE_CASE__       ,     'image_data/PSNR-example-comp-10.jpg' )       ,     1 )

    # Value expected: 29.73dB
    print('-- First Test --' )
    print(f'PSNR value is {peak_signal_to_noise_ratio(SCREAMING_SNAKE_CASE__       ,     SCREAMING_SNAKE_CASE__ )} dB' )

    # # Value expected: 31.53dB (Wikipedia Example)
    print('\n-- Second Test --' )
    print(f'PSNR value is {peak_signal_to_noise_ratio(SCREAMING_SNAKE_CASE__       ,     SCREAMING_SNAKE_CASE__ )} dB' )


if __name__ == "__main__":
     main()

