code: def test_chunking(dataset_1k_url, nrows):
      reader_fn = {
          "csv": read_csv,
          "parquet": functools.partial(read_parquet, df_lib=PANDAS_DF),
      }
  
      format = figure_data_format_dataset(dataset_1k_url)
  
      assert reader_fn[format](dataset_1k_url, nrows=nrows).shape[
          0
      ] == (nrows if nrows else 1000)
  
imports: import functools
  import json
  import logging
  import numpy
  import pandas
  import pytest
  from fsspec.config import conf
  from ludwig.api import LudwigModel
  from ludwig.data.cache.types import CacheableDataframe
  from ludwig.data.dataset_synthesizer import build_synthetic_dataset_df
  from ludwig.utils.data_utils import add_sequence_feature_column, figure_data_format_dataset, get_abs_path, hash_dict, NumpyEncoder, PANDAS_DF, read_csv, read_html, read_parquet, sanitize_column_names, use_credentials
  from tests.integration_tests.utils import private_param
label_desc: Use image node_id 9 for calling a global function with example usage: test_chunking(dataset_1k_url, nrows) without return types