code: def test_simple_read_tensorflow_with_non_petastorm_many_columns_dataset(
      many_columns_non_petastorm_dataset,
  ):
      """Read couple of rows. Make sure all tensors have static shape sizes assigned and the data matches reference
      data"""
      with make_batch_reader(
          dataset_url_or_urls=many_columns_non_petastorm_dataset.url
      ) as reader:
          row_tensors = tf_tensors(reader)
          # Make sure we have static shape info for all fields
          for column in row_tensors:
              assert column.get_shape().as_list() == [None]
  
          with _tf_session() as sess:
              batch = sess.run(row_tensors)._asdict()
              assert set(batch.keys()) == set(
                  many_columns_non_petastorm_dataset.data[0].keys()
              )
  
imports: from __future__ import division
  import datetime
  from calendar import timegm
  from collections import namedtuple
  from contextlib import contextmanager
  from decimal import Decimal
  from functools import wraps
  import inspect
  import numpy
  from packaging import version
  import pytest
  import tensorflow.compat.v1
  from unittest import mock
  from petastorm import make_reader, make_batch_reader, TransformSpec
  from petastorm.ngram import NGram
  from petastorm.tests.test_common import TestSchema
  from petastorm.tf_utils import _sanitize_field_tf_types, _numpy_to_tf_dtypes, _schema_to_tf_dtypes, tf_tensors, _ngrams_generator, _unflatten_and_set_shape
  from petastorm.unischema import Unischema, UnischemaField
label_desc: Use image node_id 17 for calling a global function with example usage: test_simple_read_tensorflow_with_non_petastorm_many_columns_dataset(many_columns_non_petastorm_dataset) without return types