code: def test_mars_dataset_script(setup_cluster):
      sess = setup_cluster
      path = os.path.join(
          os.path.dirname(os.path.abspath(__file__)), "tf_dataset.py"
      )
  
      data = mt.random.rand(1000, 32, dtype="f4")
      labels = mt.random.randint(0, 2, (1000, 10), dtype="f4")
  
      assert (
          run_tensorflow_script(
              path,
              n_workers=2,
              data={"feature_data": data, "labels": labels},
              command_argv=["multiple"],
              session=sess,
          ).fetch()["status"]
          == "ok"
      )
  
file_path: mars/mars/learn/contrib/tensorflow/tests/test_dataset.py
imports: import pytest
  import os
  from .....None import tensor
  from .....None import dataframe
  from .....utils import lazy_import
  from ..None import gen_tensorflow_dataset, run_tensorflow_script
label_desc: Use image node_id 2 for calling a global function with example usage: test_mars_dataset_script(setup_cluster) without return types