Function test_json_read

python/ray/data/tests/datasource/test_json.py:27–39

(
    ray_start_regular_shared, target_max_block_size_infinite_or_default, tmp_path
)

def test_json_read(
    ray_start_regular_shared, target_max_block_size_infinite_or_default, tmp_path
):
    df1 = pd.DataFrame({"one": [1, 2, 3], "two": ["a", "b", "c"]})
    path1 = os.path.join(tmp_path, "test1.json")
    df1.to_json(path1, orient="records", lines=True)
    ds = ray.data.read_json(path1)
    dsdf = ds.to_pandas()
    pd.testing.assert_frame_equal(df1.astype(dsdf.dtypes.to_dict()), dsdf)
    # Metadata ops.
    assert ds.count() == 3
    assert ds.input_files() == [path1]
    assert ds.schema() == Schema(pa.schema([("one", pa.int64()), ("two", pa.string())]))


def test_zipped_json_read(