code: def test_set_axis_execution(setup):
      raw_df = pd.DataFrame(
          np.random.rand(10, 5), columns=["c1", "c2", "c3", "c4", "c5"]
      )
      df = md.DataFrame(raw_df, chunk_size=3)
  
      # test axis=0
      idx_data = np.arange(0, 10)
      np.random.shuffle(idx_data)
      new_idx = md.Index(idx_data, chunk_size=4)
  
      r = df.set_axis(new_idx)
      pd.testing.assert_frame_equal(
          r.execute().fetch(), raw_df.set_axis(idx_data)
      )
  
      new_idx = pd.Index(range(9, -1, -1))
      r = df.set_axis(new_idx)
      pd.testing.assert_frame_equal(
          r.execute().fetch(), raw_df.set_axis(new_idx)
      )
  
      df1 = df.copy()
      df1.index = pd.Index(range(9, -1, -1))
      pd.testing.assert_frame_equal(
          df1.execute().fetch(), raw_df.set_axis(new_idx)
      )
  
      ser = md.Series(idx_data)
      with pytest.raises(ValueError):
          df.set_axis(ser[ser > 5]).execute()
  
      # test axis=1
      new_axis = ["a1", "a2", "a3", "a4", "a5"]
      r = df.set_axis(new_axis, axis=1)
      pd.testing.assert_frame_equal(
          r.execute().fetch(), raw_df.set_axis(new_axis, axis=1)
      )
  
      r = df.set_axis(md.Index(new_axis, store_data=True), axis=1)
      pd.testing.assert_frame_equal(
          r.execute().fetch(), raw_df.set_axis(new_axis, axis=1)
      )
  
      df1 = df.copy()
      df1.columns = new_axis
      pd.testing.assert_frame_equal(
          df1.execute().fetch(), raw_df.set_axis(new_axis, axis=1)
      )
  
      with pytest.raises(ValueError):
          df.set_axis(["a1", "a2", "a3", "a4"], axis=1)
  
      # test series
      raw_series = pd.Series(np.random.rand(10))
      s = md.Series(raw_series, chunk_size=3)
  
      idx_data = np.arange(0, 10)
      np.random.shuffle(idx_data)
      new_idx = md.Index(idx_data, chunk_size=4)
  
      r = s.set_axis(new_idx)
      pd.testing.assert_series_equal(
          r.execute().fetch(), raw_series.set_axis(idx_data)
      )
  
      s1 = s.copy()
      s1.index = new_idx
      pd.testing.assert_series_equal(
          s1.execute().fetch(), raw_series.set_axis(idx_data)
      )
  
imports: import os
  import tempfile
  import mars
  import numpy
  import pandas
  import pytest
  from ....None import dataframe
  from ....None import tensor
  from ....utils import pd_release_version
  from ...datasource.read_csv import DataFrameReadCSV
  from ...datasource.read_sql import DataFrameReadSQL
  from ...datasource.read_parquet import DataFrameReadParquet
label_desc: Use image node_id 25 for calling a global function with example usage: test_set_axis_execution(setup) without return types