code: def test_train_test_split_dataframe(setup):
      X = np.ones(10)
      types = [pd.DataFrame, md.DataFrame]
      for InputFeatureType in types:
          # X dataframe
          X_df = InputFeatureType(X)
          X_train, X_test = train_test_split(X_df)
          assert isinstance(X_train, DATAFRAME_TYPE)
          assert isinstance(X_test, DATAFRAME_TYPE)
  
imports: import itertools
  import numpy
  import pandas
  import pytest
  from ....None import dataframe
  from ....None import tensor
  from ....dataframe.core import DATAFRAME_TYPE
  from ....lib.sparse import SparseNDArray
  from ...utils.validation import _num_samples
  from ..None import train_test_split, KFold
label_desc: Use image node_id 5 for calling a global function with example usage: test_train_test_split_dataframe(setup) without return types