code: def test_many_columns_non_petastorm_dataset(
      many_columns_non_petastorm_dataset, reader_factory
  ):
      """Check if we can read a dataset with huge number of columns (1000 in this case)"""
      with reader_factory(
          many_columns_non_petastorm_dataset.url
      ) as reader:
          sample = next(reader)
          assert set(sample._fields) == set(
              many_columns_non_petastorm_dataset.data[0].keys()
          )
  
imports: import glob
  import itertools
  import numpy
  import pandas
  import pytest
  from pyarrow import parquet
  from petastorm import make_batch_reader, make_reader
  from petastorm.arrow_reader_worker import ArrowReaderWorker
  from petastorm.tests.test_common import create_test_scalar_dataset
  from petastorm.transform import TransformSpec
  from petastorm.unischema import UnischemaField
label_desc: Use image node_id 6 for calling a global function with example usage: test_many_columns_non_petastorm_dataset(many_columns_non_petastorm_dataset, reader_factory) without return types