code: def test_tree_parallel_map(input_args, expected_output):
      assert dtype_utils.cast_to_numpy(input_args) == expected_output
  
imports: import numpy
  import pytest
  from tensorflow_datasets.core.utils import dtype_utils
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: test_tree_parallel_map(input_args, expected_output) without return types