code: def _sort(data, axes):
      cur = data
      for ax in axes:
          if ax < data.ndim:
              cur = np.sort(cur, axis=ax)
      return cur
  
file_path: mars/mars/learn/utils/tests/test_shuffle.py
imports: import numpy
  import pandas
  import pytest
  from ....None import tensor
  from ....None import dataframe
  from ....core import tile
  from ..None import shuffle
  from ..shuffle import LearnShuffle
label_desc: Use image node_id 2 for calling a global function with example usage: _sort(data, axes) and returns: cur