code: def test_cast_to_int32_return_int32():
      dataset = test_utils.generate_one_hot_labels(100, 10, "dataset")
      dataset = dataset.map(lambda x: tf.cast(x, tf.uint8))
      dataset = common.CastToInt32().transform(dataset)
      for data in dataset:
          assert data.dtype == tf.int32
          break
  
file_path: autokeras/autokeras/preprocessors/common_test.py
imports: import numpy
  import tensorflow
  from autokeras import test_utils
  from autokeras.preprocessors import common
  from autokeras.utils import data_utils
label_desc: Use image node_id 3 for calling a global function with example usage: test_cast_to_int32_return_int32() without return types