code: def test_k_folds():
      sd = StructuredDataset(
          df=df,
          label_names=["label"],
          protected_attribute_names=["two"],
      )
      folds = sd.split(4)
  
      assert len(folds) == 4
      assert all(
          f.features.shape[0]
          == f.labels.shape[0]
          == f.protected_attributes.shape[0]
          == len(f.instance_names)
          == f.instance_weights.shape[0]
          == 1
          for f in folds
      )
  
      folds = sd.split(3)
      assert folds[0].features.shape[0] == 2
  
imports: import numpy
  import pandas
  from aif360.datasets import StructuredDataset
label_desc: Use image node_id 3 for calling a global function with example usage: test_k_folds() without return types