






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__    (	A       ):
   lowercase__    :						Optional[int]       =			tmp_path / """file.jsonl"""
   lowercase__    :						Optional[Any]       =			textwrap.dedent(
       """\
        {\"col_1\": -1}
        {\"col_1\": 1, \"col_2\": 2}
        {\"col_1\": 10, \"col_2\": 20}
        """       )
   with open(A		,					"""w"""       ) as f:
      f.write(A       )
   return str(A       )
@pytest.fixture
def    a__    (	A       ):
   lowercase__    :						int       =			tmp_path / """file_utf16_encoded.jsonl"""
   lowercase__    :						str       =			textwrap.dedent(
       """\
        {\"col_1\": -1}
        {\"col_1\": 1, \"col_2\": 2}
        {\"col_1\": 10, \"col_2\": 20}
        """       )
   with open(A		,					"""w"""		,					encoding="""utf-16"""       ) as f:
      f.write(A       )
   return str(A       )
@pytest.fixture
def    a__    (	A       ):
   lowercase__    :						Any       =			tmp_path / """file_with_list_of_dicts.json"""
   lowercase__    :						Union[str, Any]       =			textwrap.dedent(
       """\
        [
            {\"col_1\": -1},
            {\"col_1\": 1, \"col_2\": 2},
            {\"col_1\": 10, \"col_2\": 20}
        ]
        """       )
   with open(A		,					"""w"""       ) as f:
      f.write(A       )
   return str(A       )
@pytest.fixture
def    a__    (	A       ):
   lowercase__    :						int       =			tmp_path / """file_with_list_of_dicts_field.json"""
   lowercase__    :						List[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(A		,					"""w"""       ) as f:
      f.write(A       )
   return str(A       )
@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__    (	A		,					A		,					A       ):
   lowercase__    :						Optional[Any]       =			Json(**A       )
   lowercase__    :						List[str]       =			json._generate_tables([[request.getfixturevalue(A       )]]       )
   lowercase__    :						List[str]       =			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__    (	A		,					A		,					A       ):
   lowercase__    :						Union[str, Any]       =			Json(**A       )
   lowercase__    :						Any       =			json._generate_tables([[request.getfixturevalue(A       )]]       )
   lowercase__    :						Tuple       =			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]}










def    a__    (	A       ):
   return " ".join([word[::-1] for word in input_str.split()]       )


if __name__ == "__main__":
 import doctest

 doctest.testmod()


